@lzhzzzzwill/cofos 1.2.2 → 1.2.4

package/bin/cofos.js

@@ -106,7 +106,8 @@ function render({ modelPath, cacheDir, pdfDir }) {
 const SLASH_COMMANDS = [
   { cmd: "/help", desc: "Show help" },
   { cmd: "/exit", desc: "Quit" },
-  { cmd: "/clear", desc: "Clear conversation history" },
+  { cmd: "/clear", desc: "Clear page and conversation history" },
+  { cmd: "/cls", desc: "Clear page only" },
   { cmd: "/save", desc: "Save this session history" },
   { cmd: "/info", desc: "Show model and RAG info" },
   { cmd: "/topk 5", desc: "Set retrieval top-k" },
@@ -169,8 +170,8 @@ function createTui() {
     alwaysScroll: true,
     keys: true,
     vi: true,
-    mouse: false,
-    tags: false,
+    mouse: true,
+    tags: true,
     scrollbar: { ch: " ", track: { bg: "black" }, style: { bg: "cyan" } },
   });
 
@@ -207,10 +208,13 @@ function createTui() {
 
   let buffer = "";
   let onSubmit = () => {};
+  let onExit = () => {};
+  let submitting = false;
   let suggestionMatches = [];
   let selectedSuggestion = 0;
 
   function append(text = "") {
+    const wasAtBottom = history.getScrollPerc() >= 98;
     const data = String(text).replace(/\x1b\[[0-9;]*[A-Za-z]/g, "");
     for (const part of data.split("\r")) {
       if (part === "") continue;
@@ -223,7 +227,7 @@ function createTui() {
     }
     if (buffer.length > 250000) buffer = buffer.slice(-220000);
     history.setContent(buffer);
-    history.setScrollPerc(100);
+    if (wasAtBottom) history.setScrollPerc(100);
     screen.render();
   }
 
@@ -268,6 +272,8 @@ function createTui() {
   }
 
   function submitInput() {
+    if (submitting) return;
+    submitting = true;
     let value = input.getValue();
     if (!suggestions.hidden && selectedCommand()) value = selectedCommand();
     hideSuggestions();
@@ -275,9 +281,11 @@ function createTui() {
     input.focus();
     screen.render();
     onSubmit(String(value || ""));
+    setTimeout(() => { submitting = false; }, 0);
   }
 
   input.key(["enter"], submitInput);
+  screen.key(["enter"], submitInput);
 
   input.on("keypress", (_ch, key = {}) => {
     if (["up", "down", "tab", "enter", "escape"].includes(key.name)) return;
@@ -308,7 +316,11 @@ function createTui() {
   });
 
   screen.key(["escape"], hideSuggestions);
-  screen.key(["C-c"], () => process.emit("SIGINT"));
+  screen.key(["pageup"], () => { history.scroll(-Math.max(3, Math.floor(screen.height * 0.8))); screen.render(); });
+  screen.key(["pagedown"], () => { history.scroll(Math.max(3, Math.floor(screen.height * 0.8))); screen.render(); });
+  screen.key(["home"], () => { history.setScrollPerc(0); screen.render(); });
+  screen.key(["end"], () => { history.setScrollPerc(100); screen.render(); });
+  screen.key(["C-c"], () => onExit());
 
   input.focus();
   screen.render();
@@ -318,7 +330,10 @@ function createTui() {
     input,
     append,
     print: (text = "") => append(String(text) + "\n"),
+    clear: () => { buffer = ""; history.setContent(""); history.setScrollPerc(100); screen.render(); },
     setSubmitHandler: (fn) => { onSubmit = fn; },
+    setExitHandler: (fn) => { onExit = fn; },
+    focus: () => { input.focus(); screen.render(); },
     setPrompt: (label) => { input.setLabel(` ${label} `); screen.render(); },
     destroy: () => screen.destroy(),
   };
@@ -423,7 +438,8 @@ function main() {
   proc.stdout.setEncoding("utf-8");
   proc.stderr.setEncoding("utf-8");
 
-  let buf = "", ready = false, generating = false, responseOpen = false, closed = false;
+  let buf = "", ready = false, generating = false, responseOpen = false, commandOpen = false, closed = false;
+  let responseMaybeReasoning = false, responseReasoning = false, responseBuffer = "";
   const inputQ = [];
   let mlBuf = [];
 
@@ -435,6 +451,7 @@ function main() {
     ui.destroy();
   }
 
+  ui.setExitHandler(shutdown);
   process.on("SIGTERM", shutdown);
   process.on("SIGINT", shutdown);
 
@@ -467,9 +484,22 @@ function main() {
       ui.print(slashHint("/") + "\n  Add local papers to RAG:\n    cofos --pdf-dir ./new_pdfs\n    cofos --pdf-dir ./new_pdfs --rebuild-pdf-index\n\n  Multi-line: end a line with \\ to continue.");
       return false;
     }
+    if (/^\/(cls)$/.test(input)) {
+      ui.clear();
+      return false;
+    }
     if (/^\/(clear)$/.test(input)) {
+      ui.clear();
       mlBuf = [];
       responseOpen = false;
+      commandOpen = true;
+      proc.stdin.write(input + "\n");
+      generating = true;
+      return true;
+    }
+    if (/^\/(topk)(?:\s+\d+)?$/.test(input) || /^\/rag\s+(?:on|off)$/.test(input)) {
+      responseOpen = false;
+      commandOpen = true;
       proc.stdin.write(input + "\n");
       generating = true;
       return true;
@@ -478,8 +508,12 @@ function main() {
       ui.print(`\n  COFOS CLI — v${pkg.version}\n  Model:  ${modelPath} (Hugging Face)\n  RAG:    Willlzh/COFOS_data/runtime (KG + BM25) + optional --pdf-dir BM25 evidence\n  Cache:  ${cacheDir}\n  HF:     ${backendEnv.HF_HOME}\n  Engine: StudentAdapterInference + transformers + PyTorch\n  Hist:   ~/.cofos_history.jsonl\n`);
       return false;
     }
+    if (input.startsWith("/")) {
+      ui.print(`{red-fg}•{/red-fg} Unknown command: ${input}. Press / then Tab for commands, or run /help.`);
+      return false;
+    }
 
-    ui.print(Y + "● You" + R + "\n" + input);
+    ui.print("{yellow-fg}●{/yellow-fg} " + input);
     responseOpen = false;
     generating = true;
     proc.stdin.write(input + "\n");
@@ -493,6 +527,39 @@ function main() {
 
   proc.stderr.on("data", (chunk) => ui.append(chunk));
 
+  function appendAssistant(text) {
+    if (!text) return;
+    if (responseMaybeReasoning) {
+      responseBuffer += text;
+      const enough = responseBuffer.length > 80 || responseBuffer.includes("\n\n") || responseBuffer.includes("。") || responseBuffer.includes(".");
+      if (!enough) return;
+      const start = responseBuffer.trimStart();
+      responseReasoning = /^(The user|Okay[,，]|I need|We need|Based on the provided information[,，]? I can|用户)/i.test(start);
+      responseMaybeReasoning = false;
+      if (responseReasoning) {
+        ui.append("{gray-fg}- " + responseBuffer + "{/gray-fg}");
+      } else {
+        ui.append(responseBuffer);
+      }
+      responseBuffer = "";
+      return;
+    }
+    if (responseReasoning) {
+      const end = text.indexOf("\n\n");
+      if (end !== -1) {
+        const reason = text.slice(0, end);
+        const answer = text.slice(end).replace(/^\s+/, "");
+        if (reason) ui.append("{gray-fg}" + reason + "{/gray-fg}");
+        responseReasoning = false;
+        if (answer) ui.append("\n{green-fg}{bold}◆{/bold}{/green-fg} " + answer);
+      } else {
+        ui.append("{gray-fg}" + text + "{/gray-fg}");
+      }
+      return;
+    }
+    ui.append(text);
+  }
+
   proc.stdout.on("data", (chunk) => {
     buf += chunk;
     if (!ready) {
@@ -507,6 +574,20 @@ function main() {
     }
     if (!generating) { buf = ""; return; }
 
+    if (commandOpen) {
+      const di = buf.indexOf(DONE);
+      if (di === -1) return;
+      const pre = buf.slice(0, di).trim();
+      if (pre) ui.print("{gray-fg}•{/gray-fg} " + pre);
+      buf = buf.slice(di + DONE.length);
+      if (buf[0] === "\n") buf = buf.slice(1);
+      generating = false;
+      commandOpen = false;
+      ui.focus();
+      drain();
+      return;
+    }
+
     if (!responseOpen) {
       const ri = buf.indexOf(RESPONSE);
       if (ri === -1) {
@@ -517,24 +598,29 @@ function main() {
       buf = buf.slice(ri + RESPONSE.length);
       if (buf[0] === "\n") buf = buf.slice(1);
       responseOpen = true;
-      ui.append(GR + "◆ COFOS" + R + "\n");
+      responseMaybeReasoning = true;
+      responseReasoning = false;
+      responseBuffer = "";
+      ui.append("{green-fg}{bold}◆{/bold}{/green-fg} ");
     }
 
     const di = buf.indexOf(DONE);
     if (di !== -1) {
       const pre = buf.slice(0, di);
-      if (pre) ui.append(pre);
+      if (pre) appendAssistant(pre);
       buf = buf.slice(di + DONE.length);
       if (buf[0] === "\n") buf = buf.slice(1);
+      if (responseBuffer) { appendAssistant("\n"); }
       generating = false;
       responseOpen = false;
       ui.append("\n");
+      ui.focus();
       drain();
     } else {
       const keep = DONE.length - 1;
       if (buf.length > keep) {
         const out = buf.slice(0, -keep);
-        if (out) ui.append(out);
+        if (out) appendAssistant(out);
         buf = buf.slice(-keep);
       }
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lzhzzzzwill/cofos",
-  "version": "1.2.2",
+  "version": "1.2.4",
   "description": "COFOS 9B + RAG chat CLI with optional local PDF ingestion.",
   "type": "module",
   "bin": {