fix(ai): max_tokens 显式上限 + 默认超时 300s——修复 DeepSeek 输出截断

DeepSeek 官方不传 max_tokens 时默认 8192 token 硬切（finish_reason=length），JSON 截断 → 精修 invalid 失败；20k token 生成实测 110s，60s 超时也会切断。payload 显式 max_tokens（env SILICONFLOW_MAX_TOKENS/AI_TEXT_FILTER_MAX_TOKENS 覆盖）+ 超时默认 300s。

Author: Aparencia

app/src-tauri/src/ai_client.rs

@@ -8,16 +8,30 @@
 //! @ai-context: 既有 ai_text_filter.rs（REQ-085 文本复核）与 M2/M3 新增的
 //!              ai_note_refine.rs/ai_enrich.rs 共用本 client；审计/配额/缓存
 //!              挂钩由 command 层完成（现有模式：锁内 read-modify-write，
-//!              网络调用不持全局锁——防 60s 超时阻塞其他 AI 命令）。
+//!              网络调用不持全局锁——防 300s 超时阻塞其他 AI 命令）。
 //! @ai-context: 网络路径不单测（与 model_downloader 同口径）；payload 构建/
 //!              响应提取/JSON 解析为纯函数可单测。
 
 use crate::ai_settings::AiSettings;
 
 /// 单请求超时默认（秒；env SILICONFLOW_TIMEOUT_SECS 可覆盖）。
-pub const DEFAULT_TIMEOUT_SECS: u64 = 60;
+///
+/// @ai-context: 300s——2026-08-21 真机排查：DeepSeek 长输出实测 20k token
+///              需 ~110s，60s 默认会把长生成切断（响应体截断 → JSON 解析
+///              EOF → 精修 invalid 失败），按 5 倍余量上调。
+pub const DEFAULT_TIMEOUT_SECS: u64 = 300;
 /// 重试次数默认（429/5xx/传输错误；env SILICONFLOW_RETRIES 可覆盖）。
 pub const DEFAULT_MAX_RETRIES: u32 = 2;
+/// 输出 token 上限默认（env SILICONFLOW_MAX_TOKENS 可覆盖）。
+///
+/// @ai-context: 20000——2026-08-21 真机排查：DeepSeek 官方不传 max_tokens 时
+///              默认 8192 token 硬切（finish_reason=length，JSON 截断 →
+///              Parse EOF）；精修单片 ≤8000 字输入 → 结构化输出最坏
+///              ~15k token，20000 给足余量。其他提供商若拒绝超限值，
+///              用 env 调小（OpenAI 兼容 clamp/报错行为不一）。
+pub const DEFAULT_MAX_TOKENS: u32 = 20000;
+/// max_tokens env 覆盖键。
+const MAX_TOKENS_ENV: &str = "SILICONFLOW_MAX_TOKENS";
 
 /// 共享 client 配置（resolve 聚合：环境变量 > 设置 > 内置默认）。
 #[derive(Debug, Clone, PartialEq)]
@@ -27,6 +41,7 @@ pub struct AiClientConfig {
     pub model: String,
     pub timeout_secs: u64,
     pub max_retries: u32,
+    pub max_tokens: u32,
 }
 
 /// 归一化 AI 错误（REQ-145 失败原因四类 + 服务端/解析补充）。
@@ -87,6 +102,7 @@ impl AiClient {
             model,
             timeout_secs: env_parse("SILICONFLOW_TIMEOUT_SECS", DEFAULT_TIMEOUT_SECS),
             max_retries: env_parse("SILICONFLOW_RETRIES", DEFAULT_MAX_RETRIES as u64) as u32,
+            max_tokens: env_parse(MAX_TOKENS_ENV, DEFAULT_MAX_TOKENS as u64) as u32,
         })
     }
 
@@ -103,7 +119,7 @@ impl AiClient {
         if self.config.api_key.trim().is_empty() {
             return Err(AiClientError::Auth("未配置 API 密钥（设置页保存或配置环境变量）".to_string()));
         }
-        let payload = build_chat_payload(&self.config.model, system, user);
+        let payload = build_chat_payload(&self.config.model, system, user, self.config.max_tokens);
         let url = chat_completions_url(&self.config.base_url);
         let agent = ureq::AgentBuilder::new()
             .timeout(std::time::Duration::from_secs(self.config.timeout_secs.max(5)))
@@ -193,18 +209,23 @@ pub fn chat_completions_url(base_url: &str) -> String {
     format!("{}/chat/completions", base_url.trim_end_matches('/'))
 }
 
-/// 构建 chat/completions payload（纯函数可单测；temperature=0 + json_object）。
+/// 构建 chat/completions payload（纯函数可单测；temperature=0 + json_object +
+/// max_tokens 显式上限）。
 ///
 /// @ai-context: R1 系推理模型带 no_think=true 关闭思考标签（保 JSON 输出
 ///              稳定——2026-08 实测选型注意点，与 ai_text_filter 同款）。
-pub fn build_chat_payload(model: &str, system: &str, user: &str) -> serde_json::Value {
+/// @ai-context: max_tokens 显式传值（2026-08-21 真机排查）：DeepSeek 官方
+///              缺省 8192 token 硬切 → JSON 截断；调用方经 AiClientConfig
+///              注入（env SILICONFLOW_MAX_TOKENS 可覆盖）。
+pub fn build_chat_payload(model: &str, system: &str, user: &str, max_tokens: u32) -> serde_json::Value {
     let mut body = serde_json::json!({
         "model": model,
         "messages": [
             {"role": "system", "content": system},
             {"role": "user", "content": user}
         ],
         "temperature": 0,
+        "max_tokens": max_tokens,
         "response_format": {"type": "json_object"}
     });
     if model.to_lowercase().contains("r1") {

app/src-tauri/src/ai_client_tests.rs

@@ -30,22 +30,23 @@ fn chat_url_preserves_v1_segment() {
 
 #[test]
 fn payload_has_expected_shape() {
-    // temperature=0 + json_object + system/user 消息
-    let p = build_chat_payload("acme/model", "sys", "usr");
+    // temperature=0 + json_object + max_tokens 显式上限 + system/user 消息
+    let p = build_chat_payload("acme/model", "sys", "usr", 20000);
     assert_eq!(p["model"], "acme/model");
     assert_eq!(p["messages"][0]["role"], "system");
     assert_eq!(p["messages"][0]["content"], "sys");
     assert_eq!(p["messages"][1]["content"], "usr");
     assert_eq!(p["temperature"], 0);
+    assert_eq!(p["max_tokens"], 20000);
     assert_eq!(p["response_format"]["type"], "json_object");
 }
 
 #[test]
 fn payload_r1_disables_think() {
     // R1 系推理模型带 no_think=true（保 JSON 稳定）；非 R1 不带
-    let r1 = build_chat_payload("deepseek-ai/DeepSeek-R1-0528-Qwen3-8B", "s", "u");
+    let r1 = build_chat_payload("deepseek-ai/DeepSeek-R1-0528-Qwen3-8B", "s", "u", 20000);
     assert_eq!(r1["no_think"], true);
-    let other = build_chat_payload("qwen/qwen3", "s", "u");
+    let other = build_chat_payload("qwen/qwen3", "s", "u", 20000);
     assert!(other.get("no_think").is_none());
 }
 
@@ -81,14 +82,42 @@ fn parse_json_object_invalid_is_parse_error() {
 
 #[test]
 fn from_settings_resolves_defaults() {
-    // 无 env/凭据 → 设置值；密钥为空
+    // Arrange：清除相关环境变量（防宿主机泄漏——本机 User 级已设 DeepSeek
+    // 配置；沿用 ai_text_filter_tests 的 save/remove/restore 模式，末尾还原）
+    let keys = [
+        "SILICONFLOW_API_KEY",
+        "SILICONFLOW_BASE_URL",
+        "SILICONFLOW_MODEL",
+        "SILICONFLOW_TIMEOUT_SECS",
+        "SILICONFLOW_RETRIES",
+        "SILICONFLOW_MAX_TOKENS",
+    ];
+    let saved: Vec<(String, Option<String>)> =
+        keys.iter().map(|k| (k.to_string(), std::env::var(k).ok())).collect();
+    for k in &keys {
+        std::env::remove_var(k);
+    }
+    // Act：无 env/凭据 → 设置值；密钥为空
     let s = AiSettings::default();
     let c = AiClient::from_settings(&s, None).config;
+    // Assert：2026-08-21 真机排查后默认值——长生成超时 300s + 输出上限 20000 token
     assert_eq!(c.base_url, s.base_url);
     assert_eq!(c.model, s.model);
     assert!(c.api_key.is_empty());
-    assert_eq!(c.timeout_secs, 60);
+    assert_eq!(c.timeout_secs, 300);
     assert_eq!(c.max_retries, 2);
+    assert_eq!(c.max_tokens, 20000);
+    // env 覆盖 max_tokens（顺序执行，无并行竞争）
+    std::env::set_var("SILICONFLOW_MAX_TOKENS", "4096");
+    let c2 = AiClient::from_settings(&AiSettings::default(), None).config;
+    assert_eq!(c2.max_tokens, 4096);
+    // 清理（还原宿主环境，防污染其他测试）
+    for (k, v) in saved {
+        match v {
+            Some(val) => std::env::set_var(k, val),
+            None => std::env::remove_var(k),
+        }
+    }
 }
 
 #[test]
@@ -108,6 +137,7 @@ fn chat_json_without_key_is_auth_error() {
         model: "m".to_string(),
         timeout_secs: 5,
         max_retries: 0,
+        max_tokens: 20000,
     });
     match client.chat_json("sys", "usr") {
         Err(AiClientError::Auth(_)) => {}

app/src-tauri/src/ai_text_filter.rs

@@ -34,6 +34,9 @@ pub struct AiTextFilterConfig {
     pub timeout_secs: u64,
     /// 重试次数（429/5xx/传输错误；指数退避）
     pub max_retries: u32,
+    /// 输出 token 上限（2026-08-21 真机排查：缺省时 DeepSeek 8192 硬切
+    /// JSON 截断——与共享 client 同默认；AI_TEXT_FILTER_MAX_TOKENS 覆盖）
+    pub max_tokens: u32,
     /// 批量上限（段/请求；超量分批）
     pub batch_size: usize,
     pub prompt: TextFilterPrompt,
@@ -117,6 +120,10 @@ impl AiTextFilterConfig {
             enabled,
             timeout_secs: env_parse("AI_TEXT_FILTER_TIMEOUT_SECS", 60),
             max_retries: env_parse("AI_TEXT_FILTER_RETRIES", 2) as u32,
+            max_tokens: env_parse(
+                "AI_TEXT_FILTER_MAX_TOKENS",
+                crate::ai_client::DEFAULT_MAX_TOKENS as u64,
+            ) as u32,
             batch_size: (env_parse("AI_TEXT_FILTER_BATCH", 30) as usize).clamp(1, 100),
             prompt: TextFilterPrompt::bundled(),
         }
@@ -158,6 +165,9 @@ impl AiTextFilterAdapter {
             model: self.config.model.clone(),
             timeout_secs: self.config.timeout_secs,
             max_retries: self.config.max_retries,
+            // 2026-08-21 真机排查：显式输出上限（缺省时 DeepSeek 8192 token
+            // 硬切 JSON 截断）——与共享 client 同默认，env 可覆盖
+            max_tokens: self.config.max_tokens,
         });
         let raw = client
             .chat_text(&system, &user)

app/src-tauri/src/ai_text_filter_tests.rs

@@ -14,6 +14,7 @@ fn config() -> AiTextFilterConfig {
         enabled: true,
         timeout_secs: 30,
         max_retries: 2,
+        max_tokens: 20000,
         batch_size: 30,
         prompt: TextFilterPrompt::bundled(),
     }
@@ -60,7 +61,7 @@ fn payload_has_zero_temperature_and_json_mode() {
     let cfg = config();
     let system = build_system_prompt(&cfg.prompt);
     let user = serde_json::to_string(&request().segments).unwrap();
-    let payload = crate::ai_client::build_chat_payload(&cfg.model, &system, &user);
+    let payload = crate::ai_client::build_chat_payload(&cfg.model, &system, &user, 20000);
     // Assert
     assert_eq!(payload["temperature"], 0);
     assert_eq!(payload["response_format"]["type"], "json_object");
@@ -78,7 +79,7 @@ fn non_r1_model_no_think_absent() {
     // Arrange：非推理模型（Qwen3-30B 付费档）
     let model = "Qwen/Qwen3-30B-A3B-Instruct-2507".to_string();
     // Act
-    let payload = crate::ai_client::build_chat_payload(&model, "sys", "usr");
+    let payload = crate::ai_client::build_chat_payload(&model, "sys", "usr", 20000);
     // Assert：不注入 no_think（未知参数可能被严格端点拒绝）
     assert!(payload.get("no_think").is_none());
 }