mocode-ai 0.3.0 → 0.4.1

package/dist/tools/builtins/grep.js

@@ -5,19 +5,26 @@ import { getSandboxRoot, isInsideRoot, jailResolve } from '../../sandbox/index.j
 // ---------- grep ----------
 export const grepTool = {
     name: 'grep',
-    description: 'Search file contents by regex, returning file:line: matched lines. Recursively searches cwd excluding node_modules/.git. Optional glob restricts file types.' +
-        ' For architecture/call chains, prefer codegraph.',
+    description: 'Search file contents by regex (recursive, excludes node_modules/.git).\n' +
+        'Output: per-file header "<path>: N matches, lines [l1, l2, ...]" + first N matching lines.\n' +
+        'Use the line-number list to call read_file(offset=X, limit=Y) for each region — ' +
+        'do NOT read entire files after grepping. Prefer codegraph for call chains.',
     parameters: {
         type: 'object',
         properties: {
             pattern: { type: 'string', description: 'Regular expression' },
             glob: { type: 'string', description: 'Optional, restrict to a file glob, e.g. *.ts' },
+            max_per_file: {
+                type: 'integer',
+                description: 'Max body lines per file (default 15, cap 50). Line-number list is always full.',
+            },
         },
         required: ['pattern'],
     },
     async execute(args) {
         const pattern = String(args.pattern);
         const g = String(args.glob ?? '**/*');
+        const maxPerFile = Math.min(Math.max(Number(args.max_per_file ?? 15), 1), 50);
         let re;
         try {
             re = new RegExp(pattern);
@@ -34,11 +41,10 @@ export const grepTool = {
             followSymbolicLinks: false, // 不跟随软链目录,防经软链扫到牢外文件
             throwErrorOnBrokenSymbolicLink: false,
         })).filter((f) => isInsideRoot(f)); // 后置兜底:仅留牢内
-        const results = [];
+        const hits = [];
         let scanned = 0;
-        for (const f of files) {
-            if (results.length >= MAX_RESULTS)
-                break;
+        let truncated = false;
+        outer: for (const f of files) {
             let content;
             try {
                 // jailResolve:realpath 化,防「牢内文件软链→牢外」的内容泄露;越界/不可读均 catch 跳过
@@ -49,19 +55,50 @@ export const grepTool = {
             }
             scanned++;
             const lines = content.split(/\r?\n/);
+            const lineNos = [];
             for (let i = 0; i < lines.length; i++) {
-                if (re.test(lines[i])) {
-                    results.push(`${f}:${i + 1}: ${lines[i].trim()}`);
-                    if (results.length >= MAX_RESULTS)
-                        break;
+                if (re.test(lines[i]))
+                    lineNos.push(i + 1);
+            }
+            if (lineNos.length === 0)
+                continue;
+            // 全局配额:行号列表总是计入,body 行数也计入(行号 + body 两条共用 MAX_RESULTS)
+            const totalCost = lineNos.length + Math.min(lineNos.length, maxPerFile);
+            if (hits.length >= MAX_RESULTS || totalCost > MAX_RESULTS * 4) {
+                // 文件过多 / 配额爆:仅追加该文件行号列表,不再展开 body
+                if (hits.length < MAX_RESULTS) {
+                    hits.push({ path: f, lineNos, bodies: [] });
                 }
+                truncated = true;
+                continue;
             }
+            const bodies = lineNos.slice(0, maxPerFile).map((n) => {
+                const trimmed = lines[n - 1].trim();
+                return `  L${n}: ${trimmed}`;
+            });
+            hits.push({ path: f, lineNos, bodies });
         }
-        if (results.length === 0)
+        if (hits.length === 0)
             return `无匹配(扫描了 ${scanned} 个文件)`;
-        let out = results.join('\n');
-        if (results.length >= MAX_RESULTS)
-            out += `\n...(结果达到 ${MAX_RESULTS} 条上限)`;
-        return out;
+        const out = [];
+        let totalShown = 0;
+        for (const h of hits) {
+            const header = `${h.path}: ${h.lineNos.length} 处匹配,行号 [${h.lineNos.join(', ')}]`;
+            out.push(header);
+            if (h.bodies.length > 0) {
+                out.push(...h.bodies);
+            }
+            else {
+                out.push(`  (body 已折叠,见上方行号列表 → read_file 精读)`);
+            }
+            totalShown += h.lineNos.length + h.bodies.length;
+            if (totalShown >= MAX_RESULTS) {
+                out.push(`...(结果达到 ${MAX_RESULTS} 条上限)`);
+                break;
+            }
+        }
+        if (truncated)
+            out.push(`...(仍有更多匹配文件未展示,缩小 glob 或收窄正则)`);
+        return out.join('\n');
     },
 };

package/dist/tools/builtins/read-file.js

@@ -1,24 +1,34 @@
 import { readFile } from 'node:fs/promises';
 import { resolve } from 'node:path';
 import { MAX_FILE_LINES } from '../constants.js';
+/** 默认单次 read_file 拉取的行数。刻意压低,逼 LLM 分块读大文件,
+ * 配合 description 中的 PAGINATION IS MANDATORY 引导。
+ * 300 行 ≈ 一个屏幕的源码量,够定位一段逻辑而不至于吃光上下文。 */
+const DEFAULT_READ_LIMIT = 300;
 // ---------- read_file ----------
 export const readFileTool = {
     name: 'read_file',
-    description: 'Read file content with line numbers. Read before editing. offset (1-based, default 1), limit (default 2000).' +
-        ' For architecture/call chains, prefer codegraph over reading files one at a time.',
+    description: 'Read file content with line numbers. Read before editing.\n' +
+        'For files >500 lines: grep first to locate regions, then call read_file multiple times ' +
+        'with offset+limit (e.g. offset=350, limit=120). Do NOT read an entire large file in one call.\n' +
+        'Prefer codegraph over reading files one at a time for architecture/call-chain questions.',
     parameters: {
         type: 'object',
         properties: {
             path: { type: 'string', description: 'File path, relative to the working directory' },
-            offset: { type: 'integer', description: 'Start line (1-based), default 1' },
-            limit: { type: 'integer', description: 'Max number of lines to read, default 2000' },
+            offset: { type: 'integer', description: 'Start line, 1-based (default 1).' },
+            limit: {
+                type: 'integer',
+                description: 'Max lines to read (default 300, hard cap 2000). Keep ranges ~80-200.',
+            },
         },
         required: ['path'],
     },
     async execute(args) {
         const path = String(args.path);
         const offset = Number(args.offset ?? 1);
-        const limit = Number(args.limit ?? MAX_FILE_LINES);
+        // 无论 LLM 传多大,单次硬钳到 MAX_FILE_LINES,杜绝「绕过分页引导一把全拿」。
+        const limit = Math.min(Number(args.limit ?? DEFAULT_READ_LIMIT), MAX_FILE_LINES);
         const data = await readFile(resolve(path), 'utf8');
         const lines = data.split(/\r?\n/);
         const start = Math.max(0, offset - 1);

package/dist/tools/constants.js

@@ -22,6 +22,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 里剔除的工具(模型根本看不到 → 调不到):

package/dist/ui/prompt.js

@@ -84,24 +84,44 @@ export async function promptWithSlashMenu(opts) {
     let menuOpen = false;
     let selected = 0;
     let filtered = [];
+    const MENU_MAX_VISIBLE = 5;
+    let menuTop = 0; // 窗口首项在 filtered 中的索引,菜单最多显示 MENU_MAX_VISIBLE 条
     let resolved = false;
     let resolve;
     let reject;
-    /** 菜单行(预渲染,带色)——向上展开进内容区底,由 layout 贴入。 */
+    /** 菜单行(预渲染,带色)——向上展开进内容区底,由 layout 贴入。最多显示 MENU_MAX_VISIBLE 条,支持上下滚动。 */
     function menuLines() {
         if (!menuOpen || filtered.length === 0)
             return [];
         const cols = layout.getGeo().cols;
-        const maxName = Math.max(...filtered.map((c) => displayWidth(c.name)));
-        return filtered.map((c, i) => {
+        const visibleCount = Math.min(MENU_MAX_VISIBLE, filtered.length);
+        // 保 selected 在窗口内:selected 顶到上/下边界时才挪 menuTop
+        if (selected < menuTop)
+            menuTop = selected;
+        else if (selected >= menuTop + visibleCount)
+            menuTop = selected - visibleCount + 1;
+        const windowItems = filtered.slice(menuTop, menuTop + visibleCount);
+        const maxName = Math.max(...windowItems.map((c) => displayWidth(c.name)));
+        const hasMoreAbove = menuTop > 0;
+        const hasMoreBelow = menuTop + visibleCount < filtered.length;
+        return windowItems.map((c, i) => {
+            const globalIdx = menuTop + i;
             // 选中项:▸ 与文字均 cyan+bold(去 dim),未选中项保持 dim——选中行整体高亮。
-            const isSel = i === selected;
+            const isSel = globalIdx === selected;
             const color = isSel ? `${ui.cyan}${ui.bold}` : ui.dim;
             const marker = isSel ? `${ui.cyan}${ui.bold}▸${ui.reset}` : ' ';
             const name = padEndDisplay(c.name, maxName);
-            const descW = cols - maxName - 5; // marker + 空格 + 2 间距
-            const desc = descW > 0 ? truncateDisplay(c.desc, descW) : '';
-            return `${marker} ${color}${name}${ui.reset}  ${color}${desc}${ui.reset}`;
+            // 首/末附加滚动指示(▲/▼)而非替换整行
+            let desc = c.desc;
+            let scrollHint = '';
+            if (i === 0 && hasMoreAbove)
+                scrollHint = ' ▲';
+            if (i === windowItems.length - 1 && hasMoreBelow)
+                scrollHint = ' ▼';
+            const hintW = displayWidth(scrollHint);
+            const descW = cols - maxName - 5 - hintW; // marker + 空格 + 2 间距 + hint
+            const descStr = descW > 0 ? truncateDisplay(desc, descW) : '';
+            return `${marker} ${color}${name}${ui.reset}  ${color}${descStr}${ui.reset}${ui.dim}${scrollHint}${ui.reset}`;
         });
     }
     /** 当前光标在该行的显示列(供 layout 定位光标)。 */
@@ -203,12 +223,15 @@ export async function promptWithSlashMenu(opts) {
         if (cl === 0 && lines[0].startsWith('/')) {
             filtered = opts.commands.filter((c) => c.name.startsWith(lines[0]));
             menuOpen = filtered.length > 0;
-            if (selected >= filtered.length)
+            if (selected >= filtered.length) {
                 selected = 0;
+                menuTop = 0;
+            }
         }
         else {
             filtered = [];
             menuOpen = false;
+            menuTop = 0;
         }
     }
     function redraw() {
@@ -283,6 +306,7 @@ export async function promptWithSlashMenu(opts) {
         menuOpen = false;
         filtered = [];
         selected = 0;
+        menuTop = 0;
         if (pasteTimer) {
             clearTimeout(pasteTimer);
             pasteTimer = null;
@@ -469,6 +493,8 @@ export async function promptWithSlashMenu(opts) {
             case 'escape':
                 menuOpen = false;
                 filtered = [];
+                selected = 0;
+                menuTop = 0;
                 redraw();
                 return;
             case 'left':

package/package.json

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