Преглед изворни кода

fix weaviate delete_by_ids (#2565)

Bowen Liang пре 1 година
родитељ
комит
562ca45e07
1 измењених фајлова са 5 додато и 4 уклоњено
  1. 5 4
      api/core/rag/datasource/vdb/weaviate/weaviate_vector.py

+ 5 - 4
api/core/rag/datasource/vdb/weaviate/weaviate_vector.py

@@ -150,10 +150,11 @@ class WeaviateVector(BaseVector):
         return True
 
     def delete_by_ids(self, ids: list[str]) -> None:
-        self._client.data_object.delete(
-            ids,
-            class_name=self._collection_name
-        )
+        for uuid in ids:
+            self._client.data_object.delete(
+                class_name=self._collection_name,
+                uuid=uuid,
+            )
 
     def search_by_vector(self, query_vector: list[float], **kwargs: Any) -> list[Document]:
         """Look up similar documents by embedding vector in Weaviate."""