diff --git a/SqlDatabaseVectorSearch/DataAccessLayer/ApplicationDbContext.cs b/SqlDatabaseVectorSearch/DataAccessLayer/ApplicationDbContext.cs index 532b8e3..ec12f24 100644 --- a/SqlDatabaseVectorSearch/DataAccessLayer/ApplicationDbContext.cs +++ b/SqlDatabaseVectorSearch/DataAccessLayer/ApplicationDbContext.cs @@ -36,7 +36,7 @@ public class ApplicationDbContext(DbContextOptions options .HasMaxLength(8000) .IsVector(); - entity.HasOne(d => d.Document).WithMany(p => p.DocumentChunks) + entity.HasOne(d => d.Document).WithMany(p => p.Chunks) .HasForeignKey(d => d.DocumentId) .OnDelete(DeleteBehavior.NoAction) .HasConstraintName("FK_DocumentChunks_Documents"); diff --git a/SqlDatabaseVectorSearch/DataAccessLayer/Entities/Document.cs b/SqlDatabaseVectorSearch/DataAccessLayer/Entities/Document.cs index e041e4f..17f4b1a 100644 --- a/SqlDatabaseVectorSearch/DataAccessLayer/Entities/Document.cs +++ b/SqlDatabaseVectorSearch/DataAccessLayer/Entities/Document.cs @@ -8,5 +8,5 @@ public class Document public DateTimeOffset CreationDate { get; set; } - public virtual ICollection DocumentChunks { get; set; } = []; + public virtual ICollection Chunks { get; set; } = []; } diff --git a/SqlDatabaseVectorSearch/Services/VectorSearchService.cs b/SqlDatabaseVectorSearch/Services/VectorSearchService.cs index a579e25..b7bd990 100644 --- a/SqlDatabaseVectorSearch/Services/VectorSearchService.cs +++ b/SqlDatabaseVectorSearch/Services/VectorSearchService.cs @@ -52,7 +52,7 @@ public class VectorSearchService(ApplicationDbContext dbContext, ITextEmbeddingG public async Task> GetDocumentsAsync() { var documents = await dbContext.Documents.OrderBy(d => d.Name).AsNoTracking() - .Select(d => new Document(d.Id, d.Name, d.CreationDate, d.DocumentChunks.Count)) + .Select(d => new Document(d.Id, d.Name, d.CreationDate, d.Chunks.Count)) .ToListAsync(); return documents; @@ -60,13 +60,13 @@ public class VectorSearchService(ApplicationDbContext dbContext, ITextEmbeddingG public async Task DeleteDocumentAsync(Guid documentId, bool saveChanges = true) { - var document = await dbContext.Documents.Include(d => d.DocumentChunks).FirstOrDefaultAsync(d => d.Id == documentId); + var document = await dbContext.Documents.Include(d => d.Chunks).FirstOrDefaultAsync(d => d.Id == documentId); if (document is null) { return; } - dbContext.DocumentChunks.RemoveRange(document.DocumentChunks); + dbContext.DocumentChunks.RemoveRange(document.Chunks); dbContext.Documents.Remove(document); if (saveChanges) diff --git a/SqlDatabaseVectorSearch/Settings/AppSettings.cs b/SqlDatabaseVectorSearch/Settings/AppSettings.cs index c11aba0..4466e1a 100644 --- a/SqlDatabaseVectorSearch/Settings/AppSettings.cs +++ b/SqlDatabaseVectorSearch/Settings/AppSettings.cs @@ -8,7 +8,7 @@ public class AppSettings public int OverlapTokens { get; init; } = 100; - public int MaxRelevantChunks { get; init; } = 6; + public int MaxRelevantChunks { get; init; } = 5; public int MessageLimit { get; init; }