fix: AI_MAX_TOKENS/AI_TEMPERATURE 未配置时使用模型默认值,避免截断分析输出

This commit is contained in:
Sakurasan
2026-09-01 17:58:27 +08:00
parent 86429f43c2
commit d0eafe203c
2 changed files with 6 additions and 4 deletions
+2 -2
View File
@@ -8,5 +8,5 @@ load_dotenv()
AI_API_BASE = os.getenv("AI_API_BASE", "https://openteam.oneisall.xyz/v1") AI_API_BASE = os.getenv("AI_API_BASE", "https://openteam.oneisall.xyz/v1")
AI_API_KEY = os.getenv("AI_API_KEY", "") AI_API_KEY = os.getenv("AI_API_KEY", "")
AI_MODEL = os.getenv("AI_MODEL", "deepseek-v4-flash") AI_MODEL = os.getenv("AI_MODEL", "deepseek-v4-flash")
AI_MAX_TOKENS = int(os.getenv("AI_MAX_TOKENS", "4096")) AI_MAX_TOKENS = int(os.getenv("AI_MAX_TOKENS")) if os.getenv("AI_MAX_TOKENS") else None
AI_TEMPERATURE = float(os.getenv("AI_TEMPERATURE", "0.7")) AI_TEMPERATURE = float(os.getenv("AI_TEMPERATURE")) if os.getenv("AI_TEMPERATURE") else None
+4 -2
View File
@@ -80,9 +80,11 @@ async def call_llm(messages: list, tools: list = None) -> dict:
payload = { payload = {
"model": AI_MODEL, "model": AI_MODEL,
"messages": messages, "messages": messages,
"max_tokens": AI_MAX_TOKENS,
"temperature": AI_TEMPERATURE,
} }
if AI_MAX_TOKENS is not None:
payload["max_tokens"] = AI_MAX_TOKENS
if AI_TEMPERATURE is not None:
payload["temperature"] = AI_TEMPERATURE
if tools: if tools:
payload["tools"] = tools payload["tools"] = tools
payload["tool_choice"] = "auto" payload["tool_choice"] = "auto"