瀏覽代碼

Moving new config to under retreval

David Tippett 5 月之前
父節點
當前提交
afca48028f

+ 0 - 12
backend/open_webui/apps/rag/vector/connector.py

@@ -1,12 +0,0 @@
-from open_webui.apps.rag.vector.dbs.chroma import ChromaClient
-from open_webui.apps.rag.vector.dbs.milvus import MilvusClient
-
-
-from open_webui.config import VECTOR_DB
-
-if VECTOR_DB == "milvus":
-    VECTOR_DB_CLIENT = MilvusClient()
-elif VECTOR_DB == "opensearch":
-    VECTOR_DB_CLIENT = OpenSearchClient()
-else:
-    VECTOR_DB_CLIENT = ChromaClient()

+ 4 - 0
backend/open_webui/apps/retrieval/vector/connector.py

@@ -8,6 +8,10 @@ elif VECTOR_DB == "qdrant":
     from open_webui.apps.retrieval.vector.dbs.qdrant import QdrantClient
     from open_webui.apps.retrieval.vector.dbs.qdrant import QdrantClient
 
 
     VECTOR_DB_CLIENT = QdrantClient()
     VECTOR_DB_CLIENT = QdrantClient()
+elif VECTOR_DB == "opensearch":
+    from open_webui.apps.retrieval.vector.dbs.opensearch import OpenSearchClient
+
+    VECTOR_DB_CLIENT = OpenSearchClient()
 else:
 else:
     from open_webui.apps.retrieval.vector.dbs.chroma import ChromaClient
     from open_webui.apps.retrieval.vector.dbs.chroma import ChromaClient