support deepseek llm

This commit is contained in:
harry
2024-05-13 21:48:53 +08:00
parent 4c21a23a8b
commit fee226a149
5 changed files with 29 additions and 3 deletions

View File

@@ -56,7 +56,7 @@ listen_port = _cfg.get("listen_port", 8080)
project_name = _cfg.get("project_name", "MoneyPrinterTurbo")
project_description = _cfg.get("project_description",
"<a href='https://github.com/harry0703/MoneyPrinterTurbo'>https://github.com/harry0703/MoneyPrinterTurbo</a>")
project_version = _cfg.get("project_version", "1.1.6")
project_version = _cfg.get("project_version", "1.1.7")
reload_debug = False
imagemagick_path = app.get("imagemagick_path", "")

View File

@@ -64,6 +64,12 @@ def _generate_response(prompt: str) -> str:
model_name = config.app.get("cloudflare_model_name")
account_id = config.app.get("cloudflare_account_id")
base_url = "***"
elif llm_provider == "deepseek":
api_key = config.app.get("deepseek_api_key")
model_name = config.app.get("deepseek_model_name")
base_url = config.app.get("deepseek_base_url")
if not base_url:
base_url = "https://api.deepseek.com"
else:
raise ValueError("llm_provider is not set, please set it in the config.toml file.")