@lzhzzzzwill/cofos 1.2.3 → 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" } },
   });
 
@@ -213,6 +214,7 @@ function createTui() {
   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;
@@ -225,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();
   }
 
@@ -314,6 +316,10 @@ function createTui() {
   });
 
   screen.key(["escape"], hideSuggestions);
+  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();
@@ -324,6 +330,7 @@ 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(); },
@@ -432,6 +439,7 @@ function main() {
   proc.stderr.setEncoding("utf-8");
 
   let buf = "", ready = false, generating = false, responseOpen = false, commandOpen = false, closed = false;
+  let responseMaybeReasoning = false, responseReasoning = false, responseBuffer = "";
   const inputQ = [];
   let mlBuf = [];
 
@@ -476,7 +484,12 @@ 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;
@@ -495,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 + "● " + R + input);
+    ui.print("{yellow-fg}●{/yellow-fg} " + input);
     responseOpen = false;
     generating = true;
     proc.stdin.write(input + "\n");
@@ -510,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) {
@@ -528,7 +578,7 @@ function main() {
       const di = buf.indexOf(DONE);
       if (di === -1) return;
       const pre = buf.slice(0, di).trim();
-      if (pre) ui.print(G + "• " + R + pre);
+      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;
@@ -548,15 +598,19 @@ function main() {
       buf = buf.slice(ri + RESPONSE.length);
       if (buf[0] === "\n") buf = buf.slice(1);
       responseOpen = true;
-      ui.append(GR + B + "◆ COFOS " + R);
+      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");
@@ -566,7 +620,7 @@ function main() {
       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.3",
+  "version": "1.2.4",
   "description": "COFOS 9B + RAG chat CLI with optional local PDF ingestion.",
   "type": "module",
   "bin": {