소스 검색

Moving new config to under retreval

David Tippett 6 달 전
부모
커밋
afca48028f
2개의 변경된 파일4개의 추가작업 그리고 12개의 파일을 삭제
  1. 0 12
      backend/open_webui/apps/rag/vector/connector.py
  2. 4 0
      backend/open_webui/apps/retrieval/vector/connector.py

+ 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
 
     VECTOR_DB_CLIENT = QdrantClient()
+elif VECTOR_DB == "opensearch":
+    from open_webui.apps.retrieval.vector.dbs.opensearch import OpenSearchClient
+
+    VECTOR_DB_CLIENT = OpenSearchClient()
 else:
     from open_webui.apps.retrieval.vector.dbs.chroma import ChromaClient