fix(ai): 流式收尾校验与整包回退/预算余量/授权文案（2026-09-05 审查批B）

Author: Aparencia

app/src-tauri/src/ai_chat_stream.rs

@@ -58,6 +58,9 @@ pub struct StreamOutcome {
     pub content: String,
     pub usage_json: Option<String>,
     pub cancelled: bool,
+    /// REQ-290①：是否收到 SSE [DONE] 正常收尾（false=断流/服务端 error——
+    /// 调用方应视作失败走重试/回退，禁止把截断当成功）
+    pub completed: bool,
 }
 
 /// 发送流式 chat/completions（SSE），逐 delta 回调 emit。
@@ -104,21 +107,25 @@ pub fn stream_chat(
     let mut content = String::new();
     let mut usage_json: Option<String> = None;
     let mut cancelled = false;
+    let mut completed = false;
     for line in reader.lines() {
         if cancel.is_cancelled() {
             cancelled = true;
             break;
         }
         let line = match line {
             Ok(l) => l,
-            Err(_) => break, // 传输中途断流：以已累积内容为准（无 usage）
+            Err(_) => break, // 传输中途断流：completed=false——调用方不得当成功
         };
         match parse_sse_line(&line) {
             SseEvent::Delta(d) => {
                 content.push_str(&d);
                 emit(ChatStreamEvent::Chunk { delta: d });
             }
-            SseEvent::Done => break,
+            SseEvent::Done => {
+                completed = true;
+                break;
+            }
             SseEvent::Ignore => {
                 // usage 可能挂在非 delta 的 data 行（OpenAI 兼容末 chunk）
                 if let Some(usage) = extract_usage(&line) {
@@ -127,7 +134,7 @@ pub fn stream_chat(
             }
         }
     }
-    Ok(StreamOutcome { content, usage_json, cancelled })
+    Ok(StreamOutcome { content, usage_json, cancelled, completed })
 }
 
 /// 从 data 行提取 usage（纯函数；无 usage → None）。
@@ -170,25 +177,29 @@ pub fn stream_sse_content(
     let reader = BufReader::new(resp.into_reader());
     let mut content = String::new();
     let mut usage_json: Option<String> = None;
+    let mut completed = false;
     for line in reader.lines() {
         let line = match line {
             Ok(l) => l,
-            Err(_) => break, // 传输中途断流：以已累积内容为准
+            Err(_) => break, // 传输中途断流：completed=false——调用方不得当成功
         };
         match parse_sse_line(&line) {
             SseEvent::Delta(d) => {
                 content.push_str(&d);
                 emit(&d);
             }
-            SseEvent::Done => break,
+            SseEvent::Done => {
+                completed = true;
+                break;
+            }
             SseEvent::Ignore => {
                 if let Some(usage) = extract_usage(&line) {
                     usage_json = Some(usage);
                 }
             }
         }
     }
-    Ok(StreamOutcome { content, usage_json, cancelled: false })
+    Ok(StreamOutcome { content, usage_json, cancelled: false, completed })
 }
 
 /// HTTP 状态 → AiClientError（与 post_completions 同归一口径——四下一致）。

app/src-tauri/src/ai_note_refine.rs

@@ -221,14 +221,16 @@ impl AiNoteRefineAdapter {
         let mut system = self.prompt.build_system(&request.profile, dims);
         system.push_str("\n\n");
         system.push_str(crate::ai_refine_protocol::NDJSON_SYSTEM_SUFFIX);
-        // REQ-290② 预算（与 refine_vision 同口径——流式不豁免上限）
+        // REQ-290② 预算（与 refine_vision 同口径——流式不豁免上限；含引导段）
         let budget = crate::refine_budget::output_budget(
             dims.map(|d| d.preset_id.as_str()).unwrap_or("standard"),
             request.content.chars().count(),
         );
+        system.push_str("\n\n");
+        system.push_str(&crate::refine_budget::guidance_suffix(&budget));
         let user = serde_json::to_string(request)
             .map_err(|e| AiClientError::Parse(format!("精修请求序列化失败: {}", e)))?;
-        let payload = if images.is_empty() {
+        let mut payload = if images.is_empty() {
             crate::ai_client::build_chat_payload(
                 &self.client.config.model, &system, &user, budget.max_tokens,
             )
@@ -237,6 +239,11 @@ impl AiNoteRefineAdapter {
                 &self.client.config.model, &system, &user, images, budget.max_tokens,
             )
         };
+        // B1（审查）：json_object=单一 JSON 值约束与 NDJSON 多行对象冲突——
+        // 流式拍必须去除 response_format（非流式回退路径保留既有约束）
+        if let Some(o) = payload.as_object_mut() {
+            o.remove("response_format");
+        }
         let mut sections: Vec<crate::ai_refine_protocol::AiRefineSection> = Vec::new();
         let mut pending = String::new();
         let outcome = crate::ai_chat_stream::stream_sse_content(
@@ -258,16 +265,47 @@ impl AiNoteRefineAdapter {
             sections.push(sec.clone());
             on_section(sec);
         }
-        let _ = outcome.content;
+        // B2（审查）：未收到 [DONE] 即断流——已累积节不可信（尾节可能丢失），
+        // 整体视作失败走同拍非流式回退（禁止静默截断当成功）
+        if !outcome.completed {
+            return Err(AiClientError::Network(
+                "流式响应未正常收尾（SSE 中断）——回退非流式".to_string(),
+            ));
+        }
+        // B3（审查）：整包回退承诺兑现——模型输出完整数组（紧凑/pretty）时
+        // 对已累积全文做整包解析（零额外请求；仅此路径无 Delta 帧）
+        if sections.is_empty() {
+            let text = outcome.content.trim();
+            if !text.is_empty() {
+                let v: serde_json::Value = serde_json::from_str(text)
+                    .map_err(|e| AiClientError::Parse(format!("流式整包解析失败: {}", e)))?;
+                let arr = v
+                    .get("sections")
+                    .cloned()
+                    .unwrap_or_else(|| v.clone());
+                let parsed: Vec<crate::ai_refine_protocol::AiRefineSection> =
+                    serde_json::from_value(arr).map_err(|e| {
+                        AiClientError::Parse(format!("流式整包结构非法: {}", e))
+                    })?;
+                if !parsed.is_empty() {
+                    sections = parsed;
+                }
+            }
+        }
         if sections.is_empty() {
             return Err(AiClientError::Parse(
                 "流式响应未解析出章节（模型未按逐节输出——回退非流式）".to_string(),
             ));
         }
-        Ok(crate::ai_refine_protocol::AiRefineResponse {
+        let resp = crate::ai_refine_protocol::AiRefineResponse {
             schema_version: crate::ai_refine_protocol::SCHEMA_VERSION_V2,
             sections,
-        })
+        };
+        // B6（审查）：与 refine_vision 同口径——组装后过整包校验
+        resp.validate().map_err(|e| {
+            AiClientError::Parse(format!("流式响应校验失败（已丢弃回退）: {}", e))
+        })?;
+        Ok(resp)
     }
 }
 

app/src-tauri/src/refine_budget.rs

@@ -23,13 +23,15 @@ pub struct OutputBudget {
 ///              保持高上限（防截断毁 JSON 校验），极简提取激进收敛。
 fn budget_params(preset_id: &str) -> (f64, u32) {
     match preset_id {
-        // faithful=忠实整理（输出≈输入长度）· deep=深度改写（转述+导语≈输入）
-        "faithful" => (1.0, 16_000),
-        "standard" | "" => (0.9, 16_000),
-        "deep" => (0.95, 16_000),
+        // faithful=忠实整理（输出≈输入长度）· deep=深度改写（转述+导语≈输入）；
+        // 最坏档 8000 字片需 ~15k token 且 20k 为旧上限——18k 保留 ≥3k 包装余量
+        // （审查 B5：16k 余量 <1k 会让合法长输出先被截断再 Parse 失败）
+        "faithful" => (1.0, 18_000),
+        "standard" | "" => (0.9, 18_000),
+        "deep" => (0.95, 18_000),
         "minimal" => (0.3, 9_000),
         // custom 无档位语义 → 按标准兜底（REQ-279 已声明后端兜底口径）
-        _ => (0.9, 16_000),
+        _ => (0.9, 18_000),
     }
 }
 

app/src/components/RefineLaunchDialog.tsx

@@ -172,11 +172,21 @@ export default function RefineLaunchDialog({
           <span style={{ fontSize: 11, color: "#6b7280" }}>{isNote ? "目标：当前笔记" : "目标：来源会话"}</span>
         </div>
 
-        {/* 授权卡（首次：上传说明 + 同意） */}
+        {/* 授权卡（首次：上传说明 + 同意）——审查 B11：文案按画面理解生效态条件化，
+            开启时明确"关键帧将随本次上传"（不再声称图像永不出本机） */}
         {!settings?.authorized && (
           <div style={{ border: "1px solid #f59e0b", background: "#fffbeb", borderRadius: 8, padding: 10, marginBottom: 10, fontSize: 12 }}>
             <div style={{ fontWeight: 600, marginBottom: 4 }}>首次使用需授权</div>
-            精修将上传<strong>转写文本、笔记内容与最小上下文</strong>至 AI 服务；本地优先铁律：<strong>音视频/图像永不出本机</strong>。是否同意？
+            {effectiveVision ? (
+              <>
+                精修将上传<strong>转写文本、笔记内容与关键帧图片</strong>至 AI 服务（画面理解已开启——关键帧仅本次随请求上云，不落云端存储）。
+              </>
+            ) : (
+              <>
+                精修将上传<strong>转写文本、笔记内容与最小上下文</strong>至 AI 服务；本地优先铁律：<strong>音视频/图像永不出本机</strong>。
+              </>
+            )}
+            是否同意？
             <div style={{ marginTop: 8, display: "flex", gap: 6 }}>
               <button style={{ ...btn, background: "#0d9488", color: "#fff", border: "none" }} onClick={() => void consent()}>同意并继续</button>
               <button style={btn} onClick={onClose}>暂不</button>