OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

First pass at an OpenSearch connector

Browse Source
This commit is contained in:
David Tippett
2024-10-29 20:28:37 -04:00
parent 9d6cfe64ba
commit fb30b667e2
3 changed files with 169 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -0,0 +1,12 @@
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()