diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py index 00a5456ed..f7d2f46e5 100644 --- a/backend/open_webui/config.py +++ b/backend/open_webui/config.py @@ -1194,6 +1194,7 @@ CHROMA_HTTP_SSL = os.environ.get("CHROMA_HTTP_SSL", "false").lower() == "true" # Milvus MILVUS_URI = os.environ.get("MILVUS_URI", f"{DATA_DIR}/vector_db/milvus.db") +MILVUS_DB = os.environ.get("MILVUS_DB", "default") # Qdrant QDRANT_URI = os.environ.get("QDRANT_URI", None) diff --git a/backend/open_webui/retrieval/vector/dbs/milvus.py b/backend/open_webui/retrieval/vector/dbs/milvus.py index 31d890664..22bb87f3a 100644 --- a/backend/open_webui/retrieval/vector/dbs/milvus.py +++ b/backend/open_webui/retrieval/vector/dbs/milvus.py @@ -6,14 +6,14 @@ from typing import Optional from open_webui.retrieval.vector.main import VectorItem, SearchResult, GetResult from open_webui.config import ( - MILVUS_URI, + MILVUS_URI, MILVUS_DB, ) class MilvusClient: def __init__(self): self.collection_prefix = "open_webui" - self.client = Client(uri=MILVUS_URI) + self.client = Client(uri=MILVUS_URI, database=MILVUS_DB) def _result_to_get_result(self, result) -> GetResult: ids = []