@lzhzzzzwill/cofos 1.3.3 → 1.3.6

package/README.md

@@ -6,7 +6,7 @@ COFOS CLI is a standalone Node.js command line wrapper around the COFOS Python r
 npx @lzhzzzzwill/cofos
 ```
 
-By default COFOS uses a plain terminal interface, so normal terminal scrollback, mouse selection, copy, and `Ctrl+C` work as expected. The fixed-input full-screen interface is still available as an experimental mode with `--tui`.
+COFOS uses a plain terminal interface, so normal terminal scrollback, mouse selection, copy, and `Ctrl+C` work as expected.
 
 ## Runtime Model
 
@@ -71,18 +71,19 @@ Disable RAG for a session:
 npx @lzhzzzzwill/cofos --no-rag
 ```
 
-Ask once and exit, useful for scripts and smoke tests:
+Show model reasoning scaffold when emitted:
 
 ```bash
-npx @lzhzzzzwill/cofos -p "What ROS does TT-T-COF generate under visible-light photocatalysis?"
+npx @lzhzzzzwill/cofos --show-reasoning
 ```
 
-Use the experimental fixed-input TUI:
+Ask once and exit, useful for scripts and smoke tests:
 
 ```bash
-npx @lzhzzzzwill/cofos --tui
+npx @lzhzzzzwill/cofos -p "What ROS does TT-T-COF generate under visible-light photocatalysis?"
 ```
 
+
 ## Startup Behavior
 
 On startup, the CLI does the following:
@@ -133,7 +134,7 @@ cofos \
 | `--rebuild-pdf-index` | Force PDF parsing and BM25 rebuild |
 | `--top-k` | Retrieval top-k |
 | `-p`, `--prompt` | Ask one question and exit |
-| `--tui` | Use the experimental fixed-input full-screen TUI |
+| `--show-reasoning` | Show emitted reasoning scaffold in dim text before the final answer |
 | `--max-new-tokens` | Override generation length |
 | `--no-rag` | Run without KG/BM25 retrieval |
 

package/bin/cofos.js

@@ -158,208 +158,7 @@ function ensureDeps(py, log = console.log) {
   if (r.status !== 0) { log("[cofos] Failed to install Python dependencies."); process.exit(1); }
 }
 
-/* ─── TUI ─── */
-
-async function createTui() {
-  const { default: blessed } = await import("blessed");
-  const screen = blessed.screen({
-    smartCSR: true,
-    fullUnicode: true,
-    title: "COFOS",
-    mouse: false,
-  });
-  screen.program.disableMouse();
-
-  const history = blessed.box({
-    top: 0,
-    left: 0,
-    width: "100%",
-    height: "100%-3",
-    scrollable: true,
-    alwaysScroll: true,
-    keys: true,
-    vi: true,
-    mouse: false,
-    tags: true,
-    scrollbar: { ch: " ", track: { bg: "black" }, style: { bg: "cyan" } },
-  });
-
-  const suggestions = blessed.box({
-    bottom: 3,
-    left: 2,
-    width: "70%",
-    height: 6,
-    hidden: true,
-    border: { type: "line" },
-    tags: true,
-    style: { border: { fg: "yellow" }, fg: "white", bg: "black" },
-  });
-
-  const input = blessed.textbox({
-    bottom: 0,
-    left: 0,
-    width: "100%",
-    height: 3,
-    inputOnFocus: true,
-    keys: true,
-    mouse: false,
-    border: { type: "line" },
-    label: " cofos ",
-    style: {
-      border: { fg: "cyan" },
-      focus: { border: { fg: "green" } },
-    },
-  });
-
-  screen.append(history);
-  screen.append(suggestions);
-  screen.append(input);
-
-  let buffer = "";
-  let onSubmit = () => {};
-  let onExit = () => {};
-  let submitting = false;
-  let suggestionMatches = [];
-  let selectedSuggestion = 0;
-
-  function renderBuffer(wasAtBottom = true) {
-    history.setContent(buffer);
-    if (wasAtBottom) history.setScrollPerc(100);
-    screen.render();
-  }
-
-  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;
-      if (data.includes("\r")) {
-        const lastNewline = buffer.lastIndexOf("\n");
-        buffer = (lastNewline === -1 ? "" : buffer.slice(0, lastNewline + 1)) + part;
-      } else {
-        buffer += part;
-      }
-    }
-    if (buffer.length > 250000) buffer = buffer.slice(-220000);
-    renderBuffer(wasAtBottom);
-  }
-
-  function hideSuggestions() {
-    suggestionMatches = [];
-    selectedSuggestion = 0;
-    if (!suggestions.hidden) {
-      suggestions.hide();
-      screen.render();
-    }
-  }
-
-  function renderSuggestions() {
-    if (!suggestionMatches.length) {
-      hideSuggestions();
-      return;
-    }
-    selectedSuggestion = Math.max(0, Math.min(selectedSuggestion, suggestionMatches.length - 1));
-    suggestions.setContent(suggestionMatches.map(({ cmd, desc }, i) => {
-      const marker = i === selectedSuggestion ? "{black-fg}{yellow-bg}›" : " ";
-      const close = i === selectedSuggestion ? "{/yellow-bg}{/black-fg}" : "";
-      return `${marker} {cyan-fg}${cmd}{/cyan-fg}  ${desc}${close}`;
-    }).join("\n"));
-    suggestions.height = Math.min(8, suggestionMatches.length + 2);
-    suggestions.show();
-    screen.render();
-  }
-
-  function updateSuggestions() {
-    const value = input.getValue() || "";
-    if (!value.startsWith("/")) {
-      hideSuggestions();
-      return;
-    }
-    suggestionMatches = slashMatches(value);
-    selectedSuggestion = 0;
-    renderSuggestions();
-  }
-
-  function selectedCommand() {
-    return suggestionMatches[selectedSuggestion]?.cmd || "";
-  }
-
-  function submitInput() {
-    if (submitting) return;
-    submitting = true;
-    let value = input.getValue();
-    if (!suggestions.hidden && selectedCommand()) value = selectedCommand();
-    hideSuggestions();
-    input.clearValue();
-    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;
-    setTimeout(updateSuggestions, 0);
-  });
-
-  screen.key(["tab"], () => {
-    const value = input.getValue() || "";
-    if (!value.startsWith("/")) return;
-    if (!suggestionMatches.length) suggestionMatches = slashMatches(value);
-    if (!suggestionMatches.length) return;
-    input.setValue(selectedCommand());
-    hideSuggestions();
-    input.focus();
-    screen.render();
-  });
-
-  screen.key(["up"], () => {
-    if (suggestions.hidden || !suggestionMatches.length) return;
-    selectedSuggestion = (selectedSuggestion - 1 + suggestionMatches.length) % suggestionMatches.length;
-    renderSuggestions();
-  });
-
-  screen.key(["down"], () => {
-    if (suggestions.hidden || !suggestionMatches.length) return;
-    selectedSuggestion = (selectedSuggestion + 1) % suggestionMatches.length;
-    renderSuggestions();
-  });
-
-  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();
-  screen.render();
-
-  return {
-    supportsReplace: true,
-    echoUser: true,
-    screen,
-    input,
-    append,
-    print: (text = "") => append(String(text) + "\n"),
-    clear: () => { buffer = ""; renderBuffer(true); },
-    length: () => buffer.length,
-    replaceFrom: (index, text = "") => { buffer = buffer.slice(0, Math.max(0, index)) + String(text); renderBuffer(true); },
-    setSubmitHandler: (fn) => { onSubmit = fn; },
-    setExitHandler: (fn) => { onExit = fn; },
-    focus: () => { input.focus(); screen.render(); },
-    setPrompt: (label) => { input.setLabel(` ${label} `); screen.render(); },
-    destroy: () => {
-      try { screen.program.disableMouse(); } catch (_) {}
-      screen.destroy();
-    },
-  };
-}
-
-function blessedToAnsi(text) {
+function markupToAnsi(text) {
   return String(text)
     .replace(/\{bold\}/g, B)
     .replace(/\{\/bold\}/g, R)
@@ -399,11 +198,11 @@ function createPlainUi() {
   return {
     supportsReplace: false,
     echoUser: false,
-    append: (text = "") => process.stdout.write(blessedToAnsi(text)),
-    print: (text = "") => console.log(blessedToAnsi(text)),
+    append: (text = "") => process.stdout.write(markupToAnsi(text)),
+    print: (text = "") => console.log(markupToAnsi(text)),
     clear: () => console.clear(),
     length: () => 0,
-    replaceFrom: (_index, text = "") => process.stdout.write(blessedToAnsi(text)),
+    replaceFrom: (_index, text = "") => process.stdout.write(markupToAnsi(text)),
     setSubmitHandler: (fn) => { onSubmit = fn; },
     setExitHandler: (fn) => { onExit = fn; },
     focus: () => rl.prompt(),
@@ -417,7 +216,7 @@ function createPlainUi() {
 async function main() {
   if (process.argv.includes("-h") || process.argv.includes("--help")) {
     console.log("\n  COFOS CLI — 9B + RAG chat (Willlzh/COFOS)\n");
-    console.log("  Usage:  cofos [--tui] [-p QUESTION] [--model PATH] [--config PATH] [--pdf-dir DIR] [--cache-dir DIR] [--top-k N] [--no-rag]\n");
+    console.log("  Usage:  cofos [-p QUESTION] [--show-reasoning] [--model PATH] [--config PATH] [--pdf-dir DIR] [--cache-dir DIR] [--top-k N] [--no-rag]\n");
     console.log("  Commands in chat:");
     console.log("    /help        Show this help");
     console.log("    /exit        Quit");
@@ -430,7 +229,7 @@ async function main() {
     console.log("    --pdf-dir DIR          Add PDFs from DIR as extra BM25 evidence");
     console.log("    --cache-dir DIR        Store model/data cache here (default: ./.cofos)");
     console.log("    --max-new-tokens N     Override generation length");
-    console.log("    --tui                  Use the experimental fixed-input TUI");
+    console.log("    --show-reasoning       Show model reasoning scaffold in dim text when emitted");
     console.log("    -p, --prompt TEXT      Ask once and exit");
     console.log("    --rebuild-pdf-index    Force reparsing PDFs and rebuilding BM25\n");
     console.log("  Multi-line: end a line with \\ to continue on the next line.\n");
@@ -447,7 +246,7 @@ async function main() {
   let maxNewTokens = null;
   let cacheDir = process.env.COFOS_CACHE_DIR ? resolve(process.env.COFOS_CACHE_DIR) : resolve(process.cwd(), ".cofos");
   let rebuildPdfIndex = false;
-  let useTui = false;
+  let showReasoning = false;
   let singlePrompt = null;
   const backendArgs = [chatScript];
 
@@ -479,8 +278,8 @@ async function main() {
       cacheDir = resolve(value(++i, arg));
     } else if (arg === "--rebuild-pdf-index") {
       rebuildPdfIndex = true;
-    } else if (arg === "--tui") {
-      useTui = true;
+    } else if (arg === "--show-reasoning") {
+      showReasoning = true;
     } else if (arg === "-p" || arg === "--prompt") {
       singlePrompt = value(++i, arg);
     } else if (arg === "--no-rag") {
@@ -498,8 +297,8 @@ async function main() {
   if (!py) die("Python not found. Install Python 3.9+ and re-run.");
   if (!existsSync(chatScript)) die(`Missing script: ${chatScript}`);
 
-  const ui = useTui ? await createTui() : createPlainUi();
-  ui.print(useTui ? plain(render({ modelPath, cacheDir, pdfDir })) : render({ modelPath, cacheDir, pdfDir }));
+  const ui = createPlainUi();
+  ui.print(render({ modelPath, cacheDir, pdfDir }));
   ensureDeps(py, (msg) => ui.print(msg));
   ui.print("  Loading COFOS 9B + RAG backend from Hugging Face … first run may take a while");
   if (pdfDir) ui.print("  Local PDF RAG: enabled for this session");
@@ -516,6 +315,9 @@ async function main() {
     HF_HUB_ETAG_TIMEOUT: process.env.HF_HUB_ETAG_TIMEOUT || "30",
     HF_HUB_VERBOSITY: process.env.HF_HUB_VERBOSITY || "error",
     TRANSFORMERS_VERBOSITY: process.env.TRANSFORMERS_VERBOSITY || "error",
+    PYTHONUNBUFFERED: process.env.PYTHONUNBUFFERED || "1",
+    PYTHONIOENCODING: process.env.PYTHONIOENCODING || "utf-8",
+    COFOS_SHOW_REASONING: showReasoning ? "1" : (process.env.COFOS_SHOW_REASONING || "0"),
   };
   const proc = spawn(py, backendArgs, {
     stdio: ["pipe", "pipe", "pipe"],
@@ -551,6 +353,33 @@ async function main() {
     }
   }
 
+  function handleLocalOnly(rawInput) {
+    const input = String(rawInput || "").trim();
+    if (!input) return true;
+    if (input === "/") {
+      ui.print("Commands: " + SLASH_COMMANDS.map(({ cmd }) => cmd).join("  ") + "\nType a prefix such as /r or /to, then press Tab for completion.");
+      return true;
+    }
+    if (/^\/(exit|quit)$/.test(input)) { shutdown(); return true; }
+    if (/^\/(help)$/.test(input)) {
+      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 true;
+    }
+    if (/^\/(cls)$/.test(input)) {
+      ui.clear();
+      return true;
+    }
+    if (/^\/(info)$/.test(input)) {
+      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  Status: ${ready ? "ready" : "loading"}\n  Hist:   ~/.cofos_history.jsonl\n`);
+      return true;
+    }
+    if (input.startsWith("/") && !/^\/(clear|save|topk)(?:\s|$)/.test(input) && !/^\/rag\s+(?:on|off)$/.test(input)) {
+      ui.print(`{red-fg}•{/red-fg} Unknown command: ${input}.\n${slashPrefixHint(input)}`);
+      return true;
+    }
+    return false;
+  }
+
   function handle(rawInput) {
     let input = String(rawInput || "");
     if (input.endsWith("\\")) {
@@ -569,7 +398,7 @@ async function main() {
     if (!input) return false;
 
     if (input === "/") {
-      ui.print("Type a command prefix, for example /r or /to, then press Tab. Run /help for all commands.");
+      ui.print("Commands: " + SLASH_COMMANDS.map(({ cmd }) => cmd).join("  ") + "\nType a prefix such as /r or /to, then press Tab for completion.");
       return false;
     }
     if (/^\/(exit|quit)$/.test(input)) { shutdown(); return true; }
@@ -632,8 +461,14 @@ async function main() {
   }
 
   ui.setSubmitHandler((line) => {
+    if (handleLocalOnly(line)) {
+      if (!closed) ui.focus();
+      return;
+    }
     if (!ready || generating) {
       inputQ.push(line);
+      ui.print(ready ? "• Queued until the current answer finishes." : "• Queued until the model is ready. Use /exit to quit while loading.");
+      if (!closed) ui.focus();
       return;
     }
     const started = handle(line);
@@ -681,6 +516,16 @@ async function main() {
 
   function streamPlainAssistant(text, final = false) {
     if (ui.supportsReplace) return;
+    if (!showReasoning) {
+      if (plainAssistantMode === "undecided") {
+        if (!text && !final) return;
+        plainAssistantMode = "answer";
+        ui.append(`${GR}${B}◆${R} `);
+      }
+      if (text) ui.append(text);
+      return;
+    }
+
     plainAssistantPending += text;
     if (plainAssistantMode === "undecided") {
       const start = plainAssistantPending.trimStart();
@@ -691,12 +536,10 @@ async function main() {
         plainReasonPrinted = 0;
         plainAssistantPending = "";
         ui.append(`${G}-${R} ${G}`);
-      } else if (final || start.length >= 12 || /\s/.test(start)) {
+      } else {
         plainAssistantMode = "answer";
         plainAssistantPending = start;
         ui.append(`${GR}${B}◆${R} `);
-      } else {
-        return;
       }
     }
 
@@ -731,15 +574,14 @@ async function main() {
       }
     }
 
-    if (plainAssistantMode === "answer") {
-      if (plainAssistantPending) {
-        ui.append(plainAssistantPending);
-        plainAssistantPending = "";
-      }
+    if (plainAssistantMode === "answer" && plainAssistantPending) {
+      ui.append(plainAssistantPending);
+      plainAssistantPending = "";
     }
   }
 
 
+
   proc.stdout.on("data", (chunk) => {
     buf += chunk;
     if (!ready) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lzhzzzzwill/cofos",
-  "version": "1.3.3",
+  "version": "1.3.6",
   "description": "COFOS 9B + RAG chat CLI with optional local PDF ingestion.",
   "type": "module",
   "bin": {
@@ -38,8 +38,5 @@
   "license": "MIT",
   "engines": {
     "node": ">=18"
-  },
-  "dependencies": {
-    "blessed": "^0.1.81"
   }
 }

package/runtime/src/inference/student_adapter_inference.py

@@ -489,6 +489,11 @@ class StudentAdapterInference:
     @staticmethod
     def _sanitize_stream(chunks: Any) -> Generator[str, None, None]:
         """Drop ``<think>`` blocks and leading labels from a text stream."""
+        if os.environ.get("COFOS_SHOW_REASONING", "").lower() in {"1", "true", "yes", "on"}:
+            for chunk in chunks:
+                yield chunk
+            return
+
         buffer = ""
         started = False
         in_think = False
@@ -538,7 +543,23 @@ class StudentAdapterInference:
                 )
                 if not stripped.strip():
                     continue
-                buffer = stripped.lstrip()
+                stripped = stripped.lstrip()
+                if awaiting_initial_answer and re.match(
+                    r"(?is)^(?:the user|okay[,，]?\s+the user|用户问|用户用|i need|i should|i'll|let me|since the user|we need)\b",
+                    stripped,
+                ):
+                    paragraph_end = re.search(r"\n\s*\n", stripped)
+                    if paragraph_end is None:
+                        # Keep waiting for the end of the leading reasoning paragraph.
+                        buffer = stripped[-4000:]
+                        continue
+                    buffer = stripped[paragraph_end.end():].lstrip()
+                    awaiting_initial_answer = False
+                    if not buffer.strip():
+                        continue
+                else:
+                    buffer = stripped
+                    awaiting_initial_answer = False
                 started = True
 
             if len(buffer) > len("<think>"):