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

Feat/vdb migrate command (#2562)

Browse source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong 2024-02-26 19:47:29 +08:00 • committed by GitHub
commit 0620fa3094
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 134 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -127,7 +127,10 @@ class WeaviateVector(BaseVector):
)
def delete(self):
self._client.schema.delete_class(self._collection_name)
# check whether the index already exists
schema = self._default_schema(self._collection_name)
if self._client.schema.contains(schema):
self._client.schema.delete_class(self._collection_name)
def text_exists(self, id: str) -> bool:
collection_name = self._collection_name