Update milvus_store.py (#1212)

eosphoros-ai · Feb 29, 2024 · d6fabe9 · d6fabe9
1 parent 87ec9c6
commit d6fabe9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dbgpt/storage/vector_store/milvus_store.py b/dbgpt/storage/vector_store/milvus_store.py
@@ -165,7 +165,7 @@ def init_schema_and_load(self, vector_name, documents) -> List[str]:
             connections.connect(
                 host=self.uri or "127.0.0.1",
                 port=self.port or "19530",
-                alias="default"
+                alias="default",
                 # secure=self.secure,
             )
         texts = [d.content for d in documents]
@@ -419,7 +419,7 @@ def delete_vector_name(self, vector_name):
 
         """milvus delete collection name"""
         logger.info(f"milvus vector_name:{vector_name} begin delete...")
-        utility.drop_collection(vector_name)
+        utility.drop_collection(self.collection_name)
         return True
 
     def delete_by_ids(self, ids):