@lzhzzzzwill/cofos 1.2.2 → 1.2.3

package/bin/cofos.js

@@ -207,6 +207,8 @@ function createTui() {
 
   let buffer = "";
   let onSubmit = () => {};
+  let onExit = () => {};
+  let submitting = false;
   let suggestionMatches = [];
   let selectedSuggestion = 0;
 
@@ -268,6 +270,8 @@ function createTui() {
   }
 
   function submitInput() {
+    if (submitting) return;
+    submitting = true;
     let value = input.getValue();
     if (!suggestions.hidden && selectedCommand()) value = selectedCommand();
     hideSuggestions();
@@ -275,9 +279,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 +314,7 @@ function createTui() {
   });
 
   screen.key(["escape"], hideSuggestions);
-  screen.key(["C-c"], () => process.emit("SIGINT"));
+  screen.key(["C-c"], () => onExit());
 
   input.focus();
   screen.render();
@@ -319,6 +325,8 @@ function createTui() {
     append,
     print: (text = "") => append(String(text) + "\n"),
     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 +431,7 @@ 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;
   const inputQ = [];
   let mlBuf = [];
 
@@ -435,6 +443,7 @@ function main() {
     ui.destroy();
   }
 
+  ui.setExitHandler(shutdown);
   process.on("SIGTERM", shutdown);
   process.on("SIGINT", shutdown);
 
@@ -470,6 +479,14 @@ function main() {
     if (/^\/(clear)$/.test(input)) {
       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;
@@ -479,7 +496,7 @@ function main() {
       return false;
     }
 
-    ui.print(Y + "● You" + R + "\n" + input);
+    ui.print(Y + "● " + R + input);
     responseOpen = false;
     generating = true;
     proc.stdin.write(input + "\n");
@@ -507,6 +524,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(G + "• " + R + 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,7 +548,7 @@ function main() {
       buf = buf.slice(ri + RESPONSE.length);
       if (buf[0] === "\n") buf = buf.slice(1);
       responseOpen = true;
-      ui.append(GR + "◆ COFOS" + R + "\n");
+      ui.append(GR + B + "◆ COFOS " + R);
     }
 
     const di = buf.indexOf(DONE);
@@ -529,6 +560,7 @@ function main() {
       generating = false;
       responseOpen = false;
       ui.append("\n");
+      ui.focus();
       drain();
     } else {
       const keep = DONE.length - 1;

package/package.json

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