opensource/WrenAI
3
0
Fork 0
mirror of https://github.com/Canner/WrenAI.git synced 2026-09-24 23:29:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix (#278)

Browse Source
This commit is contained in:
Chih-Yu Yeh
2024-05-21 09:32:48 +08:00
committed by GitHub
parent 1fac9b76f3
commit 46de3b606d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wren-ai-service/src/utils.py
+4 -2
View File
@@ -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()