Skip to content
Snippets Groups Projects
Unverified Commit dfab14a5 authored by Timothy Carambat's avatar Timothy Carambat Committed by GitHub
Browse files

Patch lanceDB not deleting vectors from workspace (#655)

patch lanceDB not deleting vectors from workspace
documentVectors self-sanitize on delete of parent document
parent 9d41ff58
No related branches found
No related tags found
No related merge requests found
...@@ -106,6 +106,9 @@ const Document = { ...@@ -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);
} }
......
...@@ -207,9 +207,9 @@ const LanceDb = { ...@@ -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 });
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment