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

This commit is contained in:
harry 2024-03-16 09:00:28 +08:00
parent b5ba1e6b09
commit d8b3a1af49

View File

@ -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")