From 67ed61d022a24067789a79e5f06bc8a8e6d1e9f8 Mon Sep 17 00:00:00 2001 From: Perry Li Date: Tue, 4 Mar 2025 16:02:19 +0800 Subject: [PATCH] fixbug: correct parameter name for MilvusClient instantiation Replace incorrect parameter 'database=MILVUS_DB' with valid 'db_name=MILVUS_DB' --- backend/open_webui/retrieval/vector/dbs/milvus.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/retrieval/vector/dbs/milvus.py b/backend/open_webui/retrieval/vector/dbs/milvus.py index 3fd5b1ccd..ad05f9422 100644 --- a/backend/open_webui/retrieval/vector/dbs/milvus.py +++ b/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 = []