@yfwu2020/dsh-selection-explain 0.11.1 → 0.13.0

package/lib/config-values.js

@@ -5,6 +5,7 @@ export const CONFIG_DEFAULTS = Object.freeze({
     chatModel: '',
     reasoningEffort: 'high',
     translationReasoningEffort: 'off',
+    chatContextTokens: 200000,
     historyMaxEntries: 20,
     temperature: -1,
     maxTokens: 0,
@@ -32,8 +33,12 @@ export const CONFIG_DEFAULTS = Object.freeze({
     pillIdleMs: 10000,
     voiceCancelOnSilence: true,
     voiceAutoSend: true,
+    shotCardScope: 'window',
+    shotPanelScope: 'selection',
+    shotMargin: 48,
 });
 export const CONFIG_LIMITS = {
+    chatContextTokens: { min: 4096, max: 262144 },
     historyMaxEntries: { min: 0, max: 500 },
     temperature: { min: -1, max: 2 },
     maxTokens: { min: 0, max: 32000 },
@@ -51,5 +56,6 @@ export const CONFIG_LIMITS = {
     maxToolRounds: { min: 0, max: 8 },
     toolResultMaxChars: { min: 200, max: 20000 },
     pillIdleMs: { min: 2000, max: 600000 },
+    shotMargin: { min: 0, max: 400 },
 };
 //# sourceMappingURL=config-values.js.map

package/lib/config-values.js.map

@@ -1 +1 @@
-{"version":3,"file":"config-values.js","sourceRoot":"","sources":["../src/config-values.ts"],"names":[],"mappings":"AAyGA,MAAM,CAAC,MAAM,eAAe,GAAqB,MAAM,CAAC,MAAM,CAAC;IAC3D,QAAQ,EAAE,EAAE;IACZ,KAAK,EAAE,EAAE;IACT,YAAY,EAAE,EAAE;IAChB,SAAS,EAAE,EAAE;IACb,eAAe,EAAE,MAAM;IACvB,0BAA0B,EAAE,KAAK;IACjC,iBAAiB,EAAE,EAAE;IACrB,WAAW,EAAE,CAAC,CAAC;IACf,SAAS,EAAE,CAAC;IACZ,SAAS,EAAE,MAAM;IACjB,iBAAiB,EAAE,IAAI;IACvB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,EAAE;IACxB,cAAc,EAAE,IAAI;IACpB,yBAAyB,EAAE,EAAE;IAC7B,0BAA0B,EAAE,CAAC;IAC7B,sBAAsB,EAAE,CAAC;IACzB,kBAAkB,EAAE,CAAC;IACrB,2BAA2B,EAAE,IAAI;IACjC,oBAAoB,EAAE,IAAI;IAC1B,gBAAgB,EAAE,MAAM;IACxB,KAAK,EAAE,IAAI;IACX,SAAS,EAAE,gDAAgD;IAC3D,iBAAiB,EAAE,YAAY;IAC/B,aAAa,EAAE,CAAC;IAChB,kBAAkB,EAAE,IAAI;IACxB,YAAY,EAAE,EAAE;IAChB,mBAAmB,EAAE,MAAM;IAC3B,oBAAoB,EAAE,IAAI;IAC1B,WAAW,EAAE,IAAI;IACjB,UAAU,EAAE,KAAK;IACjB,oBAAoB,EAAE,IAAI;IAC1B,aAAa,EAAE,IAAI;CACpB,CAAC,CAAA;AAEJ,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,iBAAiB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;IACvC,WAAW,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE;IAChC,SAAS,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE;IACjC,SAAS,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE;IACrC,iBAAiB,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;IAC1C,eAAe,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE;IACzC,oBAAoB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;IAC1C,yBAAyB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;IAC/C,0BAA0B,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;IAChD,sBAAsB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE;IAChD,kBAAkB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE;IACvC,2BAA2B,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE;IACrD,oBAAoB,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE;IAC/C,gBAAgB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE;IAC3C,aAAa,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE;IACjC,kBAAkB,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE;IAC5C,UAAU,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE;CAC9B,CAAA"}
+{"version":3,"file":"config-values.js","sourceRoot":"","sources":["../src/config-values.ts"],"names":[],"mappings":"AA+HA,MAAM,CAAC,MAAM,eAAe,GAAqB,MAAM,CAAC,MAAM,CAAC;IAC3D,QAAQ,EAAE,EAAE;IACZ,KAAK,EAAE,EAAE;IACT,YAAY,EAAE,EAAE;IAChB,SAAS,EAAE,EAAE;IACb,eAAe,EAAE,MAAM;IACvB,0BAA0B,EAAE,KAAK;IACjC,iBAAiB,EAAE,MAAM;IACzB,iBAAiB,EAAE,EAAE;IACrB,WAAW,EAAE,CAAC,CAAC;IACf,SAAS,EAAE,CAAC;IACZ,SAAS,EAAE,MAAM;IACjB,iBAAiB,EAAE,IAAI;IACvB,eAAe,EAAE,KAAK;IACtB,oBAAoB,EAAE,EAAE;IACxB,cAAc,EAAE,IAAI;IACpB,yBAAyB,EAAE,EAAE;IAC7B,0BAA0B,EAAE,CAAC;IAC7B,sBAAsB,EAAE,CAAC;IACzB,kBAAkB,EAAE,CAAC;IACrB,2BAA2B,EAAE,IAAI;IACjC,oBAAoB,EAAE,IAAI;IAC1B,gBAAgB,EAAE,MAAM;IACxB,KAAK,EAAE,IAAI;IACX,SAAS,EAAE,gDAAgD;IAC3D,iBAAiB,EAAE,YAAY;IAC/B,aAAa,EAAE,CAAC;IAChB,kBAAkB,EAAE,IAAI;IACxB,YAAY,EAAE,EAAE;IAChB,mBAAmB,EAAE,MAAM;IAC3B,oBAAoB,EAAE,IAAI;IAC1B,WAAW,EAAE,IAAI;IACjB,UAAU,EAAE,KAAK;IACjB,oBAAoB,EAAE,IAAI;IAC1B,aAAa,EAAE,IAAI;IACnB,aAAa,EAAE,QAAQ;IACvB,cAAc,EAAE,WAAW;IAC3B,UAAU,EAAE,EAAE;CACf,CAAC,CAAA;AAEJ,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,iBAAiB,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE;IAC7C,iBAAiB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;IACvC,WAAW,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE;IAChC,SAAS,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE;IACjC,SAAS,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE;IACrC,iBAAiB,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;IAC1C,eAAe,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE;IACzC,oBAAoB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;IAC1C,yBAAyB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;IAC/C,0BAA0B,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;IAChD,sBAAsB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE;IAChD,kBAAkB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE;IACvC,2BAA2B,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE;IACrD,oBAAoB,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE;IAC/C,gBAAgB,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE;IAC3C,aAAa,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE;IACjC,kBAAkB,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE;IAC5C,UAAU,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE;IACtC,UAAU,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE;CACxB,CAAA"}

package/lib/host/analyze.js

@@ -4,13 +4,16 @@ import { sessionIdOf } from './context.js';
 import { createAssistantMessage, createUserMessage, ReasoningEffortId, BlockAssembler, createToolResultMessage } from '@deepseek-ai/dsh-llm';
 import { name } from './constants.js';
 import { resolveToolNames, webFetchUsable, FS_TOOL_NAMES, describeToolCall, isReadAllowed, runTool, digestToolResult } from './tools.js';
-import { CHAT_SYSTEM_PROMPT, DETAIL_PROMPT, CODE_PROMPT, TRANSLATION_PROMPT } from '../prompts.js';
+import { CHAT_SYSTEM_PROMPT, DETAIL_PROMPT, CODE_PROMPT, TRANSLATION_PROMPT, shotPromptFor, shotTitlePromptFor } from '../prompts.js';
 import { loadDesignSkill } from './design-skill.js';
 import { systemMessageOf } from './messages.js';
 import { createThinkFilter, splitLeakedReasoning, looksLikeLeakedReasoning } from './filters.js';
 import { resolve } from 'node:path';
 import { looksLikeReasoningRequired, looksLikeUnsupportedTier, EFFORT_LABEL } from './effort.js';
 import { historyContentBlocks, looksLikeAttachmentFailure, readHistoryAttachments } from './history-attachments.js';
+import { contextSource, EMPTY_DIGEST, estimateTextTokens, readContextState } from '../chat-context.js';
+import { resolveChatCapacity, fitBackground, planChatContext, estimateRequestTokens } from './chat-context.js';
+import { readMaterialCatalog, materialIndex, MATERIAL_TOOL, restoreMaterial } from './chat-materials.js';
 /** Owns analyze state for one plugin instance. */
 export function createAnalyze(ctx, config, models, sessions, historyStore) {
     /** 分钟级限流。 */
@@ -69,11 +72,24 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
         const context = historyStore.clampText(body.context, config.maxContextChars);
         const label = historyStore.clampText(body.label, 80);
         // 页面标题/地址不写进提示词：模型会把它当成「选中文字里的英文」去翻译（实测把标题 DSH 译了一遍）
+        const initialExplanation = typeof body.initialExplanation === 'string' ? body.initialExplanation : '';
+        const source = contextSource(sessionIdOf(body), text, typeof body.context === 'string' ? body.context : '', initialExplanation);
+        const restoredState = readContextState(body.contextState);
+        const historyStart = Number.isSafeInteger(body.historyStart) && Number(body.historyStart) >= 0 ? Number(body.historyStart) : 0;
+        if (isChatRound && historyStart > 0 && (!restoredState || restoredState.source !== source || restoredState.through !== historyStart)) {
+            sendJson(res, 409, { error: '对话记忆进度与当前来源不一致，请重新打开这段历史后重试' });
+            return;
+        }
+        let contextState = restoredState && restoredState.source === source && restoredState.through === historyStart
+            ? { ...restoredState, sessionId: sessionIdOf(body) } : { version: 1, sessionId: sessionIdOf(body), source, through: 0, digest: EMPTY_DIGEST, summary: '', background: null };
+        const capacity = isChatRound ? await resolveChatCapacity(ctx, config, route) : null;
         // 会话背景：从 sessionQuery 读取当前会话的模型界面（live 优先，含持久化会话）
         let background = null;
         if (config.sessionContext) {
             const sessionId = sessionIdOf(body);
-            if (sessionId) {
+            if (isChatRound && restoredState && restoredState.source === source)
+                background = contextState.background;
+            else if (sessionId) {
                 // 背景预算按阶段：首轮只要翻译（8 条够定指代）；详解与追问都是深度交互，吃满 24 条
                 const stage = typeof body.stage === 'string' ? body.stage : '';
                 const wantsFull = stage === 'detail' || typeof body.question === 'string';
@@ -84,6 +100,10 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                 });
             }
         }
+        if (isChatRound && capacity && !restoredState)
+            background = fitBackground(background, Math.max(128, capacity.budget * 0.2));
+        if (isChatRound)
+            contextState = { ...contextState, background };
         // 顺序即缓存策略：前缀必须“只增不变”才能复用 ——
         //   ① 会话背景（同一会话内基本不变）→ ② 固定说明 → ③ 动态尾部（选中文字/局部片段）
         // 反过来写（动态在前、背景在后）会让缓存前缀在“选中文字”处就断掉：
@@ -92,16 +112,36 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
         if (background && background.transcript) {
             stablePrefix.push(`【会话背景】（按时间顺序；${background.marked ? '最后一条即选中文字所在的消息，已用 ▶ 标出，其后的对话未包含' : '未能定位选中位置，取的是最近的对话'}）`, background.transcript, '', '上面是会话背景；下面是需要你解读的【选中文字】及其上下文。', '');
         }
-        const userText = [
-            ...stablePrefix,
-            '【选中文字】',
-            text,
-            '',
-            `【所在位置】${label || '页面内容'}`,
-            '',
-            '【选中处的上下文片段】（选中部分用【】标出）',
-            context || '（未能取到上下文片段，请仅根据选中文字与会话背景作答）',
-        ].join('\n');
+        const isShotMode = body.kind === 'shot';
+        /**
+         * 截图取景范围：决定提示词口径（有没有红框、图是不是只有选区）。
+         * 老客户端不带时按"整窗 + 红框"兜底；非法值也退到整窗，避免把提示词写坏。
+         */
+        const shotScope = body.shotScope === 'screen' || body.shotScope === 'window' || body.shotScope === 'margin' || body.shotScope === 'selection'
+            ? body.shotScope
+            : 'window';
+        const shotFrame = body.shotFrame !== false;
+        /**
+         * 取标题（截图提问的第二次调用）。
+         *
+         * 关键：**前缀必须与回答那次逐字节一致**（同样的 system、同样的图、同样的提问），
+         * 这样图那部分能命中前缀缓存；只在最后追加一句"只输出标题"。
+         * 并且工具轮次强制为 0 —— 这次不该去调工具。
+         */
+        const titleOnly = isShotMode && body.titleOnly === true;
+        // 截图提问：**没有选中文字**，这轮的文字就是用户说的那句话，正文不该套"选中文字"模板
+        const userText = isShotMode
+            ? text
+            : [
+                ...stablePrefix,
+                '【选中文字】',
+                text,
+                '',
+                `【所在位置】${label || '页面内容'}`,
+                '',
+                '【选中处的上下文片段】（选中部分用【】标出）',
+                context || '（未能取到上下文片段，请仅根据选中文字与会话背景作答）',
+            ].join('\n');
         const effort = typeof body.effort === 'string' && body.effort.length > 0 ? body.effort : config.reasoningEffort;
         // 追问模式：客户端带 question 时，就是「就这段选中文字继续聊」
         const question = typeof body.question === 'string' ? body.question.trim() : '';
@@ -109,7 +149,16 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
         // 这一轮的附件（追问）：/api/attach 存好之后的持久引用。形状在这里**收敛一次**，
         // 历史那条路用的也是同一个收敛器（见 history-attachments.ts）。
         const currentAttachments = readHistoryAttachments(body.attachments);
-        const rawHistory = Array.isArray(body.history) ? body.history : [];
+        const rawHistory = (Array.isArray(body.history) ? body.history : []).map(raw => {
+            const item = (raw || {});
+            const attachments = readHistoryAttachments(item.attachments);
+            return { role: item.role === 'assistant' ? 'assistant' : 'user', text: typeof item.text === 'string' ? item.text : '',
+                ...(attachments.length ? { attachments } : {}) };
+        });
+        let selectedHistory = rawHistory;
+        const catalog = readMaterialCatalog([...(Array.isArray(body.materials) ? body.materials : []),
+            ...rawHistory.flatMap((t, i) => (t.attachments || []).map(a => ({ ...a, turn: historyStart + i }))),
+            ...currentAttachments.map(a => ({ ...a, turn: historyStart + rawHistory.length }))]);
         // 预算规划的结果（图片几张回带、几张降级），只用于上报与诊断
         let historyImages = { kept: 0, omitted: 0, files: 0 };
         // 历史里有没有真带过附件：决定"附件读不回来"时值不值得降级重试一次
@@ -125,7 +174,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
         function planHistory() {
             if (!isChat)
                 return { blocks: [], kept: [], keptImages: 0, omittedImages: 0, keptFiles: 0, omittedFiles: 0 };
-            const window = rawHistory.slice(-20);
+            const window = selectedHistory;
             const turns = historyDropped
                 ? window.map((turn) => {
                     const item = (turn ?? {});
@@ -141,7 +190,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
             const out = [];
             if (!isChat)
                 return out;
-            const window = rawHistory.slice(-20);
+            const window = selectedHistory;
             const plan = planHistory();
             window.forEach((turn, index) => {
                 const item = turn;
@@ -177,7 +226,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
             for (const item of currentAttachments)
                 if (item.kind === 'file')
                     refs.push(item.ref);
-            for (const item of planHistory().kept)
+            for (const item of catalog)
                 if (item.kind === 'file')
                     refs.push(item.ref);
             for (const ref of refs) {
@@ -283,6 +332,9 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                 ctx.logger?.warn?.(`[${name}] 网页抓取不可用（域名解析到 ${probe.sample}，属保留地址段），已从工具表移除 web_fetch`);
             }
         }
+        if (isChat && config.tools && config.maxToolRounds > 0 && (rawHistory.length > 0 || catalog.length > 0 || contextState.through > 0)) {
+            toolSchemas.push(MATERIAL_TOOL);
+        }
         // 文件类工具的边界：默认只允许会话工作目录；'*' = 不限制；配置了具体目录就用它。
         // 拿不到工作目录时**不设限**（否则新会话里 read 直接全废），但会打日志说明。
         const wantsFsTools = toolSchemas.some((schema) => FS_TOOL_NAMES.includes(schema.name));
@@ -300,13 +352,17 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
         const stage = typeof body.stage === 'string' ? body.stage : '';
         // 选中文字是代码：首轮走「注释」提示词（代码块 + 逐句注释 + 小结）
         const isCode = body.kind === 'code';
-        const basePrompt = isChat
-            ? CHAT_SYSTEM_PROMPT
-            : stage === 'detail'
-                ? DETAIL_PROMPT
-                : isCode
-                    ? CODE_PROMPT
-                    : TRANSLATION_PROMPT;
+        // 截图提问是**独立模式**：没有选中文字，只有一张图 + 一个问题，不能用"追问"那套提示词
+        const isShot = body.kind === 'shot';
+        const basePrompt = isShot
+            ? shotPromptFor(shotScope, shotFrame)
+            : isChat
+                ? CHAT_SYSTEM_PROMPT
+                : stage === 'detail'
+                    ? DETAIL_PROMPT
+                    : isCode
+                        ? CODE_PROMPT
+                        : TRANSLATION_PROMPT;
         // 一个工具都没有时明确禁止伪造调用（模型会照着自己的习惯吐 <ds_safety_tool_call> 之类的格式）
         // 正文里不许写"盘算过程"（模型在工具轮里很容易把"I should search again…"这类写进正文）
         const narrateRule = '\n\n【正文只写结论】不要把思考、盘算、自查过程写进正文（"我还得再搜一次""先看看拿到的标题""让我确认一下"这类一律不写）；需要工具就直接调用，调用前后都不要解说。';
@@ -347,11 +403,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
         // 所以改成**先把可读范围说全**（项目 + 本次附带的文件），再说"除此之外会被拒"。
         let attachedPaths = collectAttachedPaths();
         const fileRule = wantsFsTools
-            ? attachedPaths.size > 0
-                ? `\n\n【文件访问】你可以用 read/grep/glob 读文件。**可读范围 = 当前项目（${readRoot || '不限'}）加上用户这次附带的文件**（它们的只读副本在附件仓里，路径就在消息里给出）。除此之外的路径会被拒绝。` +
-                    `\n  · 附带的文件**直接 read 那个路径**就行 —— 不要还没试就断定"项目外读不了"，也不要反复换 grep/glob 重试；` +
-                    `\n  · 引用文件内容时只取必要的一两行，不要整段粘贴。`
-                : `\n\n【文件访问】你可以用 read/grep/glob 查看当前项目里的文件（范围：${readRoot || '不限'}）。超出该范围的路径会被拒绝；引用文件内容时只取必要的一两行，不要整段粘贴。`
+            ? `\n\n【文件访问】你可以用read/grep/glob读取当前项目（${readRoot || '不限'}）和本小窗已提供的文件附件。附件路径以经过验证的文件引用或selection_material返回为准。需要旧文件时先用selection_material取回路径，再按问题读取必要部分；上传成功不等于格式可解析，不能读取时明确说明。除此之外的路径会被拒绝。`
             : '';
         const fetchRule = fetchBlockedSample
             ? `\n\n【抓取网页不可用】本机域名解析到保留地址段（${fetchBlockedSample}），网页抓取会被安全策略拒绝：只做联网搜索，**不要尝试抓取具体网址**（省掉一次注定失败的往返）。`
@@ -433,9 +485,55 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                 : {}),
         });
         let chars = 0;
+        let contextEstimate = 0;
+        if (isChat && capacity) {
+            try {
+                const fixedTokens = 256 + estimateTextTokens(systemPrompt + userText + initialExplanation + question)
+                    + estimateTextTokens(JSON.stringify(toolSchemas))
+                    + estimateTextTokens(materialIndex(catalog, historyStart + rawHistory.length))
+                    + estimateTextTokens(typeof body.toolDigest === 'string' ? body.toolDigest.slice(0, 2000) : '')
+                    + currentAttachments.reduce((n, a) => n + capacity.attachmentPrice(a), 0);
+                const planned = await planChatContext(ctx, route, capacity, { state: contextState, history: rawHistory, fixedTokens, signal: controller.signal, effort: sentEffort });
+                contextState = planned.state;
+                selectedHistory = planned.history;
+                contextEstimate = planned.estimatedTokens;
+                sse(res, { type: 'context-state', state: contextState,
+                    budget: capacity.budget, estimatedTokens: contextEstimate, modelWindow: capacity.modelWindow,
+                    compacted: planned.compacted, retainedMessages: selectedHistory.length });
+            }
+            catch (error) {
+                sse(res, { type: 'error', code: 'context-rejected', message: String(error?.message || error) });
+                finished = true;
+                clearTimeout(timer);
+                res.end();
+                return;
+            }
+        }
+        const memoryMessages = isChat ? [
+            ...(initialExplanation ? [createAssistantMessage({ source: historySource, content: [{ type: 'text', text: initialExplanation }] })] : []),
+            ...(contextState.summary ? [createUserMessage({ source: { kind: 'user' }, content: [{ type: 'text',
+                            text: '【较早对话摘要，可能遗漏细节】\n' + contextState.summary + '\n' + materialIndex(catalog, contextState.through) }] })] : []),
+        ] : [];
+        let upstreamStarted = false;
+        function checkedMessages(messages) {
+            if (capacity && estimateRequestTokens(messages, capacity, toolSchemas) > capacity.budget) {
+                throw Object.assign(new Error('本轮材料与工具结果超过上下文容量，请缩小问题或引用范围'), { code: 'context-capacity' });
+            }
+            return messages;
+        }
+        // ⚠️ 截图模式：图必须**在这一条消息里**。
+        // 原先只有 isChat(追问) 才拼 openingMessages，截图走的是 baseMessage 纯文本那条 ——
+        // 结果图根本没进请求，模型只会说"我没收到截图"（实测踩到）。
+        const shotImageBlocks = isShotMode
+            ? currentAttachments.map((item) => item.kind === 'image'
+                ? { type: 'image', attachment: item.ref }
+                : { type: 'file', attachment: item.ref })
+            : [];
         const baseMessage = createUserMessage({
             source: { kind: 'user' },
-            content: [{ type: 'text', text: userText }],
+            content: shotImageBlocks.length > 0
+                ? [...shotImageBlocks, { type: 'text', text: userText }]
+                : [{ type: 'text', text: userText }],
         });
         // 上一轮查过的东西一并带上：否则追问会反复重查同样的内容（一次工具往返 20~34s）
         const toolDigest = typeof body.toolDigest === 'string' ? body.toolDigest.trim().slice(0, 2000) : '';
@@ -450,13 +548,27 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
         const currentBlocks = currentAttachments.map((item) => item.kind === 'image' ? { type: 'image', attachment: item.ref } : { type: 'file', attachment: item.ref });
         const questionContent = currentBlocks.length > 0 ? [...currentBlocks, { type: 'text', text: questionText }] : [{ type: 'text', text: questionText }];
         let openingMessages = isChat
-            ? [...buildHistoryMessages(), createUserMessage({ source: { kind: 'user' }, content: questionContent })]
+            ? [
+                ...memoryMessages,
+                ...buildHistoryMessages(),
+                // 截图首轮：图 + 问题已经放进 baseMessage（只需要一份，避免图片和问题各发两遍）；
+                // 追问不走截图模式，当前问题仍然放在这里。
+                ...(isShot ? [] : [createUserMessage({ source: { kind: 'user' }, content: questionContent })]),
+            ]
             : [];
+        if (titleOnly) {
+            // 截图：base 那条消息里已经带了图 + 提问，这里只多一句"只输出标题"
+            openingMessages = [
+                createUserMessage({ source: { kind: 'user' }, content: [{ type: 'text', text: shotTitlePromptFor(shotScope, shotFrame) }] }),
+            ];
+        }
         try {
             let errored = false;
             let usage = null;
             let answer = '';
             const toolMessages = [];
+            const recoveredMaterialBlocks = [];
+            let toolContextTokens = 0;
             let toolCalls = 0;
             // 允许的工具轮 + 1 个**收尾轮**：收尾轮不带任何工具。
             // 不加这一轮的话，模型每轮都调工具时循环直接结束，正文是空的——
@@ -495,9 +607,10 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
             const triedEfforts = new Set(sentEffort ? [sentEffort] : []);
             const firstSentEffort = sentEffort;
             let roundEffort = sentEffort;
-            for (let round = 0; round <= toolRounds; round += 1) {
+            const effectiveToolRounds = titleOnly ? 0 : toolRounds;
+            for (let round = 0; round <= effectiveToolRounds; round += 1) {
                 const roundStartedAt = Date.now();
-                const wrapUp = round === toolRounds;
+                const wrapUp = round === effectiveToolRounds;
                 const stream = ctx.llm.stream({
                     provider: route.provider,
                     model: route.model,
@@ -505,7 +618,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                     // 插件以前没传，结果是所有小窗请求共用静态会话头 —— 而中继是按会话跟踪的（实测长回答被掐）。
                     ...(sessionId ? { sessionId: sessionId } : {}),
                     system: undefined,
-                    messages: [
+                    messages: checkedMessages([
                         // 第二参数（组装这段提示词的插件名）：**新老版本签名不一样，两边都得能编译**——
                         //   · npm 上按 peer 范围 `>=0.1.5-rc <2` 解析到的是 0.1.5-rc.3：`(text, plugin)` 两个必填
                         //     （CI 就是拿这份类型做构建校验的，少传就 TS2554）；
@@ -516,7 +629,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                                 '正文里不要出现任何工具调用格式（包括 DeepSeek 的 DSML 写法，如 invoke/parameter 标记）——那些会被当作乱码展示给用户。'
                             : systemPrompt, name),
                         ...(round === 0 ? [baseMessage, ...openingMessages] : [baseMessage, ...openingMessages, ...toolMessages]),
-                    ],
+                    ]),
                     // 工具**每轮都在请求里**（照主会话的做法）：收尾轮以前故意不带工具"逼结论"，
                     // 结果模型退到文本层、把工具调用标记写进正文（实测 wrapUp 轮 raw=726 → emitted=0）。
                     // 主会话从不摘工具，模型也就从不写这种标记 —— 这里与它保持一致。
@@ -526,6 +639,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                     ...(roundEffort ? { reasoningEffort: ReasoningEffortId(roundEffort) } : {}),
                     signal: controller.signal,
                 });
+                upstreamStarted = true;
                 const assembler = new BlockAssembler();
                 // 本轮吐出的正文字数：这一轮如果最后是要调工具，那这些字其实是"过程旁白"
                 // （"我还得再搜一次""先看看拿到的标题"这类），要被撤回，不能当成答案留在小窗里
@@ -611,7 +725,11 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                     if (roundText.length === 0 && historyHadAttachments && !historyDropped && looksLikeAttachmentFailure(errorMessage, errorCode)) {
                         historyDropped = true;
                         historyImages = { kept: 0, omitted: 0, files: 0 };
-                        openingMessages = [...buildHistoryMessages(), createUserMessage({ source: { kind: 'user' }, content: questionContent })];
+                        openingMessages = [
+                            ...memoryMessages,
+                            ...buildHistoryMessages(),
+                            ...(isShot ? [] : [createUserMessage({ source: { kind: 'user' }, content: questionContent })]),
+                        ];
                         // 历史附件被丢掉了 → 名单跟着缩小（只放"还看得见的"那些）
                         attachedPaths = collectAttachedPaths();
                         errored = false;
@@ -713,6 +831,8 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                     content: blocks,
                     source: { provider: route.provider, model: route.model },
                 }));
+                const restoredBlocks = [];
+                toolContextTokens += estimateTextTokens(JSON.stringify(blocks));
                 for (const call of calls) {
                     if (call.type !== 'tool-call')
                         continue;
@@ -732,7 +852,15 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                     // 只认 path 的话 read 的 file_path 会是 undefined → 边界检查被静默绕过（实测踩过）。
                     const rawPath = parsed?.file_path ?? parsed?.path;
                     const wantPath = typeof rawPath === 'string' ? rawPath : '';
-                    const execution = !toolSchemas.some((schema) => schema.name === call.name)
+                    let restored;
+                    if (call.name === MATERIAL_TOOL.name && toolSchemas.some(schema => schema.name === call.name)) {
+                        restored = await restoreMaterial(ctx, catalog, parsed, { through: contextState.through, key: typeof body.key === 'string' ? body.key : '',
+                            maxChars: config.toolResultMaxChars, readMessage: historyStore.readMessage });
+                        if (restored.restored && capacity && capacity.attachmentPrice(restored.restored) + contextEstimate + toolContextTokens > capacity.budget * 0.95) {
+                            restored = { content: [{ type: 'text', text: '这份材料超过本轮剩余容量，请减少本轮材料后重新查看。' }], isError: true };
+                        }
+                    }
+                    const execution = restored || (!toolSchemas.some((schema) => schema.name === call.name)
                         ? { content: [{ type: 'text', text: `（调用已拒绝：${call.name} 不在本次允许的工具中）` }], isError: true }
                         : FS_TOOL_NAMES.includes(call.name) && !isReadAllowed(readRoot, wantPath, attachedPaths)
                             ? {
@@ -744,7 +872,21 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                                 ],
                                 isError: true,
                             }
-                            : await runTool(ctx, call.name, parsed, controller.signal, config.toolResultMaxChars, agent);
+                            : await runTool(ctx, call.name, parsed, controller.signal, config.toolResultMaxChars, agent));
+                    toolContextTokens += estimateTextTokens(JSON.stringify(execution.content));
+                    if (restored?.restored) {
+                        const item = restored.restored;
+                        restoredBlocks.push(item.kind === 'image' ? { type: 'image', attachment: item.ref } : { type: 'file', attachment: item.ref });
+                        if (capacity)
+                            toolContextTokens += capacity.attachmentPrice(item);
+                        if (item.kind === 'file') {
+                            const store = ctx.get('attachments');
+                            const path = store?.fileHostPath?.(item.ref);
+                            if (path)
+                                attachedPaths.add(resolve(path));
+                        }
+                        sse(res, { type: 'material-restored', attachment: item });
+                    }
                     // 结果也回传：小窗里能看见"查了什么、命中什么、链接是什么"，否则工具是个黑盒
                     const digest = digestToolResult(execution.content);
                     sse(res, {
@@ -764,6 +906,9 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                     // 结论轮要用：把"查了什么 + 拿到了什么"写成纯文本，避免重放工具调用链
                     toolDigestLines.push(`· ${call.name}（${label2 || '查询'}）：${digest.preview}`);
                 }
+                recoveredMaterialBlocks.push(...restoredBlocks);
+                if (restoredBlocks.length > 0)
+                    toolMessages.push(createUserMessage({ source: { kind: 'user' }, content: restoredBlocks }));
             }
             if (!errored && answer.trim().length === 0 && toolCalls > 0 && triedEfforts.size === 1) {
                 // 收尾轮常常"只吐一段工具调用、不写正文"（实测「今天的新闻」一个字都没有）。
@@ -772,6 +917,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                 conclusionInfo.tried = true;
                 sse(res, { type: 'notice', code: 'conclusion-retry', text: '查询已完成，正在整理结论…' });
                 const conclusionUser = [
+                    '【补充结论】',
                     // ⚠️ **必须带上用户这一轮问的是什么**：这一段是最终正文的来源，而它以前只有
                     // 「选区块 + 检索资料 + 请给出最终答案」—— 模型看不到问题，就只能顺着选中文字写。
                     // 实测症状：问"搜索今天的AI新闻。"，正文却在解释选区那个词（如 diagLog / exit）。
@@ -779,9 +925,6 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                     // 缺了问题就等于每次都答非所问。
                     question ? `【用户这一轮的问题】（这就是你要回答的对象）\n${question}` : '',
                     '',
-                    '【选中文字】（只是背景资料：问题需要它时才引用，不要拿它当回答的对象）',
-                    userText,
-                    '',
                     '【已经查到的资料】',
                     ...toolDigestLines.slice(0, 12),
                     '',
@@ -810,11 +953,11 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                             // 以前这一轮不带工具，模型要调工具时只能退回文本层 —— 那正是正文里出现
                             // DSML / ds_safety_tool_call 标记的唯一来源。带上之后它走协议层，正文就是正文。
                             ...(toolSchemas.length > 0 ? { tools: toolSchemas } : {}),
-                            messages: [
-                                // 同上：多传的插件名在老版本运行时会被忽略，新版本类型上却是必填
+                            messages: checkedMessages([
                                 systemMessageOf(systemPrompt, name),
-                                createUserMessage({ source: { kind: 'user' }, content: [{ type: 'text', text: conclusionUser }] }),
-                            ],
+                                baseMessage, ...openingMessages,
+                                createUserMessage({ source: { kind: 'user' }, content: [...recoveredMaterialBlocks, { type: 'text', text: conclusionUser }] }),
+                            ]),
                             ...(config.temperature >= 0 ? { temperature: config.temperature } : {}),
                             ...(config.maxTokens > 0 ? { maxTokens: config.maxTokens } : {}),
                             ...(attemptEffort ? { reasoningEffort: ReasoningEffortId(attemptEffort) } : {}),
@@ -843,9 +986,15 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                                 }
                             }
                         }
-                        for (const tail of [thinkAgain.flush()])
-                            if (tail.think)
-                                sse(res, { type: 'thought', text: tail.think });
+                        const tail = thinkAgain.flush();
+                        if (tail.think)
+                            sse(res, { type: 'thought', text: tail.think });
+                        if (tail.text) {
+                            chars += tail.text.length;
+                            answer += tail.text;
+                            conclusionInfo.chars += tail.text.length;
+                            sse(res, { type: 'delta', text: tail.text });
+                        }
                         // 结论轮也可能把"英文核对过程"写进正文：一律按泄漏裁一次
                         const leaked = splitLeakedReasoning(answer);
                         if (leaked.prefix) {
@@ -856,6 +1005,8 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
                         }
                     }
                     catch (error) {
+                        if (error?.code === 'context-capacity')
+                            throw error;
                         conclusionInfo.error = String(error?.message ?? error);
                         ctx.logger?.warn?.(`[${name}] 结论轮失败：${conclusionInfo.error}`);
                     }
@@ -896,6 +1047,7 @@ export function createAnalyze(ctx, config, models, sessions, historyStore) {
             const aborted = err?.name === 'AbortError';
             sse(res, {
                 type: 'error',
+                ...(!upstreamStarted && err?.code === 'context-capacity' ? { code: 'context-rejected' } : {}),
                 message: aborted ? `调用超时或已取消（${config.timeoutMs}ms）` : String(err?.message ?? err),
             });
         }