Patch lanceDB not deleting vectors from workspace (#655)

patch lanceDB not deleting vectors from workspace documentVectors self-sanitize on delete of parent document
Mintplex-Labs · Jan 29, 2024 · dfab14a · dfab14a
1 parent 9d41ff5
commit dfab14a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/server/models/documents.js b/server/models/documents.js
@@ -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);
       }

diff --git a/server/utils/vectorDbProviders/lance/index.js b/server/utils/vectorDbProviders/lance/index.js
@@ -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 });
         }