Jelajahi Sumber

Merge pull request #11141 from Youggls/dev

fix: correct parameter name for MilvusClient instantiation
Timothy Jaeryang Baek 1 bulan lalu
induk
melakukan
7b442e4be0
1 mengubah file dengan 2 tambahan dan 2 penghapusan
  1. 2 2
      backend/open_webui/retrieval/vector/dbs/milvus.py

+ 2 - 2
backend/open_webui/retrieval/vector/dbs/milvus.py

@@ -20,9 +20,9 @@ class MilvusClient:
     def __init__(self):
         self.collection_prefix = "open_webui"
         if MILVUS_TOKEN is None:
-            self.client = Client(uri=MILVUS_URI, database=MILVUS_DB)
+            self.client = Client(uri=MILVUS_URI, db_name=MILVUS_DB)
         else:
-            self.client = Client(uri=MILVUS_URI, database=MILVUS_DB, token=MILVUS_TOKEN)
+            self.client = Client(uri=MILVUS_URI, db_name=MILVUS_DB, token=MILVUS_TOKEN)
 
     def _result_to_get_result(self, result) -> GetResult:
         ids = []