From 46de3b606d17ff72d72b0dca18850f03ed7ee107 Mon Sep 17 00:00:00 2001 From: Chih-Yu Yeh Date: Tue, 21 May 2024 09:32:48 +0800 Subject: [PATCH] fix (#278) --- wren-ai-service/src/utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/wren-ai-service/src/utils.py b/wren-ai-service/src/utils.py index 3a4a64fbc..04b901aaf 100644 --- a/wren-ai-service/src/utils.py +++ b/wren-ai-service/src/utils.py @@ -171,6 +171,8 @@ def init_providers() -> Tuple[LLMProvider, DocumentStoreProvider]: logger.info("Initializing providers...") loader.import_mods() - llm_provider = loader.get_provider(os.getenv("LLM_PROVIDER")) - document_store_provider = loader.get_provider(os.getenv("DOCUMENT_STORE_PROVIDER")) + llm_provider = loader.get_provider(os.getenv("LLM_PROVIDER", "openai")) + document_store_provider = loader.get_provider( + os.getenv("DOCUMENT_STORE_PROVIDER", "qdrant") + ) return llm_provider(), document_store_provider()