mocode-ai 0.4.0 → 0.4.2

package/dist/tools/constants.js

@@ -1,4 +1,5 @@
 /** 工具共享的截断 / 上限 / 忽略规则。 */
+import { isMemoryEnabled } from '../config/index.js';
 export const MAX_FILE_LINES = 2000;
 export const MAX_OUTPUT = 20000;
 export const MAX_RESULTS = 100;
@@ -22,6 +23,23 @@ export const GC_DAYS = 90;
 /** memory_search 结果(召回的记忆正文)的放宽上限:指令性内容,中截破坏语义,对齐 use_skill。 */
 export const MAX_MEMORY_RESULT = 64000;
 export const IGNORE = ['**/node_modules/**', '**/.git/**'];
+// ── Context Budget Scheduler(五区分账)────────────────────────────────────
+/** Hot/Cold 划分:当前 step 起往前 N 个 user turn 之内的工具结果视为 Hot(绝对不压),
+ * 之外的视为 Cold(可调度器压)。默认 4 = 跨过 4 个用户问题仍生效。 */
+export const HOT_TURN_WINDOW = 4;
+/** Cold 区内可被就地 stub 的 tool 消息最低 age(经过的消费者 push 数)。默认 2,
+ * 与 lifecycle.ts DEFAULT_AGE_THRESHOLD 对齐。 */
+export const TOOL_OLD_AGE = 2;
+/** 五区预算占比(总和 0.95,留 5% 给 Reserve)。对齐 user 修正版:
+ * System 15 / History 20 / Hot Tool 25 / Cold Tool 25 / Summary 10。 */
+export const BUDGET_RATIO = {
+    system: 0.15,
+    history: 0.20,
+    toolRecent: 0.25,
+    toolOld: 0.25,
+    summary: 0.10,
+    reserve: 0.05,
+};
 // ── plan 模式(只读规划,不执行)──────────────────────────────────────────────
 /**
  * plan 模式下从工具 schema 里剔除的工具(模型根本看不到 → 调不到):
@@ -38,3 +56,18 @@ export const PLAN_DISABLED_TOOLS = new Set([
     'memory_forget',
     'task',
 ]);
+/**
+ * 按当前 isMemoryEnabled() 现算 plan 模式应屏蔽的工具。
+ * memoryEnabled=false 时记忆工具整体不在 builtinTools 里,plan 屏蔽集里也无须再列 ——
+ * 反之留着只是死名字。统一过滤,避免 Set 里残留与已下架工具不一致的概念性冗余。
+ * 调用方(agent/core 串行分支、llm/planChatTools)每次 chat 时调本函数拿当前值。
+ */
+export function getPlanDisabledTools() {
+    if (isMemoryEnabled())
+        return PLAN_DISABLED_TOOLS;
+    const next = new Set(PLAN_DISABLED_TOOLS);
+    next.delete('memory_save');
+    next.delete('memory_update');
+    next.delete('memory_forget');
+    return next;
+}

package/dist/ui/layout.js

@@ -742,29 +742,46 @@ function composeSpinnerLine(status, cols) {
     const rightStr = tail ? `${ui.yellow}${tail}${ui.reset}` : '';
     return twoColumn(lead, leadW, rightStr, tailW, cols);
 }
-/** 下线之下那行(model 行):左 = 模式标识;右 = context + cwd,右端对齐。
+/** 下线之下那行(model 行):左 = 模式标识 + 本轮 token chip;右 = context + cwd,右端对齐。
  *  活跃 plan chip 不再放这里,改放 spinner 行上方的「虚拟空行」(contentBottom+1,见 drawStatusBar),
  *  既不挤 model 行,又给输入区上方留出可视分隔带。 */
 function composeModelLine(status, cols) {
     const ctx = status.contextBar; // 已带色
     const ctxW = ansiDisplayWidth(ctx);
-    // 左段:仅模式标识
+    // 左段:模式标识 + 本轮 token chip。token chip 仅展示总量,用 mid 灰,不抢主色。
     const modeTag = status.modeTag ?? '';
     const modeColor = modeTag === 'plan' ? ui.yellow : ui.brightCyan;
-    const modePart = modeTag
-        ? `${modeColor}${modeTag}${ui.reset}`
-        : '';
-    const leftStr = modePart;
-    const leftW = modeTag ? displayWidth(modeTag) : 0;
+    const modePart = modeTag ? `${modeColor}${modeTag}${ui.reset}` : '';
+    const modeW = modeTag ? displayWidth(modeTag) : 0;
+    const tokChip = formatTurnTokenChip(status.lastTurnUsage);
+    const tokW = displayWidth(stripAnsi(tokChip));
+    // 合并左段:chip 前留 2 空格分隔,无 modeTag 也允许仅显示 chip(兜底边角)。
+    const sep = modePart && tokChip ? '  ' : '';
+    const leftStr = `${modePart}${sep}${tokChip}`;
+    const leftW = modeW + (modePart && tokChip ? sep.length : 0) + tokW;
     // 右段:ctx + sep + cwd,右端对齐。cwd 按预算截断,极窄(<6)隐藏。
+    // 任一 chip 极宽时收紧 cwd(toolbar 列挤压场景),先从 cwd 砍、再隐藏 cwd、再隐藏 token chip。
     const minGap = 2;
-    const cwdBudget = cols - leftW - minGap - ctxW - STATUS_SEP_W - 1;
-    const cwd = cwdBudget >= 6 ? truncateDisplay(status.cwd, cwdBudget) : '';
-    const cwdW = displayWidth(cwd);
-    const rightStr = `${ctx}${STATUS_SEP}${ui.dim}${cwd}${ui.reset}`;
-    const rightW = ctxW + STATUS_SEP_W + cwdW;
+    let cwdBudget = cols - leftW - minGap - ctxW - STATUS_SEP_W - 1;
+    let cwd = cwdBudget >= 6 ? truncateDisplay(status.cwd, cwdBudget) : '';
+    let cwdW = displayWidth(cwd);
+    let rightStr = `${ctx}${STATUS_SEP}${ui.dim}${cwd}${ui.reset}`;
+    let rightW = ctxW + STATUS_SEP_W + cwdW;
+    // 极窄(<24 列含 ctx):藏 token chip
+    if (modePart && tokChip && rightW + minGap + leftW > cols) {
+        return twoColumn(modePart, modeW, rightStr, rightW, cols);
+    }
     return twoColumn(leftStr, leftW, rightStr, rightW, cols);
 }
+/** 把本轮 token 总量格式化成 chip 文本(纯字符串,带 ANSI 色)。无 usage 返空串。 */
+function formatTurnTokenChip(usage) {
+    if (!usage || !usage.totalTokens)
+        return '';
+    const n = usage.totalTokens;
+    const text = n < 1000 ? `${n}` : `${(n / 1000).toFixed(n >= 10000 ? 0 : 1)}k`;
+    // 单段:量足够稳定时不显分项,避免无谓视觉负担。chip 用 mid 灰(降优先级)— 模式仍是主色。
+    return `${ui.dim}${text} tokens${ui.reset}`;
+}
 /** spinner 行上方的「虚拟空行」(contentBottom+1)。
  *  - 有活跃 plan:显「plan: <summary> ▸ N. step」整行左对齐(yellow + dim)
  *  - 无活跃 plan:空(保留原分隔视觉,避免内容贴输入区)
@@ -929,7 +946,7 @@ export function clearLiveAtCursor() {
     frameRow = 0;
     frameCol = 0;
 }
-/** 更新状态行基线(模型 / context / cwd / 模式标识 / 活跃 plan chip)。repl 在轮次边界与切模式时调。 */
+/** 更新状态行基线(模型 / context / cwd / 模式标识 / 活跃 plan chip / 本轮 token chip)。repl 在轮次边界与切模式时调。 */
 export function setStatusBase(b) {
     base = b;
 }

package/package.json

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