If the llm_provider parameter is not provided, openai will be used as the default

2024-03-16 09:00:28 +08:00
parent b5ba1e6b09
commit d8b3a1af49
1 changed files with 1 additions and 1 deletions
--- a/app/services/llm.py
+++ b/app/services/llm.py
@@ -9,7 +9,7 @@ from app.config import config


 def _generate_response(prompt: str) -> str:
-    llm_provider = config.app.get("llm_provider")
+    llm_provider = config.app.get("llm_provider", "openai")
    if llm_provider == "moonshot":
        api_key = config.app.get("moonshot_api_key")
        model_name = config.app.get("moonshot_model_name")