open-webui/backend/open_webui/apps/retrieval/vector/connector.py

from open_webui.config import VECTOR_DB

if VECTOR_DB == "milvus":
    from open_webui.apps.retrieval.vector.dbs.milvus import MilvusClient

    VECTOR_DB_CLIENT = MilvusClient()
else:
    from open_webui.apps.retrieval.vector.dbs.chroma import ChromaClient

    VECTOR_DB_CLIENT = ChromaClient()
refac 2024-09-10 09:27:50 +08:00			`from open_webui.config import VECTOR_DB`

refac 2024-09-10 11:37:06 +08:00			`if VECTOR_DB == "milvus":`
refac: lazy load big dependencies to reduce min memory usage 2024-10-03 05:18:42 +08:00			`from open_webui.apps.retrieval.vector.dbs.milvus import MilvusClient`

refac 2024-09-10 11:37:06 +08:00			`VECTOR_DB_CLIENT = MilvusClient()`
			`else:`
refac: lazy load big dependencies to reduce min memory usage 2024-10-03 05:18:42 +08:00			`from open_webui.apps.retrieval.vector.dbs.chroma import ChromaClient`

refac 2024-09-10 11:37:06 +08:00			`VECTOR_DB_CLIENT = ChromaClient()`