Mirrors/SqlDatabaseVectorSearch
1
0
Fork 0
mirror of https://github.com/marcominerva/SqlDatabaseVectorSearch.git synced 2026-06-20 12:23:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Little refactoring

Browse Source
This commit is contained in:
Marco Minerva
2024-07-01 09:49:34 +02:00
parent 8ef8836075
commit f3a0ec7c31
4 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SqlDatabaseVectorSearch/DataAccessLayer/ApplicationDbContext.cs
+1 -1
View File
@@ -36,7 +36,7 @@ public class ApplicationDbContext(DbContextOptions<ApplicationDbContext> options
.HasMaxLength(8000) .HasMaxLength(8000)
.IsVector(); .IsVector();
entity.HasOne(d => d.Document).WithMany(p => p.DocumentChunks) entity.HasOne(d => d.Document).WithMany(p => p.Chunks)
.HasForeignKey(d => d.DocumentId) .HasForeignKey(d => d.DocumentId)
.OnDelete(DeleteBehavior.NoAction) .OnDelete(DeleteBehavior.NoAction)
.HasConstraintName("FK_DocumentChunks_Documents"); .HasConstraintName("FK_DocumentChunks_Documents");
SqlDatabaseVectorSearch/DataAccessLayer/Entities/Document.cs
+1 -1
View File
@@ -8,5 +8,5 @@ public class Document
public DateTimeOffset CreationDate { get; set; } public DateTimeOffset CreationDate { get; set; }
public virtual ICollection<DocumentChunk> DocumentChunks { get; set; } = []; public virtual ICollection<DocumentChunk> Chunks { get; set; } = [];
} }
SqlDatabaseVectorSearch/Services/VectorSearchService.cs
+3 -3
View File
@@ -52,7 +52,7 @@ public class VectorSearchService(ApplicationDbContext dbContext, ITextEmbeddingG
public async Task<IEnumerable<Document>> GetDocumentsAsync() public async Task<IEnumerable<Document>> GetDocumentsAsync()
{ {
var documents = await dbContext.Documents.OrderBy(d => d.Name).AsNoTracking() 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(); .ToListAsync();
return documents; return documents;
@@ -60,13 +60,13 @@ public class VectorSearchService(ApplicationDbContext dbContext, ITextEmbeddingG
public async Task DeleteDocumentAsync(Guid documentId, bool saveChanges = true) 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) if (document is null)
{ {
return; return;
} }
dbContext.DocumentChunks.RemoveRange(document.DocumentChunks); dbContext.DocumentChunks.RemoveRange(document.Chunks);
dbContext.Documents.Remove(document); dbContext.Documents.Remove(document);
if (saveChanges) if (saveChanges)
SqlDatabaseVectorSearch/Settings/AppSettings.cs
+1 -1
View File
@@ -8,7 +8,7 @@ public class AppSettings
public int OverlapTokens { get; init; } = 100; public int OverlapTokens { get; init; } = 100;
public int MaxRelevantChunks { get; init; } = 6; public int MaxRelevantChunks { get; init; } = 5;
public int MessageLimit { get; init; } public int MessageLimit { get; init; }