@noobdemon/noob-cli 1.10.4 → 1.10.5

package/package.json

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

package/src/agent.js

@@ -392,7 +392,7 @@ export function relTime(ts) {
 // chèn ngay sau SYSTEM để model biết và dùng được.
 // recentSessions: breadcrumbs các phiên trước cùng workspace (repl.js cung cấp)
 // → chèn ngay sau memoryBlock() để model "thấy" lịch sử dù chưa /resume.
-function buildSystem(history, extraToolsDoc, goal, recentSessions) {
+export function buildSystem(history, extraToolsDoc, goal, recentSessions) {
   const parts = [SYSTEM, "", memoryBlock()];
   if (recentSessions && recentSessions.length) {
     parts.push("", recentSessionsBlock(recentSessions));
@@ -403,7 +403,7 @@ function buildSystem(history, extraToolsDoc, goal, recentSessions) {
   return parts.join("\n");
 }
 
-function buildUserMessage(history) {
+export function buildUserMessage(history) {
   const msgs = compact(history, MAX_PROMPT_CHARS);
   const parts = [filesLedger(history), "", "=".repeat(60), "# CONVERSATION", ""];
   for (const m of msgs) {
@@ -496,11 +496,14 @@ export async function runAgent({ history, model, signal, onTool, onStatus, onDel
   // chạy không giới hạn token. Dừng theo: GOAL đạt, <<LOOP_DONE>>, <<ULTRA_DONE>>,
   // model tự kết thúc reply không có tool block, hoặc user Ctrl+C.
   const recentCalls = []; // {name, inputStr} — theo dõi vòng lặp
-  let idleRecoveries = 0; // đếm số lần idle recovery —max 5 để tránh infinite loop
   // Effort classifier: phân loại task từ user message gốc → set effort level.
   // Chỉ classify 1 lần ở bước đầu, giữ nguyên suốt task (thay đổi giữa chừng gây bất ổn).
   const effort = classifyEffort(history.find((m) => m.role === "user")?.content || "");
   for (let step = 0; step < MAX_STEPS; step++) {
+    // Check abort signal ngay đầu mỗi iteration để Ctrl+C dừng ngay lập tức
+    if (signal?.aborted) {
+      throw new Error("aborted");
+    }
     // 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…`);
 
@@ -556,22 +559,7 @@ export async function runAgent({ history, model, signal, onTool, onStatus, onDel
         });
         continue;
       }
-      // ── Idle recovery: model dừng nhưng còn task chưa xong ────────────
-      // Nếu pendingTasks còn items → KHÔNG return, inject nudge rồi continue loop.
-      // Giới hạn 5 lần recovery để tránh infinite loop.
-      {
-        const tasks = pendingTasks || [];
-        const remaining = tasks.length;
-        if (remaining > 0 && idleRecoveries < 5) {
-          idleRecoveries++;
-          const next = tasks[0];
-          history.push({
-            role: "user",
-            content: `[SYSTEM] Bạn vừa dừng giữa chừng. Còn ${remaining} việc chưa hoàn thành: ${tasks.map((t) => `"${t}"`).join(", ")}. Việc tiếp theo BẮT BUỘC phải làm ngay: "${next}". Gọi tool (write_file/edit_file/run_command) để làm việc này. KHÔNG dừng, KHÔNG tóm tắt, KHÔNG đợi user.`,
-          });
-          continue;
-        }
-      }
+      // Model dừng (không tool call, không incomplete) → return để repl quyết định tiếp tục hay không
       return text; // final answer
     }
 

package/src/repl.js

@@ -3,9 +3,9 @@ import fs from "node:fs";
 import path from "node:path";
 import chalk from "chalk";
 import { createTui } from "./tui.js";
-import { runAgent, maybeSummarize } from "./agent.js";
+import { runAgent, maybeSummarize, buildSystem, buildUserMessage } from "./agent.js";
 import { runSubAgent, spawnAgentToolsDoc, MAX_SUBAGENT_DEPTH } from "./subagent.js";
-import { TokenMeter, countMessages, CONTEXT_WINDOW } from "./tokens.js";
+import { TokenMeter, countMessages, CONTEXT_WINDOW, countTokens } from "./tokens.js";
 import { stream, usage, ApiError, resetMemoryToken } from "./api.js";
 import { runTool, describe, DESTRUCTIVE, addRoot, removeRoot, listRoots, OutOfScopeError, nearestExistingDir } from "./tools.js";
 import { MODELS, PROVIDERS, findModel, providerColor, DEFAULT_MODEL } from "./models.js";
@@ -1254,7 +1254,9 @@ NGUYÊN TẮC:
         updateTitle();
       }
       // Tính context tokens realtime — đếm system prompt + history trước khi gửi.
-      tokenMeter.setContext(countMessages(state.history));
+      const systemPrompt = buildSystem(state.history, state.agentMode ? spawnAgentToolsDoc(0) : "", state.goal, sessions.list(5, process.cwd()).filter((s) => s.id !== session?.id));
+      const userMessage = buildUserMessage(state.history);
+      tokenMeter.setContext(countTokens(systemPrompt) + countTokens(userMessage));
       if (process.stdin.isTTY && !state.steerHintShown) {
         console.log(c.dim("  " + t.steerHint));
         state.steerHintShown = true;
@@ -1348,6 +1350,76 @@ NGUYÊN TẮC:
       // Lượt cuối không stream ra gì (vd model suy luận trả 1 cục) → in fallback.
       if ((!printer || !printer.started) && answer?.trim())
         printAnswer(answer, state.model.name, providerColor(state.model.provider));
+
+      // ── Idle verification: model dừng → gọi model check task hoàn thành chưa ───
+      // Nếu chưa xong → chạy runAgent tiếp với prompt verify
+      // Giới hạn 3 lần để tránh infinite loop
+      const MAX_IDLE_VERIFY = 3;
+      let idleVerifyCount = 0;
+      while (idleVerifyCount < MAX_IDLE_VERIFY) {
+        // Kiểm tra xem model có thực sự xong không
+        const verifyPrompt = `[SYSTEM] Bạn vừa dừng lại. Hãy KIỂM TRA LẠI xem task đã hoàn thành chưa:
+1. Đọc file kết quả (nếu có) bằng read_file để xác nhận nội dung
+2. Chạy test/build nếu task liên quan code
+3. So sánh với yêu cầu ban đầu của user
+
+Nếu thực sự HOÀN THÀNH → trả lời "XÁC NHẬN: [tóm tắt ngắn những gì đã làm]"
+Nếu CHƯA XONG → gọi tool (write_file/edit_file/run_command) để hoàn thành tiếp. KHÔNG dừng.`;
+
+        state.history.push({ role: "user", content: verifyPrompt });
+        
+        const nextAnswer = await runAgent({
+          history: state.history,
+          model: state.model.id,
+          signal: abort.signal,
+          tokenMeter,
+          goal: state.goal,
+          recentSessions: sessions.list(5, process.cwd()).filter((s) => s.id !== session?.id),
+          extraToolsDoc: state.agentMode ? spawnAgentToolsDoc(0) : "",
+          pendingTasks: (state.todos || []).filter((t) => !t.done).map((t) => t.text),
+          onStatus: () => tick(t.thinking),
+          onSteer: () => {
+            if (!pending.length) return [];
+            const msgs = pending.splice(0);
+            stopSpin();
+            for (const msg of msgs) console.log(c.user("  " + t.steerInject(truncate(msg, 70))));
+            startSpin(t.thinking);
+            return msgs;
+          },
+          onDelta: (ev) => {
+            if (ev.type === "step-start") {
+              printer = makeStreamPrinter(state.model.name, providerColor(state.model.provider));
+            } else if (ev.type === "delta") {
+              if (printer.suppressing) return printer.push(ev.text);
+              stopSpin();
+              printer.push(ev.text);
+              if (printer.suppressing) startSpin(t.thinking);
+            } else if (ev.type === "step-end") {
+              printer?.flush();
+            }
+          },
+          onTool: (name, input) => dispatchTool(name, input, 0),
+        });
+
+        if (!nextAnswer?.trim()) {
+          idleVerifyCount++;
+          continue;
+        }
+
+        // In response nếu có
+        if ((!printer || !printer.started) && nextAnswer?.trim())
+          printAnswer(nextAnswer, state.model.name, providerColor(state.model.provider));
+
+        // Check nếu model confirm "XÁC NHẬN"
+        if (/XÁC NHẬN/i.test(nextAnswer)) {
+          answer = nextAnswer;
+          break;
+        }
+
+        idleVerifyCount++;
+        // Nếu model gọi tool thì tiếp tục loop (sẽ được xử lý bởi runAgent tiếp theo)
+      }
+
       // Parse todo từ model output → render trên status bar.
       state.todos = parseTodosFromHistory(state.history);
       tui.setTodos(state.todos);