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()