Optimize chunks retrieval

This commit is contained in:
Marco Minerva
2024-07-19 11:01:44 +02:00
parent 21b61281a7
commit c1180d3a4f
2 changed files with 3 additions and 3 deletions
@@ -104,6 +104,7 @@ public class VectorSearchService(ApplicationDbContext dbContext, ITextEmbeddingG
var chunks = await dbContext.DocumentChunks
.OrderBy(c => EF.Functions.VectorDistance("cosine", c.Embedding, questionEmbedding.ToArray()))
.Select(c => c.Content)
.Take(appSettings.MaxRelevantChunks)
.ToListAsync();