@noobdemon/noob-cli 1.6.0 → 1.7.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@noobdemon/noob-cli",
-  "version": "1.6.0",
+  "version": "1.7.0",
   "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.
@@ -634,10 +638,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 +702,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 +811,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;
 }