From 7dbbb277e6d0ce04a9f4925ad8b762c48e763dc7 Mon Sep 17 00:00:00 2001 From: jyong <718720800@qq.com> Date: Tue, 18 Mar 2025 16:29:59 +0800 Subject: [PATCH] fix weaviate delete issue --- .../vdb/weaviate/weaviate_vector.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/api/core/rag/datasource/vdb/weaviate/weaviate_vector.py b/api/core/rag/datasource/vdb/weaviate/weaviate_vector.py index 753f3987a9..15bed54783 100644 --- a/api/core/rag/datasource/vdb/weaviate/weaviate_vector.py +++ b/api/core/rag/datasource/vdb/weaviate/weaviate_vector.py @@ -168,16 +168,16 @@ class WeaviateVector(BaseVector): # check whether the index already exists schema = self._default_schema(self._collection_name) if self._client.schema.contains(schema): - try: - self._client.batch.delete_objects( - class_name=self._collection_name, - where={"operator": "ContainsAny", "path": ["id"], "valueTextArray": ids}, - output="minimal", - ) - except weaviate.UnexpectedStatusCodeException as e: - # tolerate not found error - if e.status_code != 404: - raise e + for uuid in ids: + try: + self._client.data_object.delete( + class_name=self._collection_name, + uuid=uuid, + ) + except weaviate.UnexpectedStatusCodeException as e: + # tolerate not found error + if e.status_code != 404: + raise e def search_by_vector(self, query_vector: list[float], **kwargs: Any) -> list[Document]: """Look up similar documents by embedding vector in Weaviate."""