connector.py 437 B

1234567891011121314
  1. from open_webui.config import VECTOR_DB
  2. if VECTOR_DB == "milvus":
  3. from open_webui.apps.retrieval.vector.dbs.milvus import MilvusClient
  4. VECTOR_DB_CLIENT = MilvusClient()
  5. elif VECTOR_DB == "qdrant":
  6. from open_webui.apps.retrieval.vector.dbs.qdrant import QdrantClient
  7. VECTOR_DB_CLIENT = QdrantClient()
  8. else:
  9. from open_webui.apps.retrieval.vector.dbs.chroma import ChromaClient
  10. VECTOR_DB_CLIENT = ChromaClient()