@noobdemon/noob-cli 1.5.1 → 1.5.3

package/package.json

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

package/src/api.js

@@ -41,10 +41,59 @@ async function parseError(resp) {
 
 /**
  * Stream a chat/merge/search request from the gateway.
+ *
+ * Auto-continue (chat only): the chat upstream runs on Vercel, which kills the
+ * function at ~300s. A long reply gets cut mid-stream — the gateway flags this
+ * with `{truncated:true}` (or the connection just drops with no `{done}`). When
+ * that happens we re-send the SAME transcript plus the partial reply so far and
+ * ask the model to write ONLY the rest, then append it. The caller sees one
+ * seamless stream. Capped by `maxContinues` so a genuinely broken upstream can't
+ * loop forever.
+ *
  * @returns {Promise<{text:string, reasoning:string}>}
  */
-export async function stream({ mode = "chat", message, model, signal, onDelta, onReasoning, onStatus, idleMs = 120000 }) {
+export async function stream({ mode = "chat", message, model, signal, onDelta, onReasoning, onStatus, idleMs = 120000, maxContinues = 6 }) {
   const endpoint = mode === "search" ? "/api/search" : mode === "merge" ? "/api/merge" : "/api/chat";
+
+  let fullText = "";
+  let reasoning = "";
+  let prompt = message; // prompt gửi đi: lần đầu = nguyên bản, các lần sau = nối tiếp
+
+  for (let attempt = 0; ; attempt++) {
+    const r = await streamOnce({ endpoint, mode, message: prompt, model, signal, idleMs, onStatus, onDelta, onReasoning });
+    fullText = mode === "chat" ? fullText + r.text : r.text; // chat: ghép các đoạn nối tiếp; mode khác: thay thế
+    if (r.reasoning) reasoning = r.reasoning;
+
+    // Còn nối tiếp được không? Chỉ với chat, khi bị cắt, còn lượt, và lần này có
+    // ra chữ thật (đoạn rỗng → coi như xong, tránh lặp vô tận).
+    if (!r.truncated || mode !== "chat" || attempt >= maxContinues || !r.text.trim()) break;
+    prompt = continuationPrompt(message, fullText);
+  }
+
+  return { text: fullText.trim(), reasoning: reasoning.trim() };
+}
+
+// Dựng prompt "nối tiếp" khi câu trả lời bị cắt giữa chừng: gửi lại nguyên ngữ
+// cảnh gốc + phần model đã viết dở + yêu cầu viết TIẾP đúng chỗ dừng, không lặp.
+function continuationPrompt(message, partial) {
+  const bar = "=".repeat(60);
+  return (
+    message +
+    "\n\n" + bar +
+    "\n## ASSISTANT (bị ngắt giữa chừng — phần trả lời dưới đây CHƯA hoàn tất)\n" +
+    partial +
+    "\n\n" + bar +
+    "\n# SYSTEM: Phần trả lời ngay trên bị mạng/timeout cắt ngang trước khi xong. " +
+    "Hãy VIẾT TIẾP liền mạch từ ĐÚNG ký tự cuối cùng ở trên — KHÔNG lặp lại hay diễn đạt lại bất kỳ chữ nào đã hiện, KHÔNG mở đầu lại, KHÔNG thêm lời dẫn. " +
+    "Chỉ xuất phần CÒN LẠI. Nếu đang viết dở một khối tool thì hoàn tất đúng khối đó. Nếu thật ra đã xong, chỉ xuất một dấu cách rồi dừng."
+  );
+}
+
+/**
+ * One network attempt of the stream. Returns this attempt's accumulated text +
+ * a `truncated` flag telling the caller whether the reply was cut short.
+ */
+async function streamOnce({ endpoint, mode, message, model, signal, idleMs, onStatus, onDelta, onReasoning }) {
   const body = mode === "search" ? { query: message } : mode === "merge" ? { message } : { message, model };
 
   // Idle-timeout: nếu KHÔNG nhận được byte nào trong idleMs (kết nối treo), tự
@@ -65,6 +114,8 @@ export async function stream({ mode = "chat", message, model, signal, onDelta, o
 
   let text = "";
   let reasoning = "";
+  let sawDone = false; // thấy {done} = stream kết thúc tử tế (không bị cắt)
+  let truncated = false; // gateway báo upstream bị cắt giữa chừng (Vercel 300s)
 
   // Một dòng SSE → cập nhật text/reasoning. Tách ra để dùng lại khi flush dòng cuối.
   const processLine = (rawLine) => {
@@ -88,6 +139,8 @@ export async function stream({ mode = "chat", message, model, signal, onDelta, o
       onReasoning?.(p.reasoning);
       if (p.answer) text = p.answer;
     }
+    if (p.truncated) truncated = true;
+    if (p.done) sawDone = true;
     if (p.error) throw new ApiError(p.error, {});
   };
 
@@ -118,9 +171,17 @@ export async function stream({ mode = "chat", message, model, signal, onDelta, o
     buf += decoder.decode(); // flush decoder
     if (buf.trim()) processLine(buf); // dòng SSE cuối không có '\n' — đừng bỏ sót
 
-    return { text: text.trim(), reasoning: reasoning.trim() };
+    // Chat: gateway gửi {done} khi xong sạch. Stream EOF mà chưa thấy {done} dù
+    // đã có chữ → kết nối/edge rớt giữa chừng → coi như bị cắt để nối tiếp.
+    if (mode === "chat" && !sawDone && text) truncated = true;
+
+    return { text, reasoning, truncated };
   } catch (err) {
+    if (signal?.aborted) throw err; // người dùng bấm Ctrl+C → huỷ thật, không nối tiếp
     if (timedOut) throw new ApiError("Kết nối tới máy chủ quá thời gian chờ (treo).", { code: "timeout" });
+    // Rớt mạng giữa chừng (không phải huỷ, không phải treo): với chat, nếu đã có
+    // chữ thì trả phần đã nhận + cờ truncated để lớp trên nối tiếp.
+    if (mode === "chat" && text) return { text, reasoning, truncated: true };
     throw err;
   } finally {
     clearTimeout(idle);

package/src/repl.js

@@ -282,7 +282,7 @@ export async function startRepl(opts = {}) {
     console.log("");
   }
   async function pickSession() {
-    const items = sessions.list(20);
+    const items = sessions.list(20, process.cwd()); // chỉ phiên của workspace hiện tại
     if (!items.length) {
       console.log(c.dim("  " + t.sessionEmpty) + "\n");
       return null;
@@ -307,7 +307,7 @@ export async function startRepl(opts = {}) {
     return full;
   }
   function listSessions() {
-    const items = sessions.list(20);
+    const items = sessions.list(20, process.cwd()); // chỉ phiên của workspace hiện tại
     if (!items.length) return console.log(c.dim("  " + t.sessionEmpty));
     console.log("\n" + chalk.bold("  " + t.sessionListTitle));
     items.forEach((s) =>
@@ -457,7 +457,7 @@ Tự đánh giá: còn THIẾU gì để đạt mục tiêu? Chọn 1 nhiệm v
 
   // Khôi phục phiên theo cờ dòng lệnh, hoặc mở phiên mới.
   if (opts.continue) {
-    const s = sessions.latest();
+    const s = sessions.latest(process.cwd()); // phiên gần nhất CỦA workspace này
     if (s) await restore(s);
     else {
       startFresh();
@@ -663,18 +663,28 @@ Tự đánh giá: còn THIẾU gì để đạt mục tiêu? Chọn 1 nhiệm v
   // "tự tắt" trước đây) — thì KHÔNG coi là từ chối: xếp nó vào hàng đợi tin
   // nhắn rồi HỎI LẠI. Nhờ vậy không thao tác nào bị quyết định bởi rác.
   async function askPermission(name) {
-    while (true) {
-      const raw = await ask(c.tool("  cho phép? ") + c.dim("[y] có / [n] không / [a] luôn " + name + " › "));
-      if (raw == null) return "n"; // stdin đóng thật
-      const a = raw.trim().toLowerCase();
-      if (a === "" || a === "y" || a === "yes" || a === "có") return "y";
-      if (a === "n" || a === "no" || a === "không") return "n";
-      if (a === "a" || a === "always" || a === "luôn") return "a";
-      if (raw.trim().length > 3) {
-        pending.push(raw); // tin nhắn lạc → xếp hàng, gửi sau khi xong lượt
-        console.log(c.dim("  " + t.queued(pending.length, truncate(raw, 60))));
+    // Tắt spinner "đang chạy" trong lúc chờ duyệt. Nếu để nguyên, thanh dưới hiện
+    // "đang chạy · Ctrl+C để dừng" → người dùng tưởng đang bận, không biết phải gõ
+    // y/n nên lượt TREO. Báo bằng 1 dòng cố định (vào scrollback, không bị vẽ đè)
+    // + bỏ spinner để prompt nổi bật. finally khôi phục trạng thái chạy.
+    tui.setBusy(false);
+    console.log(c.tool("  ⏸ Cần quyền: " + name) + c.dim(" — gõ y (đồng ý) / n (từ chối) / a (luôn cho phép)"));
+    try {
+      while (true) {
+        const raw = await ask(c.tool("  cho phép? ") + c.dim("[y] có / [n] không / [a] luôn " + name + " › "));
+        if (raw == null) return "n"; // stdin đóng thật
+        const a = raw.trim().toLowerCase();
+        if (a === "" || a === "y" || a === "yes" || a === "có") return "y";
+        if (a === "n" || a === "no" || a === "không") return "n";
+        if (a === "a" || a === "always" || a === "luôn") return "a";
+        if (raw.trim().length > 3) {
+          pending.push(raw); // tin nhắn lạc → xếp hàng, gửi sau khi xong lượt
+          console.log(c.dim("  " + t.queued(pending.length, truncate(raw, 60))));
+        }
+        console.log(c.dim("  " + t.permRetry));
       }
-      console.log(c.dim("  " + t.permRetry));
+    } finally {
+      tui.setBusy(true, t.thinking); // khôi phục "đang chạy" cho phần còn lại của lượt
     }
   }
 
@@ -926,7 +936,7 @@ function printError(err) {
 }
 
 function printUsage(u) {
-  const planName = { pro: "Pro", proplus: "Pro+", admin: "Admin", trial: "Trial" }[u.plan] || u.plan;
+  const planName = { pro: "Pro", proplus: "Pro+", ultra: "Ultra", admin: "Admin", trial: "Trial" }[u.plan] || u.plan;
   const lines = [
     chalk.bold(t.usageTitle),
     `  ${t.plan}: ${chalk.bold(planName)}    ${t.status}: ${u.status === "active" ? c.ok(u.status) : c.err(u.status)}`,

package/src/sessions.js

@@ -63,9 +63,20 @@ export function load(id) {
   }
 }
 
-/** Tóm tắt nhẹ (không tải toàn bộ history vào view), sắp xếp mới nhất trước. */
-export function list(limit = 30) {
+// Chuẩn hoá thư mục workspace để so khớp: resolve + lowercase trên Windows nên
+// "D:\x", "D:\x\" và "d:\x" coi là một.
+const normDir = (p) => {
+  const r = path.resolve(p || "");
+  return process.platform === "win32" ? r.toLowerCase() : r;
+};
+
+/**
+ * Tóm tắt nhẹ (không tải toàn bộ history vào view), sắp xếp mới nhất trước.
+ * cwd != null → chỉ trả phiên thuộc đúng thư mục workspace đó (resume theo dự án).
+ */
+export function list(limit = 30, cwd = null) {
   ensure();
+  const want = cwd != null ? normDir(cwd) : null;
   let files;
   try {
     files = fs.readdirSync(DIR).filter((f) => f.endsWith(".json"));
@@ -76,6 +87,7 @@ export function list(limit = 30) {
   for (const f of files) {
     try {
       const s = JSON.parse(fs.readFileSync(path.join(DIR, f), "utf8"));
+      if (want != null && normDir(s.cwd) !== want) continue; // khác workspace → bỏ
       out.push({
         id: s.id,
         updatedAt: s.updatedAt || s.createdAt || 0,
@@ -92,8 +104,8 @@ export function list(limit = 30) {
   return out.slice(0, limit);
 }
 
-export function latest() {
-  const l = list(1);
+export function latest(cwd = null) {
+  const l = list(1, cwd);
   return l.length ? load(l[0].id) : null;
 }