sij/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2025-04-23 13:08:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions1

Patch lanceDB not deleting vectors from workspace ()

Browse source 
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
server
models
documents.js
utils/vectorDbProviders/lance
index.js

3
server/models/documents.js
View file

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

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

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