Feat/add annotation migrate (#2675)

Co-authored-by: jyong <jyong@dify.ai>
2024-03-04 17:22:06 +08:00
parent f322d9bddb
commit 3631e53ff0
2 changed files with 111 additions and 3 deletions
--- a/api/core/rag/datasource/vdb/milvus/milvus_vector.py
+++ b/api/core/rag/datasource/vdb/milvus/milvus_vector.py
@@ -140,7 +140,8 @@ class MilvusVector(BaseVector):
        connections.connect(alias=alias, uri=uri, user=self._client_config.user, password=self._client_config.password)

        from pymilvus import utility
-        utility.drop_collection(self._collection_name, None, using=alias)
+        if utility.has_collection(self._collection_name, using=alias):
+            utility.drop_collection(self._collection_name, None, using=alias)

    def text_exists(self, id: str) -> bool: