mocode-ai 0.5.4 → 0.5.5

package/dist/context/utils.js

@@ -0,0 +1,68 @@
+// 跨上下文模块共享的小工具函数。
+//
+// 目的:消除 `relevance.ts` / `lifecycle.ts` / `budget.ts` / `compact.ts` / `session/drop.ts`
+//      之间的 copy-paste。这些函数逻辑完全一致,统一维护于此,避免行为漂移。
+//
+// 不变量:
+//  - 永不抛错(对齐「调度器永不抛错」契约)。
+//  - 仅依赖 `ChatMessage` 的最小形状,不反向 import agent / session / tools。
+/** 把消息 content 拍平成字符串(OpenAI 可能 string / null / 多模态数组)。
+ *  用于估算 token、内容匹配、stub 拼接等场景。 */
+export function toText(content) {
+    if (content == null)
+        return '';
+    if (typeof content === 'string')
+        return content;
+    try {
+        return JSON.stringify(content);
+    }
+    catch {
+        return String(content);
+    }
+}
+/** 从 history 末尾向前找最后一个 user 消息的索引;无 user 返 -1。
+ *  用于划定「当前轮保护区」——该 user 及其之后的消息(agent 本轮还在用)不动。
+ *  注:history[0] 是 system,故下界为 1。 */
+export function lastUserIndex(history) {
+    for (let i = history.length - 1; i >= 1; i--) {
+        if (history[i].role === 'user')
+            return i;
+    }
+    return -1;
+}
+/** 取 tool 消息对应的工具名(从紧邻的前导 assistant.tool_calls 按 tool_call_id 配对找)。
+ *  返回 null 表示找不到(孤儿 tool 消息,极少见),调用方保守跳过。 */
+export function toolNameOf(history, idx) {
+    const tcId = history[idx].tool_call_id;
+    if (!tcId)
+        return null;
+    for (let j = idx - 1; j >= 1; j--) {
+        const m = history[j];
+        if (m.role !== 'assistant')
+            continue;
+        const tcs = m.tool_calls;
+        if (!tcs)
+            continue;
+        const hit = tcs.find((tc) => tc?.id === tcId);
+        if (hit)
+            return hit.function?.name ?? null;
+    }
+    return null;
+}
+/** 解析工具 arguments JSON,只取 `path` 字段;非法 / 无 path 返 null。
+ *  用于 read_file / edit_file / write_file 等以 path 为关键字的工具,供跨消息关联使用。 */
+export function extractPath(argsRaw) {
+    if (!argsRaw)
+        return null;
+    let parsed;
+    try {
+        parsed = JSON.parse(argsRaw);
+    }
+    catch {
+        return null;
+    }
+    if (!parsed || typeof parsed !== 'object')
+        return null;
+    const p = parsed.path;
+    return typeof p === 'string' && p ? p : null;
+}

package/dist/repl/index.js

@@ -455,14 +455,15 @@ function textOf(c) {
  * 回放默认全折叠;用户可鼠标点击摘要行展开(由 BatchRenderer 接管,见 ui/batch.ts)。
  */
 export function renderHistory(history) {
-    const idToName = new Map();
-    // 当前累积的 batch(assistant.tool_calls + 后续 tool 消息);一旦遇到非 tool 消息即收尾出摘要
-    let pendingBatch = [];
+    const idToEntry = new Map();
+    // 普通工具可跨 assistant 步聚合；mutation 各自占一个 group，并切断前后普通工具。
+    let pendingBatches = [];
+    let normalBatch = null;
     const flushBatch = () => {
-        if (pendingBatch.length === 0)
-            return;
-        batch.writeSummaryOnly(pendingBatch, layout);
-        pendingBatch = [];
+        for (const entries of pendingBatches)
+            batch.writeSummaryOnly(entries, layout);
+        pendingBatches = [];
+        normalBatch = null;
     };
     for (let idx = 0; idx < history.length; idx++) {
         const m = history[idx];
@@ -505,14 +506,25 @@ export function renderHistory(history) {
                 for (const tc of tcs) {
                     const name = tc?.function?.name ?? '';
                     const args = tc?.function?.arguments ?? '';
-                    if (tc?.id && name)
-                        idToName.set(tc.id, name);
-                    pendingBatch.push({
+                    const entry = {
                         name,
                         callSummary: summarizeToolCall(name, args),
                         resultSummary: '',
                         diffBlock: null,
-                    });
+                    };
+                    if (batch.isMutationToolName(name)) {
+                        normalBatch = null;
+                        pendingBatches.push([entry]);
+                    }
+                    else {
+                        if (!normalBatch) {
+                            normalBatch = [];
+                            pendingBatches.push(normalBatch);
+                        }
+                        normalBatch.push(entry);
+                    }
+                    if (tc?.id)
+                        idToEntry.set(tc.id, entry);
                 }
                 continue; // 跳过后续 tool 消息处理循环(由下一分支填 result)
             }
@@ -522,18 +534,10 @@ export function renderHistory(history) {
         }
         if (m.role === 'tool') {
             const id = m.tool_call_id ?? '';
-            const name = idToName.get(id) ?? '';
+            const target = idToEntry.get(id);
+            const name = target?.name ?? '';
             const output = textOf(m.content);
             const preview = summarizeToolResult(name, output);
-            // 匹配 pendingBatch 中尚未填 result 的同名 entry;同名前缀 tool 较罕见(并行工具同 id 不同名)
-            let target;
-            for (let i = pendingBatch.length - 1; i >= 0; i--) {
-                const e = pendingBatch[i];
-                if (e.name === name && !e.resultSummary) {
-                    target = e;
-                    break;
-                }
-            }
             if (target) {
                 target.resultSummary = preview;
                 target.fullOutput = output;
@@ -853,6 +857,7 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null,
         if (!loadSnapshots(loaded.id))
             rebuildFromHistory(history);
         contextState.lastUsage = undefined;
+        contextState.correction = 1;
         lastTurnUsage = undefined; // 续接:旧会话的 token 累计已无意义,清空等下轮覆写
         layout.clearContent();
         renderHistory(history);
@@ -985,6 +990,7 @@ export async function startRepl(initialHistory, sessionId, updateNotice = null,
             currentSessionId = undefined; // 下轮起新会话文件
             turnCount = 0; // 反思 cadence 重新计数
             contextState.lastUsage = undefined;
+            contextState.correction = 1;
             lastTurnUsage = undefined; // 清空旧轮的 token 累计
             pendingAttachments = []; // 一并清空待发图片
             layout.clearContent();

package/dist/rollback/index.js

@@ -2,24 +2,12 @@ import { existsSync, mkdirSync, readFileSync, unlinkSync, writeFileSync, } from
 import path from 'node:path';
 import { config } from '../config/index.js';
 import { truncateDisplay } from '../ui/render.js';
+import { toText } from '../context/utils.js';
 let turnIdCounter = 0;
 let currentTurnId = 0;
 let turns = [];
 let snapshots = [];
 const MUTATION_TOOLS = new Set(['write_file', 'edit_file']);
-/** 把任意 content 拍平成字符串(OpenAI 可能是 string / 多模态数组)。 */
-function toText(content) {
-    if (content == null)
-        return '';
-    if (typeof content === 'string')
-        return content;
-    try {
-        return JSON.stringify(content);
-    }
-    catch {
-        return String(content);
-    }
-}
 /** 规整成 cwd 相对路径(快照存相对,跨 /resume 同项目可识别;resolve 已归一 ./ 和 ..)。 */
 function toRel(p) {
     try {

package/dist/session/compact.js

@@ -5,10 +5,14 @@ import { ui } from '../ui/theme.js';
 import { Spinner } from '../ui/spinner.js';
 import * as layout from '../ui/layout.js';
 import { pruneAfterCompaction } from '../rollback/index.js';
+import { toText } from '../context/utils.js';
 /** 跨模块共享的上下文状态:agent 写 lastUsage,compact 写 lastEstimate,repl 的 /context 读。
- *  scheduler.ts 写最近一次调度日志(可选,repl 可读不到时 no-op)。 */
+ *  scheduler.ts 写最近一次调度日志(可选,repl 可读不到时 no-op)。
+ *  correction:API 实测 token / 估算 token 的校正系数(1.0 = 无偏差;>1 = 低估;<1 = 高估)。
+ *  由 agent/core.ts 在每次 chat 响应后更新;compact/repl 在 usage 失效时同步重置。 */
 export const contextState = {
     lastEstimate: 0,
+    correction: 1,
 };
 /** 中截:text 太长时保 head + 标记 + tail,总长 ≤ max。 */
 export function truncateMid(text, max) {
@@ -97,19 +101,7 @@ export function capToolResultForHistory(name, output) {
         return output;
     return truncateMid(output, MAX_HISTORY_RESULT);
 }
-// ── 内部:消息内容拍平 / group 划分 ──────────────────────────────────────
-function toText(content) {
-    if (content == null)
-        return '';
-    if (typeof content === 'string')
-        return content;
-    try {
-        return JSON.stringify(content);
-    }
-    catch {
-        return String(content);
-    }
-}
+// ── 内部:group 划分(toText 已移至 context/utils.ts 统一维护) ────────────
 /**
  * 把多模态 content 拍平成纯文本(供摘要 transcript 用):text parts 拼接;image_url parts
  * 替换为 `[图片已剥离: <mime>]` stub,避免 base64 进摘要 prompt(LLM 看到也无意义,反而撑爆 token)。
@@ -323,6 +315,7 @@ export async function compactHistory(history, opts) {
             const estimateAfter = estimateMessagesTokens(history) + schemaTokens;
             contextState.lastEstimate = estimateAfter;
             contextState.lastUsage = undefined;
+            contextState.correction = 1;
             layout.contentWrite(`  ${ui.bold}${ui.accent}●${ui.reset} ${ui.accent}强制压缩(focus on early history)${ui.reset}  ${ui.dim}${estimateBefore} → ${estimateAfter} tokens${ui.reset}\n`);
             return {
                 compacted: true,
@@ -409,6 +402,7 @@ export async function compactHistory(history, opts) {
         const estimateAfter = estimateMessagesTokens(history) + schemaTokens;
         contextState.lastEstimate = estimateAfter;
         contextState.lastUsage = undefined; // 压缩后旧 usage 失效,/context 改用估算
+        contextState.correction = 1;
         layout.contentWrite(`  ${ui.bold}${ui.accent}●${ui.reset} ${ui.accent}压缩上下文${ui.reset}  ${ui.dim}${estimateBefore} → ${estimateAfter} tokens${ui.reset}\n`);
         // 抖动保护:压缩后仍超阈 → 提示 /clear,不死循环
         if (estimateAfter >= opts.threshold * opts.window) {
@@ -426,6 +420,7 @@ export async function compactHistory(history, opts) {
     const estimateAfter = estimateMessagesTokens(history) + schemaTokens;
     contextState.lastEstimate = estimateAfter;
     contextState.lastUsage = undefined; // 结构虽未变,但 token 数已变,旧 usage 失效
+    contextState.correction = 1;
     if (microcompactDone) {
         layout.contentWrite(`  ${ui.bold}${ui.accent}●${ui.reset} ${ui.accent}微压缩旧工具结果${ui.reset}  ${ui.dim}${estimateBefore} → ${estimateAfter} tokens${ui.reset}\n`);
         return {

package/dist/session/drop.js

@@ -11,49 +11,7 @@
 //  - 原地修改 history(同 compact:length=0;push 重建,repl 持有同一引用)。
 //  - 永不抛错(对齐「调度器永不抛错」契约);无匹配 / 无可剔除 → 返 dropped=0。
 import { messageTokens, estimateTokens, } from '../llm/index.js';
-/** 把消息 content 拍平成字符串(OpenAI 可能 string / null / 多模态数组)。 */
-function toText(content) {
-    if (content == null)
-        return '';
-    if (typeof content === 'string')
-        return content;
-    try {
-        return JSON.stringify(content);
-    }
-    catch {
-        return String(content);
-    }
-}
-/**
- * 从 history 末尾向前找最后一个 user 消息的索引;无 user 返 -1。
- * 用于划定"当前轮保护区"——该 user 及其之后的消息一律不剔除。
- */
-function lastUserIndex(history) {
-    for (let i = history.length - 1; i >= 1; i--) {
-        if (history[i].role === 'user')
-            return i;
-    }
-    return -1;
-}
-/** 取 tool 消息对应的工具名(从紧邻的前导 assistant.tool_calls 按 tool_call_id 配对找)。 */
-function toolNameOf(history, idx) {
-    const tcId = history[idx].tool_call_id;
-    if (!tcId)
-        return null;
-    for (let j = idx - 1; j >= 1; j--) {
-        const m = history[j];
-        if (m.role !== 'assistant')
-            continue;
-        const tcs = m
-            .tool_calls;
-        if (!tcs)
-            continue;
-        const hit = tcs.find((tc) => tc?.id === tcId);
-        if (hit)
-            return hit.function?.name ?? null;
-    }
-    return null;
-}
+import { lastUserIndex, toText, toolNameOf } from '../context/utils.js';
 /**
  * 剔除历史里命中的旧 tool 结果(原地修改 history)。
  *

package/dist/session/scheduler.js

@@ -45,7 +45,7 @@ export function createBudgetScheduler() {
             // 占位:push-time 三闸(cap / pruner / lifecycle)已自动跑;此接缝供将来演进。
         },
         async runStep(history, step) {
-            const report = evaluateBudget(history, config.contextWindowTokens, step);
+            const report = evaluateBudget(history, config.contextWindowTokens, step, contextState.correction);
             const actions = scheduleActions(report);
             let compactHistoryCalled = false;
             for (const a of actions) {
@@ -104,7 +104,7 @@ export async function manualCompact(history, focus, opts) {
         }));
         return {
             step: -1,
-            report: evaluateBudget(history, config.contextWindowTokens, -1),
+            report: evaluateBudget(history, config.contextWindowTokens, -1, contextState.correction),
             actions: [{ kind: 'compact_history', focus }],
             compactHistoryCalled: true,
             ts: Date.now(),
@@ -118,7 +118,7 @@ export async function manualCompact(history, focus, opts) {
             },
         };
     }
-    const report = evaluateBudget(history, config.contextWindowTokens, -1);
+    const report = evaluateBudget(history, config.contextWindowTokens, -1, contextState.correction);
     let actions = scheduleActions(report);
     // 用户显式说「要压」:即使 report 不含 history 触发,仍追加 compact_history
     const hasCompact = actions.some(a => a.kind === 'compact_history');

package/dist/tools/builtins/todolist.js

@@ -36,10 +36,8 @@ export const todolistTool = {
         'BAD: "打开文件A → 修改函数X → 保存文件A → 运行测试" (too fine-grained, just do it)',
         '',
         '## UPDATE WORKFLOW',
-        'Update the plan AFTER completing each step — not before, not in bulk at the end.',
-        '- Mark the current step `done` + set the next step `in_progress` in the same `todolist` call.',
-        '- Use `update` for single steps; `batch_update` ONLY when 2-3 steps finish simultaneously.',
-        '- ⛔ NEVER defer all updates to the end — this defeats the purpose of the progress chip.',
+        'Update in real-time after completing each step (one step = one update, or batch_update for 2-3 at once).',
+        'Do NOT batch all updates at the end — update as you go so the chip reflects real progress.',
         'All steps done/skipped → plan auto-finishes, archives, and chip disappears.',
         '',
         '## LIFECYCLE',