Explorar el Código

refac: Rename Qdrant multi-tenancy variable for improved clarity and consistency

Kiet Trinh hace 4 meses
padre
commit
418ac1a8da

+ 1 - 1
backend/open_webui/config.py

@@ -1751,7 +1751,7 @@ QDRANT_API_KEY = os.environ.get("QDRANT_API_KEY", None)
 QDRANT_ON_DISK = os.environ.get("QDRANT_ON_DISK", "false").lower() == "true"
 QDRANT_PREFER_GRPC = os.environ.get("QDRANT_PREFER_GRPC", "False").lower() == "true"
 QDRANT_GRPC_PORT = int(os.environ.get("QDRANT_GRPC_PORT", "6334"))
-QDRANT_MULTITENANCY = os.environ.get("QDRANT_MULTITENANCY", "false").lower() == "true"
+ENABLE_QDRANT_MULTITENANCY_MODE = os.environ.get("ENABLE_QDRANT_MULTITENANCY_MODE", "false").lower() == "true"
 
 # OpenSearch
 OPENSEARCH_URI = os.environ.get("OPENSEARCH_URI", "https://localhost:9200")

+ 2 - 2
backend/open_webui/retrieval/vector/factory.py

@@ -1,6 +1,6 @@
 from open_webui.retrieval.vector.main import VectorDBBase
 from open_webui.retrieval.vector.type import VectorType
-from open_webui.config import VECTOR_DB, QDRANT_MULTITENANCY
+from open_webui.config import VECTOR_DB, ENABLE_QDRANT_MULTITENANCY_MODE
 
 
 class Vector:
@@ -16,7 +16,7 @@ class Vector:
 
                 return MilvusClient()
             case VectorType.QDRANT:
-                if QDRANT_MULTITENANCY:
+                if ENABLE_QDRANT_MULTITENANCY_MODE:
                     from open_webui.retrieval.vector.dbs.qdrant_multitenancy import QdrantClient
 
                     return QdrantClient()