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/Services/VectorSearchService.cs
+3 -3
View File
@@ -52,7 +52,7 @@ public class VectorSearchService(ApplicationDbContext dbContext, ITextEmbeddingG
public async Task<IEnumerable<Document>> 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)