You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
716 B
22 lines
716 B
from open_webui.config import VECTOR_DB
|
|
|
|
if VECTOR_DB == "milvus":
|
|
from open_webui.retrieval.vector.dbs.milvus import MilvusClient
|
|
|
|
VECTOR_DB_CLIENT = MilvusClient()
|
|
elif VECTOR_DB == "qdrant":
|
|
from open_webui.retrieval.vector.dbs.qdrant import QdrantClient
|
|
|
|
VECTOR_DB_CLIENT = QdrantClient()
|
|
elif VECTOR_DB == "opensearch":
|
|
from open_webui.retrieval.vector.dbs.opensearch import OpenSearchClient
|
|
|
|
VECTOR_DB_CLIENT = OpenSearchClient()
|
|
elif VECTOR_DB == "pgvector":
|
|
from open_webui.retrieval.vector.dbs.pgvector import PgvectorClient
|
|
|
|
VECTOR_DB_CLIENT = PgvectorClient()
|
|
else:
|
|
from open_webui.retrieval.vector.dbs.chroma import ChromaClient
|
|
|
|
VECTOR_DB_CLIENT = ChromaClient()
|
|
|