outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix weaviate delete issue

Browse Source
This commit is contained in:
jyong 2025-03-18 16:29:59 +08:00
parent d57c82bfa8
commit 7dbbb277e6
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
api/core/rag/datasource/vdb/weaviate/weaviate_vector.py
View File

@ -168,16 +168,16 @@ class WeaviateVector(BaseVector):
# check whether the index already exists # check whether the index already exists
schema = self._default_schema(self._collection_name) schema = self._default_schema(self._collection_name)
if self._client.schema.contains(schema): if self._client.schema.contains(schema):
try: for uuid in ids:
self._client.batch.delete_objects( try:
class_name=self._collection_name, self._client.data_object.delete(
where={"operator": "ContainsAny", "path": ["id"], "valueTextArray": ids}, class_name=self._collection_name,
output="minimal", uuid=uuid,
) )
except weaviate.UnexpectedStatusCodeException as e: except weaviate.UnexpectedStatusCodeException as e:
# tolerate not found error # tolerate not found error
if e.status_code != 404: if e.status_code != 404:
raise e raise e
def search_by_vector(self, query_vector: list[float], **kwargs: Any) -> list[Document]: def search_by_vector(self, query_vector: list[float], **kwargs: Any) -> list[Document]:
"""Look up similar documents by embedding vector in Weaviate.""" """Look up similar documents by embedding vector in Weaviate."""