jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: unable to reindex documents (#5276)

Browse source
This commit is contained in:
sino 2024-06-17 17:19:43 +08:00 • committed by GitHub
commit be3512aa57
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
api/core/rag/datasource/vdb/weaviate/weaviate_vector.py
View file

@ -174,10 +174,15 @@ class WeaviateVector(BaseVector):
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):
for uuid in ids: for uuid in ids:
self._client.data_object.delete( try:
class_name=self._collection_name, self._client.data_object.delete(
uuid=uuid, 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]: 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."""