kamp
/
anything-llm
mirror of https://github.com/Mintplex-Labs/anything-llm.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Patch lanceDB not deleting vectors from workspace (#655) 

patch lanceDB not deleting vectors from workspace
documentVectors self-sanitize on delete of parent document
This commit is contained in:
Timothy Carambat 2024-01-29 09:49:22 -08:00 committed by GitHub
parent 9d41ff58e2
commit dfab14a5d2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/models/documents.js
View File

@ -106,6 +106,9 @@ const Document = {
await prisma.workspace_documents.delete({
where: { id: document.id, workspaceId: workspace.id },
});
await prisma.document_vectors.deleteMany({
where: { docId: document.docId },
});
} catch (error) {
console.error(error.message);
}

2
server/utils/vectorDbProviders/lance/index.js
View File

@ -207,9 +207,9 @@ const LanceDb = {
vectors.push(vectorRecord);
submissions.push({
...vectorRecord.metadata,
id: vectorRecord.id,
vector: vectorRecord.values,
...vectorRecord.metadata,
});
documentVectors.push({ docId, vectorId: vectorRecord.id });
}