UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 21:42:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Moving new config to under retreval

Browse Source
This commit is contained in:
David Tippett 2024-11-04 15:10:14 -05:00
parent 4baab4bce8
commit afca48028f
2 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/open_webui/apps/rag/vector/connector.py
View File

@ -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
backend/open_webui/apps/retrieval/vector/connector.py
View File

@ -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