refac: lazy load big dependencies to reduce min memory usage

This commit is contained in:
Jun Siang Cheah committed 2024-10-02 22:18:42 +01:00
1 parent d353ea449a
commit 318f61161e
2 files changed
+6 -7

No files matched your search

@@ -1,10 +1,10 @@
from open_webui.apps.retrieval.vector.dbs.chroma import ChromaClient
from open_webui.apps.retrieval.vector.dbs.milvus import MilvusClient
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()