Browse Source

Merge pull request #11141 from Youggls/dev

fix: correct parameter name for MilvusClient instantiation
Timothy Jaeryang Baek 1 month ago
parent
commit
7b442e4be0
1 changed files with 2 additions and 2 deletions
  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 = []