mirror of
https://github.com/open-webui/open-webui
synced 2024-11-17 14:02:51 +00:00
13 lines
351 B
Python
13 lines
351 B
Python
|
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()
|