mocode-ai 1.7.6 → 1.7.7

package/dist/context/vision-window.js

@@ -22,6 +22,13 @@ const ATTACHMENT_TAIL = '. Analyze the attached image content directly.';
 const UNKNOWN_FRAME_NAME = 'frame';
 /** 恢复提示只写进**第一条**被剪占位,避免 N 条占位重复同一句话侵蚀 token。 */
 const RECOVERY_HINT = ' Call computer{screenshot} if you need the current screen.';
+/**
+ * 帧被剪枝后占位行的固定前缀(见 pruneFrameMessage,完整行为 `[screenshot omitted: <帧名> · step N]`)。
+ * 剪枝会把合成附件消息整个 content 换成占位(前言 ATTACHMENT_PREAMBLE 随之消失),
+ * rollback 对齐真实用户轮时必须认出这个占位、不把它算成一轮——那里因循环依赖无法
+ * import 本模块,故在 rollback/store.ts 内联了同一行格式(按整行正则判定),修改此串时两处必须同步。
+ */
+const PRUNED_FRAME_PREFIX = '[screenshot omitted: ';
 /**
  * 白名单:参与滑动窗口的屏幕帧名前缀。
  *
@@ -139,11 +146,11 @@ function pruneFrameMessage(message, names, step, withHint) {
     for (const part of parts) {
         if (part.type !== 'image_url')
             continue;
-        lines.push(`[screenshot omitted: ${names[index] ?? UNKNOWN_FRAME_NAME} · step ${step}]`);
+        lines.push(`${PRUNED_FRAME_PREFIX}${names[index] ?? UNKNOWN_FRAME_NAME} · step ${step}]`);
         index++;
     }
     if (lines.length === 0)
-        lines.push(`[screenshot omitted: ${names[0] ?? UNKNOWN_FRAME_NAME} · step ${step}]`);
+        lines.push(`${PRUNED_FRAME_PREFIX}${names[0] ?? UNKNOWN_FRAME_NAME} · step ${step}]`);
     if (withHint)
         lines[0] = `${lines[0]}${RECOVERY_HINT}`;
     const content = [{ type: 'text', text: lines.join('\n') }];

package/dist/repl/runtime.js

@@ -352,12 +352,6 @@ export async function startRepl(initialHistory, sessionId, sandboxRootOverride,
             layout.contentWrite(`${ui.dim}(没有可回滚的轮次)${ui.reset}\n`);
             return;
         }
-        // 各轮 user 全文(预填用;按 user 消息顺序与 turnList 对齐)
-        const userTexts = [];
-        for (const m of history) {
-            if (m.role === 'user')
-                userTexts.push(textOf(m.content));
-        }
         const items = turnList.map((t) => ({ firstLine: t.firstLine }));
         let picked;
         try {
@@ -368,8 +362,11 @@ export async function startRepl(initialHistory, sessionId, sandboxRootOverride,
         }
         if (picked === null)
             return; // Esc 取消
-        // picked(0-based)= 第 (picked+1) 轮:删该轮及之后(planRollback(picked) 保 1..picked),预填该轮 user 输入
-        const prefillText = userTexts[picked] ?? '';
+        // picked(0-based)= 第 (picked+1) 轮:删该轮及之后(planRollback(picked) 保前 picked 轮),
+        // 预填该轮 user 输入。必须经 store 按"尾部对齐的真实用户轮消息"定位——直接按 history 里
+        // role==='user' 数下标会把工具附件的合成 user 消息/被剪枝老轮算进去,预填错轮。
+        const pickedMessage = runtime.rollback.turnUserMessage(picked, history);
+        const prefillText = pickedMessage ? textOf(pickedMessage.content) : '';
         const plan = runtime.rollback.plan(picked, history);
         // 清屏(擦选轮菜单 + /rollback 回显)+ 复位 lastView(dim 空),给文件询问一个干净、resize 安全的画面
         layout.clearContent();

package/dist/rollback/store.js

@@ -47,6 +47,61 @@ const EXCLUDED_WORKSPACE_DIRS = Object.freeze([
     '.ruff_cache',
     '.gradle',
 ]);
+/**
+ * 工具经 modelAttachments 注入的合成 user 消息前言(见 agent/tool-turn.ts 的
+ * ATTACHMENT_PREAMBLE)。这类消息 role 也是 'user',但不对应 beginTurn 的真实用户轮,
+ * 回滚按 role==='user' 数轮次会与 turns 数组错位,导致"回退一条却连带删掉好几轮"。
+ * 两处常量必须保持一致;不 import tool-turn 是为避免 rollback→agent 的层级反向依赖。
+ */
+const TOOL_ATTACHMENT_PREAMBLE = 'The read_file tool loaded the following visual input: ';
+/**
+ * 视觉滑动窗口剪枝占位的机器固定格式(必须与 context/vision-window.ts 的
+ * pruneFrameMessage 输出逐字一致):一行或多行 `[screenshot omitted: <帧名> · step <数字>]`,
+ * 首行可能再跟固定恢复提示后缀。用整串正则而不是前缀匹配,避免把恰好以
+ * "[screenshot omitted: " 开头的真实用户输入误判成占位。
+ *
+ * 这里内联而不是 import vision-window:vision-window → llm → … → rollback/index 会形成
+ * 循环依赖,在共享进程的测试里触发 "Cannot access 'defaultRollbackStore' before initialization"。
+ */
+const PRUNED_FRAME_LINE_RE = /^\[screenshot omitted: .+? · step \d+\]$/;
+const PRUNED_FRAME_HINT_SUFFIX = ' Call computer{screenshot} if you need the current screen.';
+function isPrunedFrameText(text) {
+    if (typeof text !== 'string')
+        return false;
+    const lines = text.split('\n');
+    return lines.every((line, i) => {
+        const core = i === 0 && line.endsWith(PRUNED_FRAME_HINT_SUFFIX) ? line.slice(0, -PRUNED_FRAME_HINT_SUFFIX.length) : line;
+        return PRUNED_FRAME_LINE_RE.test(core);
+    });
+}
+/**
+ * 判断一条 user 消息是否对应一个真实用户轮(= beginTurn 记进 turns 的那一条)。
+ *
+ * 排除两种合成 user 消息(都是 role:'user' 但不计轮次):
+ *  1. 工具批次提交时由 modelAttachments 注入的附件消息:ContentPart[] 的 text part
+ *     以工具附件前言开头;
+ *  2. 上述附件消息经视觉滑动窗口剪枝后的占位:整个 content 被替换成
+ *     "[screenshot omitted: …]" 文本,前言已不存在(见 vision-window.pruneFrameMessage),
+ *     形态可能是数组 text part,也可能是纯字符串。
+ *
+ * 用户自己发的多模态消息没有这两个标记,不会被误判。
+ */
+function isRealTurnUserMessage(message) {
+    if (message.role !== 'user')
+        return false;
+    const content = message.content;
+    if (typeof content === 'string')
+        return !isPrunedFrameText(content);
+    if (!Array.isArray(content))
+        return false;
+    for (const part of content) {
+        if (part && typeof part === 'object' && part.type === 'text' && typeof part.text === 'string') {
+            if (part.text.startsWith(TOOL_ATTACHMENT_PREAMBLE) || isPrunedFrameText(part.text))
+                return false;
+        }
+    }
+    return true;
+}
 const CAPTURE_FILE_LIMIT = 1024 * 1024;
 const CAPTURE_TOTAL_LIMIT = 32 * 1024 * 1024;
 const CAPTURE_ENTRY_LIMIT = 20000;
@@ -440,6 +495,32 @@ export class RollbackStore {
     listTurns() {
         return this.turns.slice();
     }
+    /**
+     * 取菜单第 pickedIndex(0-based)轮对应的真实用户消息(用于 /rollback 后预填)。
+     * 与 findCutoffIndex 同一套尾对齐规则。无对应消息(续接历史缺轮)时返回 undefined。
+     */
+    turnUserMessage(pickedIndex, history) {
+        const index = this.findTurnUserIndex(pickedIndex, history);
+        return index < 0 ? undefined : history[index];
+    }
+    /**
+     * 菜单第 pickedIndex(0-based)轮对应的真实用户消息在 history 中的下标。
+     * turns 是真实轮次的尾部连续子集(滚动剪枝/压缩只丢旧轮),故菜单第 k 轮 =
+     * history 中倒数第 (turns.length-k) 条真实用户轮消息。找不到返回 -1。
+     */
+    findTurnUserIndex(pickedIndex, history) {
+        let wantReverseRank = this.turns.length - pickedIndex;
+        if (wantReverseRank <= 0)
+            return -1;
+        for (let i = history.length - 1; i >= 0; i--) {
+            if (!isRealTurnUserMessage(history[i]))
+                continue;
+            wantReverseRank -= 1;
+            if (wantReverseRank === 0)
+                return i;
+        }
+        return -1;
+    }
     /**
      * 只读侧门(供 /diff 渲染):返回 turnId > sinceTurnId 的全部快照,浅拷贝数组。
      * 快照内容含 before 原文,调用方不得 mutate 返回值;用于展示,不影响回滚语义。
@@ -447,19 +528,21 @@ export class RollbackStore {
     snapshotsSince(sinceTurnId) {
         return this.snapshots.filter((snapshot) => snapshot.turnId > sinceTurnId);
     }
+    /**
+     * 回滚切点:保留菜单前 n 轮,从第 n+1 轮(菜单下标 n,0-based)的真实 user 消息处切断。
+     *
+     * turns(菜单)= 当前仍可回滚的轮次,是真实轮次的**尾部连续子集**(滚动剪枝/压缩只丢旧轮),
+     * 故从尾部倒数定位,而不是正数第 n 条——压缩的「user 逐字通道」会把更老的 user 原文
+     * 捞回保留区,正数下标与 turns 不再连续对齐。找不到(无第 n+1 轮)时返回 history.length。
+     */
     findCutoffIndex(n, history) {
-        let seen = 0;
-        for (let i = 0; i < history.length; i++) {
-            if (history[i].role === 'user') {
-                seen += 1;
-                if (seen === n + 1)
-                    return i;
-            }
-        }
-        return history.length;
+        const index = this.findTurnUserIndex(n, history);
+        return index < 0 ? history.length : index;
     }
     planRollback(n, history) {
         const cutoffTurnId = this.turns[n - 1]?.turnId ?? 0;
+        // turns 与 history 真实 user 消息尾对齐后再取切点:滚动剪枝(>50 轮)/压缩/续接后
+        // turns 只是尾部子集,不能再把 n 当成全局第 n 条 user 消息,否则会多删 offset 轮。
         const cutoffIndex = this.findCutoffIndex(n, history);
         const order = [];
         const map = new Map();
@@ -587,7 +670,9 @@ export class RollbackStore {
         return { deletedMsgs, revertedFiles, conflictedFiles };
     }
     pruneAfterCompaction(history) {
-        const count = history.filter((message) => message.role === 'user').length;
+        // 只数真实用户轮(排除工具附件合成 user 消息);压缩会整体丢弃旧轮,
+        // 保留下来的总是最近的轮次,故 turns 同样从尾部对齐截取。
+        const count = history.filter((message) => isRealTurnUserMessage(message)).length;
         this.turns = count >= this.turns.length ? this.turns : this.turns.slice(-count);
         const alive = new Set(this.turns.map((turn) => turn.turnId));
         this.snapshots = this.snapshots.filter((snapshot) => alive.has(snapshot.turnId));
@@ -603,7 +688,9 @@ export class RollbackStore {
     rebuildFromHistory(history) {
         const rebuilt = [];
         for (const message of history) {
-            if (message.role !== 'user')
+            // 跳过工具附件注入的合成 user 消息,否则 turns 比真实轮次多,
+            // /rollback 菜单与 history 切点(planRollback 按 turns[n-1])整体错位。
+            if (!isRealTurnUserMessage(message))
                 continue;
             const first = toText(message.content).split('\n')[0] ?? '';
             rebuilt.push({ turnId: rebuilt.length + 1, firstLine: truncateDisplay(first, 40) });

package/dist/runtime/runtime.js

@@ -90,6 +90,10 @@ export class Runtime {
                 this.assertOpen();
                 return this.context.rollbackStore.listTurns();
             },
+            turnUserMessage: (pickedIndex, history) => {
+                this.assertOpen();
+                return this.context.rollbackStore.turnUserMessage(pickedIndex, history);
+            },
             plan: (n, history) => {
                 this.assertIdle('plan rollback');
                 const plan = this.context.rollbackStore.planRollback(n, history);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mocode-ai",
-  "version": "1.7.6",
+  "version": "1.7.7",
   "description": "终端编码 agent:LLM + tool-call 循环 + 流式输出(含思考)+ 27 个工具,接任意 OpenAI 兼容后端。",
   "type": "module",
   "bin": {