@noobdemon/noob-cli 1.6.0 → 1.7.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@noobdemon/noob-cli",
-  "version": "1.6.0",
+  "version": "1.7.1",
   "publishConfig": {
     "access": "public"
   },

package/src/repl.js

@@ -30,6 +30,8 @@ const SLASH = [
   { 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: "/agent", desc: "bật/tắt agent mode (spawn sub-agent)" },
+  { name: "/tokens", desc: "xem số token đã dùng phiên này" },
   { name: "/learn", desc: "chưng cất bài học vào noob.md" },
   { name: "/memory", desc: "xem bộ nhớ noob.md" },
   { name: "/login", desc: "đăng nhập bằng API key" },
@@ -129,7 +131,9 @@ export async function startRepl(opts = {}) {
     autoApprove: new Set(),
     yolo: !!opts.yolo || config.yoloDefault, // cờ --yolo HOẶC mặc định đã lưu (/auto-yolo)
     ultra: false, // chế độ tự hành (self-quest) đang chạy?
+    agentMode: false, // /agent on → cho phép spawn_agent / spawn_agents
   };
+  const tokenMeter = new TokenMeter();
 
   // Prompt = dòng trạng thái sống. Luôn phản ánh yolo + version theo thời gian
   // thực (vẽ lại mỗi lượt và ngay khi Shift+Tab), nên không cần gõ /status.
@@ -379,24 +383,39 @@ Tự đánh giá: còn THIẾU gì để đạt mục tiêu? Chọn 1 nhiệm v
     state.mode = "chat"; // tự hành chỉ chạy ở chế độ agent
     state.ultra = true;
     console.log(c.accent("  🚀 " + t.ultraOn));
-    let answer = await handle(ultraStart(goal));
-    persist();
-    let i = 0;
-    // Lượt đầu = lập kế hoạch → KHÔNG xét hoàn thành. Mỗi vòng sau là một lượt
-    // "tiếp tục" có cổng kiểm chứng; chỉ dừng khi token nằm ở CUỐI câu trả lời.
-    while (state.ultra && i < MAX_QUESTS) {
-      if (!answer) break; // lượt bị ngắt/ lỗi → dừng tự hành, đừng quay vô ích
-      i++;
-      console.log(c.accent("  ↻ " + t.ultraQuest(i)));
-      answer = await handle(ultraContinue(goal));
+    // Mốc history TRƯỚC khi ULTRA bơm prompt — kết thúc thì cắt về để các lượt sau không bị "dính" mục tiêu cũ.
+    const baseLen = state.history.length;
+    try {
+      let answer = await handle(ultraStart(goal));
       persist();
-      if (answer && ultraIsDone(answer)) {
-        console.log(c.ok("  ✓ " + t.ultraDone));
-        break;
+      let i = 0;
+      // Lượt đầu = lập kế hoạch → KHÔNG xét hoàn thành. Mỗi vòng sau là một lượt
+      // "tiếp tục" có cổng kiểm chứng; chỉ dừng khi token nằm ở CUỐI câu trả lời.
+      while (state.ultra && i < MAX_QUESTS) {
+        if (!answer) break; // lượt bị ngắt/ lỗi → dừng tự hành, đừng quay vô ích
+        i++;
+        console.log(c.accent("  ↻ " + t.ultraQuest(i)));
+        answer = await handle(ultraContinue(goal));
+        persist();
+        if (answer && ultraIsDone(answer)) {
+          console.log(c.ok("  ✓ " + t.ultraDone));
+          break;
+        }
       }
+      if (state.ultra && i >= MAX_QUESTS) console.log(c.tool("  " + t.ultraMax));
+    } finally {
+      // Dọn dấu vết ULTRA khỏi history (prompt khởi động, các lượt "tiếp tục",
+      // token <<ULTRA_DONE>>…) để các yêu cầu SAU đó không bị model coi như vẫn
+      // đang tự hành / vẫn theo đuổi mục tiêu cũ.
+      state.ultra = false;
+      if (state.history.length > baseLen) state.history.length = baseLen;
+      state.history.push({
+        role: "user",
+        content: "[Phiên ULTRA đã KẾT THÚC — mục tiêu cũ: " + goal + ". Bỏ qua mọi chỉ dẫn ULTRA trước đó, KHÔNG tự hành tiếp, KHÔNG phát token " + ULTRA_DONE + ". Chờ yêu cầu mới.]",
+      });
+      state.history.push({ role: "assistant", content: "OK, đã thoát chế độ ULTRA." });
+      persist();
     }
-    if (state.ultra && i >= MAX_QUESTS) console.log(c.tool("  " + t.ultraMax));
-    state.ultra = false;
   }
 
   // /init — quét dự án & sinh noob.md tổng quan (giống `/init` của Claude Code).
@@ -584,7 +603,9 @@ NGUYÊN TẮC:
     let timer = null;
     const tick = (label) => {
       const elapsed = ((Date.now() - t0) / 1000).toFixed(0);
-      tui.status(c.dim(`${label}… ${elapsed}s`));
+      // Chèn token usage realtime (↑input ↓output) ngay cạnh spinner để người
+      // dùng thấy được số token đang cộng dồn trong khi model stream.
+      tui.status(c.dim(`${label}… ${elapsed}s · ${tokenMeter.format()}`));
     };
     const stopSpin = () => {
       if (timer) {
@@ -609,7 +630,8 @@ NGUYÊN TẮC:
           message: text,
           signal: abort.signal,
           onStatus: (s) => {
-            if (!printer.started) tui.status(c.dim("  " + s));
+            // Kèm token meter để nhánh merge/search cũng thấy ↑input ↓output realtime.
+            if (!printer.started) tui.status(c.dim(`  ${s} · ${tokenMeter.format()}`));
           },
           onDelta: (d) => {
             stopSpin();
@@ -634,10 +656,49 @@ NGUYÊN TẮC:
       startSpin(t.thinking);
       let printer = null;
 
+      const dispatchTool = async (name, input, depth = 0) => {
+        // spawn_agent / spawn_agents chỉ được phép khi agentMode bật; depth giới hạn
+        // bởi MAX_SUBAGENT_DEPTH để tránh đệ quy nổ.
+        if (name === "spawn_agent" || name === "spawn_agents") {
+          if (!state.agentMode)
+            return { allow: true, result: "ERROR: agent mode đang TẮT — gõ /agent on để bật trước khi spawn." };
+          if (depth >= MAX_SUBAGENT_DEPTH)
+            return { allow: true, result: `ERROR: đã đạt depth tối đa (${MAX_SUBAGENT_DEPTH}) — không spawn thêm.` };
+          const tasks = name === "spawn_agent" ? [input] : (Array.isArray(input?.agents) ? input.agents : []);
+          if (!tasks.length) return { allow: true, result: "ERROR: thiếu task cho sub-agent." };
+          stopSpin();
+          console.log(chalk.hex("#8b5cf6")(`  ⊕ spawn ${tasks.length} sub-agent (depth ${depth + 1}/${MAX_SUBAGENT_DEPTH})`));
+          startSpin(t.thinking);
+          try {
+            const results = await Promise.all(tasks.map((task, i) =>
+              runSubAgent({
+                task: task?.task || task?.prompt || "",
+                context: task?.context || "",
+                depth: depth + 1,
+                model: state.model.id,
+                signal: abort.signal,
+                tokenMeter,
+                dispatchTool: (n, inp) => dispatchTool(n, inp, depth + 1),
+                onLog: (msg) => { stopSpin(); console.log(chalk.hex("#8b5cf6")("  " + msg)); startSpin(t.thinking); },
+              }).then((r) => `── sub-agent #${i + 1} ──\n${r}`).catch((e) => `── sub-agent #${i + 1} (LỖI) ──\n${e?.message || String(e)}`)
+            ));
+            return { allow: true, result: results.join("\n\n") };
+          } catch (err) {
+            return { allow: true, result: "ERROR sub-agent: " + (err?.message || String(err)) };
+          }
+        }
+        stopSpin();
+        const res = await execTool(name, input);
+        startSpin(t.thinking);
+        return res;
+      };
+
       const answer = await runAgent({
         history: state.history,
         model: state.model.id,
         signal: abort.signal,
+        tokenMeter,
+        extraToolsDoc: state.agentMode ? spawnAgentToolsDoc(0) : "",
         onStatus: () => tick(t.thinking),
         onSteer: () => {
           if (!pending.length) return [];
@@ -659,12 +720,7 @@ NGUYÊN TẮC:
             printer?.flush();
           }
         },
-        onTool: async (name, input) => {
-          stopSpin();
-          const res = await execTool(name, input);
-          startSpin(t.thinking);
-          return res;
-        },
+        onTool: (name, input) => dispatchTool(name, input, 0),
       });
 
       stopSpin();
@@ -773,6 +829,18 @@ NGUYÊN TẮC:
         state.yolo = !state.yolo;
         console.log((state.yolo ? c.err : c.ok)("  " + (state.yolo ? t.yoloOn : t.yoloOff)));
         break;
+      case "agent": {
+        const v = arg.toLowerCase();
+        if (v === "on" || v === "bật" || v === "bat") state.agentMode = true;
+        else if (v === "off" || v === "tắt" || v === "tat") state.agentMode = false;
+        else state.agentMode = !state.agentMode;
+        console.log((state.agentMode ? c.accent : c.dim)("  agent mode: " + (state.agentMode ? "BẬT (spawn_agent / spawn_agents khả dụng, depth tối đa " + MAX_SUBAGENT_DEPTH + ")" : "tắt")));
+        break;
+      }
+      case "tokens": {
+        console.log(c.dim(`  tokens — input: ${tokenMeter.input.toLocaleString("vi-VN")} · output: ${tokenMeter.output.toLocaleString("vi-VN")} · tổng: ${tokenMeter.total.toLocaleString("vi-VN")} · ${tokenMeter.format()}`));
+        break;
+      }
       case "auto-yolo":
       case "autoyolo":
         await toggleAutoYolo();

package/src/subagent.js

@@ -31,14 +31,17 @@ Ví dụ phân cấp: cha giao "build full app" → đẻ 1 sub-agent "build bac
 
 // 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 }) {
+export async function runSubAgent({ task, context, model, signal, dispatchTool, depth = 1, onLog, tokenMeter }) {
   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();
+  // Dùng chung meter của cha nếu được truyền vào → token sub-agent cộng dồn
+  // vào tổng phiên. Nếu không có thì tự tạo cục bộ (giữ tương thích cũ).
+  const meter = tokenMeter || new TokenMeter();
+  const before = { input: meter.input, output: meter.output };
   onLog?.(`↳ sub-agent (depth=${depth}) bắt đầu: ${task.slice(0, 80)}${task.length > 80 ? "…" : ""}`);
   const result = await runAgent({
     history,
@@ -51,6 +54,8 @@ ${context ? `\n# NGỮ CẢNH TỪ CHA\n${context}` : ""}`;
     onDelta: () => {},
     onSteer: () => [],
   });
-  onLog?.(`↳ sub-agent (depth=${depth}) xong (${meter.format()})`);
-  return { result, tokens: { input: meter.input, output: meter.output } };
+  const used = { input: meter.input - before.input, output: meter.output - before.output };
+  onLog?.(`↳ sub-agent (depth=${depth}) xong (↑${used.input} ↓${used.output})`);
+  // Trả về string sạch để cha (model) đọc dễ. Token đã cộng vào meter rồi.
+  return result;
 }