@noobdemon/noob-cli 1.5.4 → 1.6.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@noobdemon/noob-cli",
-  "version": "1.5.4",
+  "version": "1.6.0",
   "publishConfig": {
     "access": "public"
   },
@@ -16,7 +16,8 @@
     "LICENSE"
   ],
   "scripts": {
-    "start": "node bin/noob.js"
+    "start": "node bin/noob.js",
+    "postpublish": "node scripts/notify-discord.js"
   },
   "engines": {
     "node": ">=18"
@@ -38,7 +39,10 @@
     "boxen": "^8.0.1",
     "chalk": "^5.4.1",
     "cli-highlight": "^2.1.11",
+    "gpt-tokenizer": "^3.4.0",
     "gradient-string": "^3.0.0",
+    "marked": "^15.0.12",
+    "marked-terminal": "^7.3.0",
     "ora": "^8.2.0"
   }
 }

package/src/agent.js

@@ -2,6 +2,7 @@ import os from "node:os";
 import { stream } from "./api.js";
 import { loadMemory } from "./memory.js";
 import { t } from "./i18n.js";
+import { countTokens } from "./tokens.js";
 
 export const SYSTEM = `You are noob, an agentic coding assistant in the spirit of Claude Code. You help with software engineering tasks by reading and editing files and running commands in the user's current working directory.
 
@@ -71,8 +72,13 @@ Có — cả 12 test đều pass.
 
 Follow this pattern exactly. Your very first response to a task that needs the filesystem MUST be a tool block — do not refuse or explain limitations.`;
 
-const MAX_STEPS = 300;
+// Số bước tool tối đa cho một lượt. Đặt rất cao theo yêu cầu người dùng: task
+// dài cứ chạy, đừng tự dừng. Người dùng vẫn có thể Ctrl+C bất cứ lúc nào.
+const MAX_STEPS = 10000;
 const MAX_PROMPT_CHARS = 80000; // ngân sách ký tự cho phần hội thoại gửi lên model
+// Khi history vượt ngưỡng này, gọi model phụ tóm tắt các lượt cũ thay vì cắt cụt
+// → giữ được "trí nhớ dài hạn" trong phiên mà không nổ context.
+const SUMMARIZE_THRESHOLD_CHARS = 60000;
 
 // Môi trường chạy thực: model cần biết OS + shell để emit lệnh ĐÚNG. Không có
 // khối này, trên Windows model hay emit lệnh Unix (wc/ls/cat/grep) → run_command
@@ -101,6 +107,7 @@ function runtimeContext() {
 
 // Lược ngữ cảnh để không vượt context khi phiên dài. KHÔNG đụng vào history thật
 // (vẫn lưu/đầy đủ để resume) — chỉ thu gọn BẢN SAO dùng cho prompt.
+// Nếu history đã có summary (do summarizeHistory ghi vào _summary), dùng làm head.
 function compact(history, budget) {
   const len = (m) => (m.content || "").length + 24;
   let total = history.reduce((s, m) => s + len(m), 0);
@@ -125,6 +132,68 @@ function compact(history, budget) {
   return [...head, elided, ...out.slice(tailStart)];
 }
 
+// Bộ nhớ dài hạn cho phiên: khi history phình to, gọi model phụ TÓM TẮT các
+// lượt cũ thành một message system gọn (giữ quyết định, file đã sửa, lý do,
+// việc dở) rồi thay phần đầu history bằng tóm tắt đó. Mutates `history` in place.
+// Trả về true nếu có tóm tắt (để caller persist phiên ngay).
+export async function maybeSummarize(history, { model, signal }) {
+  if (!history?.length) return false;
+  const totalChars = history.reduce((s, m) => s + (m.content?.length || 0), 0);
+  if (totalChars < SUMMARIZE_THRESHOLD_CHARS) return false;
+  // Giữ 8 message cuối nguyên vẹn; tóm tắt phần trước.
+  const keepTail = 8;
+  if (history.length <= keepTail + 2) return false;
+  // Nếu lượt đầu đã là summary (role=system, name=summary) → tóm tắt thêm.
+  const head = history.slice(0, history.length - keepTail);
+  const tail = history.slice(history.length - keepTail);
+  const transcript = head.map((m) => {
+    const role = m.role === "tool" ? `TOOL(${m.name || "?"})` : m.role.toUpperCase();
+    return `## ${role}\n${(m.content || "").slice(0, 2000)}`;
+  }).join("\n\n");
+  const ask = `Tóm tắt phần hội thoại sau thành BẢN GHI NGẮN GỌN (~25-40 dòng, tiếng Việt) để bạn — chính bạn — đọc lại sau và tiếp tục công việc mà không quên ngữ cảnh quan trọng.
+
+BẮT BUỘC giữ lại:
+- Mục tiêu/nhiệm vụ tổng người dùng giao (nguyên văn nếu ngắn).
+- Các quyết định thiết kế đã chốt và LÝ DO.
+- Danh sách file đã tạo/sửa kèm mô tả NGẮN (1 dòng/file) — vai trò của file.
+- Các phát hiện quan trọng từ tool (cấu trúc dự án, lệnh build/test, lỗi đã gặp & cách fix).
+- Việc CÒN DANG DỞ và bước kế tiếp đã dự định.
+- Sở thích/yêu cầu đặc biệt của người dùng.
+
+LOẠI BỎ: chi tiết nội dung file đọc được, output dài của tool, các lượt khảo sát lặp lại.
+
+Định dạng:
+# Tóm tắt phiên (đến lượt thứ ${head.length})
+## Mục tiêu
+- ...
+## Quyết định & lý do
+- ...
+## File đã đụng
+- path — vai trò
+## Phát hiện & lệnh quan trọng
+- ...
+## Đang làm dở / bước tiếp
+- ...
+
+--- HỘI THOẠI CẦN TÓM TẮT ---
+${transcript}`;
+  try {
+    const { text } = await stream({ mode: "chat", model, message: ask, signal });
+    const summary = (text || "").trim();
+    if (!summary || summary.length < 50) return false;
+    // Thay head bằng MỘT message tool tên "session_summary".
+    const summaryMsg = {
+      role: "tool",
+      name: "session_summary",
+      content: `[BỘ NHỚ DÀI HẠN — tóm tắt ${head.length} lượt đầu để không quên ngữ cảnh]\n\n${summary}`,
+    };
+    history.splice(0, history.length, summaryMsg, ...tail);
+    return true;
+  } catch {
+    return false; // tóm tắt lỗi → cứ để compact() cắt cụt như cũ
+  }
+}
+
 // GROUND TRUTH: liệt kê những file ĐÃ THỰC SỰ được ghi/sửa, suy ra từ KẾT QUẢ
 // tool có thật (không phải từ lời model tự kể). Chống lỗi model "tưởng đã tạo
 // file" (chỉ kể trong văn xuôi, quên gọi write_file) rồi khăng khăng "file bị
@@ -148,21 +217,31 @@ function filesLedger(history) {
 
 // Chèn bộ nhớ noob.md (nếu có) vào prompt — đây là phần "tự học" mà noob đọc
 // lại mỗi lượt. Không có file → nhắc model tự tạo khi rút ra điều đáng nhớ.
+// Framing MẠNH: phần `## Rules` là binding (luật dự án), `## Notes` mới là
+// tham khảo. Đặt ngay sau SYSTEM trong buildPrompt() để không bị lost-in-the-middle.
 function memoryBlock() {
   const mem = loadMemory();
   if (!mem)
-    return "# PROJECT MEMORY (noob.md)\n(Chưa có noob.md trong thư mục này. Khi rút ra điều đáng nhớ lâu dài — lệnh build/test/run, quy ước, kiến trúc, sở thích người dùng, tiến độ — hãy TẠO noob.md bằng write_file và ghi vào đó.)";
+    return "# PROJECT RULES & MEMORY (noob.md)\n(Chưa có noob.md trong thư mục này. Khi rút ra điều đáng nhớ lâu dài — lệnh build/test/run, quy ước, kiến trúc, sở thích người dùng, tiến độ — hãy TẠO noob.md bằng write_file và ghi vào đó.)";
   return (
-    "# PROJECT MEMORY (noob.md — điều bạn đã tự học trước đó; xác minh với filesystem trước khi tin tuyệt đối)\n" +
+    "# PROJECT RULES & MEMORY (noob.md) — BINDING\n" +
+    "Phần `## Rules` dưới đây là LUẬT DỰ ÁN bạn PHẢI tuân theo trong mọi hành động ở lượt này — coi như mở rộng của SYSTEM, không phải gợi ý. Phần `## Notes` là quan sát tham khảo, có thể xác minh lại với filesystem nếu nghi ngờ.\n\n" +
     mem +
-    "\n(Học thêm điều đáng nhớ → cập nhật noob.md bằng edit_file/write_file.)"
+    "\n\nTrước khi emit hành động, đối chiếu với `## Rules` ở trên. Học thêm điều đáng nhớ → cập nhật noob.md bằng edit_file/write_file (Notes mới, promote lên Rules khi đã chứng minh)."
   );
 }
 
 // The proxy is stateless, so we serialize the whole transcript into one prompt.
-function buildPrompt(history) {
+// extraToolsDoc: chuỗi mô tả thêm tool (vd spawn_agent khi agent mode bật) được
+// chèn ngay sau SYSTEM để model biết và dùng được.
+function buildPrompt(history, extraToolsDoc) {
   const msgs = compact(history, MAX_PROMPT_CHARS);
-  const parts = [SYSTEM, "", runtimeContext(), "", memoryBlock(), "", filesLedger(history), "", "=".repeat(60), "# CONVERSATION", ""];
+  // Thứ tự CÓ CHỦ ĐÍCH: SYSTEM → memoryBlock (Rules dự án, vị trí mạnh thứ 2,
+  // tránh lost-in-the-middle) → extraToolsDoc → runtimeContext → filesLedger →
+  // CONVERSATION. noob.md (đặc biệt phần `## Rules`) phải nằm sát SYSTEM để model coi là luật.
+  const parts = [SYSTEM, "", memoryBlock()];
+  if (extraToolsDoc) parts.push("", extraToolsDoc);
+  parts.push("", runtimeContext(), "", filesLedger(history), "", "=".repeat(60), "# CONVERSATION", "");
   for (const m of msgs) {
     if (m.role === "user") parts.push(`## USER\n${m.content}`);
     else if (m.role === "assistant") parts.push(`## ASSISTANT\n${m.content}`);
@@ -230,14 +309,21 @@ function extractJsonObject(s, from) {
  * @param {(msg:string)=>void} opts.onStatus      thinking/streaming status
  * @returns {Promise<string>} the final assistant answer (no tool block)
  */
-export async function runAgent({ history, model, signal, onTool, onStatus, onDelta, onSteer }) {
+export async function runAgent({ history, model, signal, onTool, onStatus, onDelta, onSteer, tokenMeter, extraToolsDoc }) {
   for (let step = 0; step < MAX_STEPS; step++) {
+    // Mỗi 100 bước log một mốc để người dùng biết noob vẫn đang chạy (task dài).
+    if (step > 0 && step % 100 === 0) onStatus?.(`đã chạy ${step} bước…`);
+
     // Steering: tin nhắn người dùng gõ GIỮA CHỪNG được chèn vào hội thoại TRƯỚC
     // lần gọi model kế tiếp → model thấy và điều chỉnh ngay trong cùng task.
     const steer = onSteer?.() || [];
     for (const msg of steer) history.push({ role: "user", content: msg });
 
-    const prompt = buildPrompt(history);
+    // Bộ nhớ dài hạn: thử tóm tắt nếu history đã phình. Im lặng nếu không cần.
+    try { await maybeSummarize(history, { model, signal }); } catch {}
+
+    const prompt = buildPrompt(history, extraToolsDoc);
+    tokenMeter?.addInput(countTokens(prompt));
     onStatus?.("thinking");
     onDelta?.({ type: "step-start" });
     const { text } = await stream({
@@ -245,8 +331,12 @@ export async function runAgent({ history, model, signal, onTool, onStatus, onDel
       model,
       message: prompt,
       signal,
-      onDelta: (d) => onDelta?.({ type: "delta", text: d }),
+      onDelta: (d) => {
+        tokenMeter?.pushOutputDelta(d);
+        onDelta?.({ type: "delta", text: d });
+      },
     });
+    tokenMeter?.endOutput();
     onDelta?.({ type: "step-end" });
     history.push({ role: "assistant", content: text });
 

package/src/i18n.js

@@ -57,7 +57,10 @@ export const t = {
   cmdSearch: "/search        bật/tắt chế độ tìm web",
   cmdChat: "/chat          quay lại chế độ chat thường",
   cmdYolo: "/yolo          bật/tắt tự duyệt (hoặc nhấn Shift+Tab)",
+  cmdAgent: "/agent on|off  bật/tắt agent mode (model đẻ sub-agent song song/tuần tự/phân cấp)",
+  cmdTokens: "/tokens        xem số token đã dùng trong phiên",
   cmdAutoYolo: "/auto-yolo     lưu/bỏ yolo làm mặc định mỗi lần chạy (cần xác nhận)",
+  cmdInit: "/init           quét dự án & tạo noob.md (tổng quan + quy ước, như Claude Code)",
   cmdKarpathy: "/karpathy [path]  rà soát code theo 4 nguyên tắc Karpathy (/kc)",
   cmdUltra: "/ultra <mục tiêu>  tự hành: noob tự nghĩ & tự làm nhiệm vụ tới khi xong (/u)",
   cmdLearn: "/learn [ghi chú]  chưng cất bài học của phiên vào noob.md",
@@ -108,6 +111,10 @@ export const t = {
   ultraNeedGoal: "Cần mục tiêu. Dùng: /ultra <mô tả mục tiêu>",
   ultraQuest: (n) => `tự nghĩ nhiệm vụ kế tiếp (vòng ${n})…`,
   learning: "đang chưng cất bài học vào noob.md…",
+  initRunning: "đang quét dự án & soạn noob.md…",
+  initOverwriteWarn: (p) => `⚠ Đã có noob.md tại ${p}. /init sẽ ghi đè nội dung hiện tại.`,
+  initOverwriteConfirm: "Ghi đè? gõ 'y' để xác nhận, phím khác để huỷ › ",
+  initCancel: "Huỷ /init — giữ nguyên noob.md.",
   memoryEmpty: (p) => `Chưa có noob.md. noob sẽ tự tạo ở: ${p}`,
   memoryStat: (n) => `  ·  ${n} dòng / ~200`,
 

package/src/repl.js

@@ -4,6 +4,8 @@ import path from "node:path";
 import chalk from "chalk";
 import { createTui } from "./tui.js";
 import { runAgent } from "./agent.js";
+import { runSubAgent, spawnAgentToolsDoc, MAX_SUBAGENT_DEPTH } from "./subagent.js";
+import { TokenMeter } from "./tokens.js";
 import { stream, usage, ApiError } from "./api.js";
 import { runTool, describe, DESTRUCTIVE } from "./tools.js";
 import { MODELS, PROVIDERS, findModel, providerColor, DEFAULT_MODEL } from "./models.js";
@@ -25,6 +27,7 @@ const SLASH = [
   { name: "/chat", desc: "chế độ chat thường" },
   { name: "/yolo", desc: "bật/tắt tự duyệt" },
   { name: "/auto-yolo", desc: "lưu yolo làm mặc định (cần xác nhận)" },
+  { name: "/init", desc: "quét dự án & tạo noob.md" },
   { name: "/karpathy", desc: "rà soát code (Karpathy)" },
   { name: "/ultra", desc: "tự hành: tự nghĩ & làm nhiệm vụ" },
   { name: "/learn", desc: "chưng cất bài học vào noob.md" },
@@ -396,6 +399,58 @@ Tự đánh giá: còn THIẾU gì để đạt mục tiêu? Chọn 1 nhiệm v
     state.ultra = false;
   }
 
+  // /init — quét dự án & sinh noob.md tổng quan (giống `/init` của Claude Code).
+  // Nếu noob.md đã có: hỏi xác nhận ghi đè trước khi giao việc cho model.
+  async function runInit() {
+    if (!config.apiKey) return console.log(c.tool("  " + t.notLoggedIn));
+    const mem = loadMemory();
+    if (mem) {
+      console.log(c.err("  " + t.initOverwriteWarn(memoryPath())));
+      const ans = ((await ask(c.tool("  " + t.initOverwriteConfirm))) ?? "").trim().toLowerCase();
+      if (ans !== "y" && ans !== "yes" && ans !== "có") {
+        console.log(c.dim("  " + t.initCancel));
+        return;
+      }
+    }
+    const prompt = `Hãy KHỞI TẠO bộ nhớ dự án \`noob.md\` ở thư mục gốc (giống cách Claude Code chạy /init).
+
+QUY TRÌNH (làm bằng tool, không nói suông):
+1. list_dir thư mục gốc để nhìn tổng quan.
+2. Đọc các file quan trọng nếu có: package.json, README.md, pyproject.toml, requirements.txt, Cargo.toml, go.mod, tsconfig.json, vite.config*, next.config*, Makefile, Dockerfile, .editorconfig, .eslintrc*, .prettierrc*.
+3. Lướt qua thư mục mã nguồn chính (src/, lib/, app/…) bằng list_dir/glob để hiểu kiến trúc — đừng đọc hết, chỉ đủ để nắm cấu trúc.
+4. Nếu repo có test/build script, ghi lại lệnh CHÍNH XÁC (ví dụ \`npm test\`, \`pytest\`, \`cargo build\`).
+
+SAU ĐÓ ghi \`noob.md\` bằng write_file (ghi đè nếu đã có) với cấu trúc CHÍNH XÁC sau:
+
+# noob.md
+
+## Tổng quan
+- 2–5 gạch đầu dòng: tên dự án, mục đích, ngôn ngữ/runtime, framework chính.
+
+## Lệnh thường dùng
+- build: <lệnh hoặc "chưa rõ">
+- test: <lệnh>
+- run/dev: <lệnh>
+- lint/format: <lệnh nếu có>
+
+## Kiến trúc
+- 3–8 gạch đầu dòng mô tả các thư mục/module chính & vai trò.
+
+## Rules
+- (để trống hoặc thêm quy ước ĐÃ CHỐT rút ra từ config: ví dụ "dùng ES modules (type: module)", "Node >=18"…)
+
+## Notes
+- (để trống — sẽ tự bổ sung sau khi học thêm trong quá trình làm việc)
+
+NGUYÊN TẮC:
+- Chỉ ghi sự thật rút ra từ file thật. KHÔNG bịa lệnh/quy ước không có cơ sở.
+- Ngắn gọn (~80–150 dòng), mỗi ý một gạch đầu dòng.
+- Khi xong, in 1 đoạn tóm tắt rất ngắn về những gì đã ghi vào noob.md.`;
+    console.log(c.tool("  📋 " + t.initRunning));
+    await handle(prompt);
+    persist();
+  }
+
   // /learn [ghi chú] — bắt noob chưng cất điều đáng nhớ của phiên vào noob.md.
   async function runLearn(arg) {
     if (!config.apiKey) return console.log(c.tool("  " + t.notLoggedIn));
@@ -731,6 +786,9 @@ Tự đánh giá: còn THIẾU gì để đạt mục tiêu? Chọn 1 nhiệm v
       case "u":
         await runUltra(arg);
         break;
+      case "init":
+        await runInit();
+        break;
       case "learn":
         await runLearn(arg);
         break;
@@ -960,7 +1018,10 @@ function printHelp() {
         "  " + t.cmdSearch,
         "  " + t.cmdChat,
         "  " + t.cmdYolo,
+        "  " + t.cmdAgent,
+        "  " + t.cmdTokens,
         "  " + t.cmdAutoYolo,
+        "  " + t.cmdInit,
         "  " + t.cmdKarpathy,
         "  " + t.cmdUltra,
         "  " + t.cmdLearn,

package/src/subagent.js

@@ -0,0 +1,56 @@
+// Sub-agent: chạy một runAgent() con với history độc lập, dùng chung tool
+// runtime của cha. Hỗ trợ phân cấp (sub-agent có thể đẻ sub-agent tiếp) nhưng
+// giới hạn độ sâu để không nổ. Hỗ trợ song song qua spawn_agents (mảng).
+import { runAgent } from "./agent.js";
+import { TokenMeter } from "./tokens.js";
+
+export const MAX_SUBAGENT_DEPTH = 3;
+
+// Tài liệu tool spawn_agent — chèn vào prompt khi agent mode bật. Mô tả cho
+// model biết WHEN dùng (task lớn, chia được) và HOW (song song vs tuần tự).
+export function spawnAgentToolsDoc(depth = 0) {
+  const canSpawn = depth < MAX_SUBAGENT_DEPTH;
+  return `# AGENT MODE — multi-agent
+
+Bạn đang ở chế độ AGENT. Khi gặp task LỚN có thể chia nhỏ, hãy ủy thác cho sub-agent thay vì tự làm hết một mình:
+
+- spawn_agent  {"task": str, "context"?: str}  — đẻ MỘT sub-agent làm "task". "context" là phần ngữ cảnh cần truyền (file paths, quyết định đã chốt, ràng buộc). Sub-agent có TOÀN BỘ tool (read/write/edit/run/grep/glob…) và history RIÊNG (không thấy hội thoại của bạn). Nó trả về một chuỗi tóm tắt kết quả.
+- spawn_agents {"tasks": [{"task": str, "context"?: str}, …]} — đẻ NHIỀU sub-agent CHẠY SONG SONG. Chỉ dùng khi các task ĐỘC LẬP (không phụ thuộc kết quả của nhau). Trả về mảng tóm tắt theo đúng thứ tự tasks.
+
+Quy tắc dùng:
+1. TUẦN TỰ (task B cần kết quả task A): gọi spawn_agent cho A, đọc kết quả, rồi spawn_agent cho B. KHÔNG dùng spawn_agents.
+2. SONG SONG (các task không liên quan): dùng MỘT lần spawn_agents với mảng tasks. Tiết kiệm thời gian.
+3. PHÂN CẤP (task phức tạp): sub-agent của bạn cũng có spawn_agent, nó tự chia tiếp. Độ sâu tối đa hiện tại: ${MAX_SUBAGENT_DEPTH} (bạn đang ở depth=${depth}${canSpawn ? "" : " — đã chạm trần, KHÔNG được spawn nữa, tự làm"}).
+4. Việc NHỎ/đơn giản: cứ tự làm, đừng spawn cho có. Spawn có overhead (mỗi sub-agent là 1 phiên model riêng → tốn token).
+5. Sau khi gom kết quả từ sub-agent, BẠN là người tổng hợp + trả lời cuối cho user. Sub-agent không nói chuyện trực tiếp với user.
+
+Ví dụ song song: "viết test cho 3 module độc lập" → spawn_agents với 3 tasks.
+Ví dụ tuần tự: "thiết kế schema rồi viết migration" → spawn_agent(thiết kế) → đọc → spawn_agent(viết migration với schema đó).
+Ví dụ phân cấp: cha giao "build full app" → đẻ 1 sub-agent "build backend" → sub-agent đó tự đẻ tiếp "viết auth", "viết CRUD" song song.`;
+}
+
+// Chạy một sub-agent. dispatchTool: hàm để thực thi tool con (chia sẻ với cha).
+// model: dùng chung model của cha. onLog: callback để log tiến độ ra UI cha.
+export async function runSubAgent({ task, context, model, signal, dispatchTool, depth = 1, onLog }) {
+  const sys = `Bạn là SUB-AGENT (depth=${depth}) được agent cha ủy thác MỘT nhiệm vụ cụ thể. Làm xong → trả lời NGẮN GỌN bằng Markdown tóm tắt KẾT QUẢ (file đã đụng, phát hiện, lỗi nếu có). Không tán gẫu. Không hỏi lại cha — tự quyết với thông tin được cấp.
+
+# NHIỆM VỤ
+${task}
+${context ? `\n# NGỮ CẢNH TỪ CHA\n${context}` : ""}`;
+  const history = [{ role: "user", content: sys }];
+  const meter = new TokenMeter();
+  onLog?.(`↳ sub-agent (depth=${depth}) bắt đầu: ${task.slice(0, 80)}${task.length > 80 ? "…" : ""}`);
+  const result = await runAgent({
+    history,
+    model,
+    signal,
+    tokenMeter: meter,
+    extraToolsDoc: spawnAgentToolsDoc(depth),
+    onTool: (name, input) => dispatchTool(name, input, depth),
+    onStatus: () => {},
+    onDelta: () => {},
+    onSteer: () => [],
+  });
+  onLog?.(`↳ sub-agent (depth=${depth}) xong (${meter.format()})`);
+  return { result, tokens: { input: meter.input, output: meter.output } };
+}

package/src/tokens.js

@@ -0,0 +1,70 @@
+// Đếm token cục bộ bằng gpt-tokenizer (cl100k_base - tokenizer của GPT-4).
+// CHÍNH XÁC cho GPT, XẤP XỈ cho Claude/Gemini/khác (sai số ~5-15%) — đủ để
+// hiển thị mang tính tham khảo realtime trong CLI.
+import { encode } from "gpt-tokenizer";
+
+export function countTokens(text) {
+  if (!text) return 0;
+  try {
+    return encode(String(text)).length;
+  } catch {
+    // Fallback heuristic nếu encoder lỗi: ~4 ký tự / token.
+    return Math.ceil(String(text).length / 4);
+  }
+}
+
+// Đếm token cho cả mảng messages dạng {role, content}. Cộng overhead nhỏ
+// (~4 token/message cho role + format) để gần đúng cách provider tính prompt.
+export function countMessages(messages = []) {
+  let n = 0;
+  for (const m of messages) {
+    n += countTokens(m?.content || "") + 4;
+  }
+  return n;
+}
+
+// Bộ đếm cộng dồn cho 1 phiên: input (prompt gửi đi) + output (text stream về).
+// Hỗ trợ cộng dồn theo delta để hiển thị realtime trong lúc stream.
+export class TokenMeter {
+  constructor() {
+    this.input = 0;
+    this.output = 0;
+    this._outBuf = ""; // gom delta để đếm theo batch (đỡ tốn CPU)
+    this._outBufN = 0; // số token đã đếm từ _outBuf (để cộng dồn chính xác)
+  }
+  addInput(n) {
+    this.input += Math.max(0, n | 0);
+  }
+  // Mỗi delta text từ stream: gom vào buffer, định kỳ encode lại để cập nhật
+  // số token. Encode toàn buffer thay vì delta riêng lẻ → chính xác hơn (BPE
+  // gộp các byte qua ranh giới delta).
+  pushOutputDelta(text) {
+    if (!text) return;
+    this._outBuf += text;
+    // Re-encode toàn bộ buffer hiện tại; cập nhật delta vào this.output.
+    const total = countTokens(this._outBuf);
+    if (total > this._outBufN) {
+      this.output += total - this._outBufN;
+      this._outBufN = total;
+    }
+  }
+  // Kết thúc một lượt output → reset buffer (bắt đầu lượt mới).
+  endOutput() {
+    this._outBuf = "";
+    this._outBufN = 0;
+  }
+  get total() {
+    return this.input + this.output;
+  }
+  // Định dạng ngắn để hiển thị ở status bar: "↑1.2k ↓340 (1.5k)".
+  format() {
+    const fmt = (n) => (n >= 1000 ? (n / 1000).toFixed(1) + "k" : String(n));
+    return `↑${fmt(this.input)} ↓${fmt(this.output)} (${fmt(this.total)})`;
+  }
+  reset() {
+    this.input = 0;
+    this.output = 0;
+    this._outBuf = "";
+    this._outBufN = 0;
+  }
+}

package/src/tools.js

@@ -348,6 +348,10 @@ export function describe(name, input) {
       return input.id != null ? `xem tiến trình nền #${input.id}` : "liệt kê tiến trình nền";
     case "kill_bg":
       return `dừng tiến trình nền #${input.id}`;
+    case "spawn_agent":
+      return `↳ sub-agent: ${String(input.task || "").slice(0, 80)}`;
+    case "spawn_agents":
+      return `↳ ${(input.tasks || []).length} sub-agent song song`;
     default:
       return name;
   }

package/src/tui.js

@@ -74,11 +74,12 @@ export function createTui({ onLine, onInterrupt, onEOF, onShiftTab, completer }
 
   let liveOut = ""; // dòng output dở dang (chưa có '\n') hiện ngay trên thanh
   let statusText = null; // text spinner khi đang nghĩ
-  // `busy` = một lượt/tool ĐANG chạy. Hiện spinner dự phòng suốt lượt kể cả lúc
+  // `busy` = một lượt/tool ĐANG chạy. Hiện status bar suốt lượt kể cả lúc
   // statusText tạm trống (vd model ngừng phun token giữa các bước) → người dùng
   // LUÔN thấy rõ "đang chạy", không bị tưởng treo.
   let busy = false;
   let busyLabel = "";
+  let busyStartedAt = 0; // mốc thời gian để hiển thị elapsed
   let frame = 0;
   let frameTimer = null;
   let prevRows = 0;

package/src/ui.js

@@ -1,7 +1,9 @@
 import chalk from "chalk";
 import gradient from "gradient-string";
 import boxen from "boxen";
-import { highlight, supportsLanguage } from "cli-highlight";
+import { supportsLanguage } from "cli-highlight";
+import { marked } from "marked";
+import { markedTerminal } from "marked-terminal";
 import { PROVIDERS, providerColor } from "./models.js";
 import { t } from "./i18n.js";
 
@@ -47,60 +49,46 @@ export function modelBadge(model) {
 }
 
 // ── Markdown → ANSI ────────────────────────────────────────────────────────
-function inline(s) {
-  return s
-    .replace(/\*\*(.+?)\*\*/g, (_, t) => chalk.bold(t))
-    .replace(/(^|[^*])\*(?!\*)(.+?)\*(?!\*)/g, (_, p, t) => p + chalk.italic(t))
-    .replace(/`([^`]+)`/g, (_, t) => chalk.hex("#f59e0b")(t))
-    .replace(/\[(.+?)\]\((.+?)\)/g, (_, t, u) => chalk.underline.cyan(t) + c.dim(` (${u})`));
-}
+// Dùng marked + marked-terminal. Một vài lưu ý từ source marked-terminal:
+//   * option `code` là STYLE fallback cho code block, không phải callback render.
+//     Body code block đã được cli-highlight xử lý sẵn → ta truyền highlightOptions
+//     để tô màu, và post-process để thêm viền trái `│`.
+//   * option `listitem` chạy AFTER bullet `*` được prepend, nên KHÔNG thêm `•` ở đây.
+//     Đổi bullet ở bước post-process.
+//   * option `href` không nên bọc ngoặc — wrapper tự thêm `(...)`.
+const BULLET = c.accent("•");
 
-function renderCode(code, lang) {
-  let body = code.replace(/\n$/, "");
-  try {
-    if (lang && supportsLanguage(lang)) body = highlight(body, { language: lang });
-    else body = highlight(body, { ignoreIllegals: true });
-  } catch {
-    /* fall through to raw */
-  }
-  const label = c.dim((lang || "code") + " ▸ ");
-  const lines = body.split("\n").map((l) => c.dim("│ ") + l);
-  return "\n" + label + "\n" + lines.join("\n") + "\n";
+marked.use(
+  markedTerminal(
+    {
+      width: Math.min(term(), 100),
+      reflowText: true,
+      tab: 2,
+      showSectionPrefix: false,
+      firstHeading: (s) => brand(s),
+      heading: chalk.hex("#a78bfa").bold,
+      blockquote: chalk.hex("#6b7280").italic,
+      strong: chalk.bold,
+      em: chalk.italic,
+      codespan: chalk.bgHex("#1f2937").hex("#fbbf24"),
+      hr: () => rule(),
+      link: chalk.hex("#06b6d4").underline,
+      href: chalk.hex("#9ca3af"),
+      code: chalk.hex("#f59e0b"),
+    },
+    { ignoreIllegals: true },
+  ),
+);
+
+// Post-process: đổi bullet `*` thành `•` màu accent, thêm viền `│` cho block code (4-space indent).
+function prettify(s) {
+  return s
+    .replace(/^( *)\* /gm, (_, sp) => sp + BULLET + " ")
+    .replace(/^ {4}(.*)$/gm, (_, rest) => c.dim("│ ") + rest);
 }
 
 export function renderMarkdown(md) {
-  const out = [];
-  const lines = md.split("\n");
-  let i = 0;
-  while (i < lines.length) {
-    const line = lines[i];
-    const fence = line.match(/^```(\w*)\s*$/);
-    if (fence) {
-      const lang = fence[1];
-      const buf = [];
-      i++;
-      while (i < lines.length && !/^```\s*$/.test(lines[i])) buf.push(lines[i++]);
-      i++; // skip closing fence
-      out.push(renderCode(buf.join("\n"), lang));
-      continue;
-    }
-    let m;
-    if ((m = line.match(/^(#{1,6})\s+(.*)$/))) {
-      out.push(brand(chalk.bold(m[2])));
-    } else if (/^\s*[-*+]\s+/.test(line)) {
-      out.push(line.replace(/^(\s*)[-*+]\s+/, (_, sp) => sp + c.accent(" • ")) .replace(/^(\s*\S+\s)(.*)$/, (_, pre, rest) => pre + inline(rest)));
-    } else if ((m = line.match(/^(\s*)(\d+)\.\s+(.*)$/))) {
-      out.push(m[1] + c.accent(` ${m[2]}. `) + inline(m[3]));
-    } else if (/^\s*>\s?/.test(line)) {
-      out.push(c.dim("┃ ") + c.dim(inline(line.replace(/^\s*>\s?/, ""))));
-    } else if (/^\s*([-*_])\1{2,}\s*$/.test(line)) {
-      out.push(rule());
-    } else {
-      out.push(inline(line));
-    }
-    i++;
-  }
-  return out.join("\n");
+  return prettify(marked.parse(md || "")).trimEnd();
 }
 
 export function box(content, title, color = "#a78bfa") {