@krmxd/onegpt 0.2.2-beta → 0.2.3-beta

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@krmxd/onegpt",
-  "version": "0.2.2-beta",
+  "version": "0.2.3-beta",
   "description": "OGPT - AI coding assistant for the terminal with a built-in local engine",
   "main": "src/index.js",
   "bin": {

package/src/agent.js

@@ -10,6 +10,41 @@ const path = require("path");
 const { getConfig } = require("./config");
 const { getSharedProvider } = require("./ollama");
 const { ToolRegistry } = require("./tools");
+const { spawnSync } = require("child_process");
+
+// --- auto syntax check for AI-written code -------------------------------
+const CHECKABLE_EXT = new Set([".py", ".js", ".mjs", ".cjs"]);
+
+function _pick(binaries, args, timeoutMs) {
+  for (const bin of binaries) {
+    try {
+      const r = spawnSync(bin, args, { timeout: timeoutMs });
+      if (r.status === 0) return null;
+      if (r.error && r.error.code === "ENOENT") continue;
+      const err = (r.stderr ? r.stderr.toString() : "")
+        || (r.stdout ? r.stdout.toString() : "") || `exit ${r.status}`;
+      return err.split("\n").filter((l) => l.trim()).slice(0, 4).join("\n").slice(0, 500);
+    } catch { /* try next */ }
+  }
+  return null; // no interpreter available - skip silently
+}
+
+function syntaxCheckError(filePath) {
+  const ext = path.extname(filePath).toLowerCase();
+  if (!CHECKABLE_EXT.has(ext)) return null;
+  if (!fs.existsSync(filePath)) return null;
+  if (ext === ".py") {
+    return _pick(["python3", "python"], ["-m", "py_compile", filePath], 15000);
+  }
+  return _pick([process.execPath || "node", "node"], ["--check", filePath], 15000);
+}
+
+function autoCheckMsg(p, err) {
+  // Loop-break redirect style: tiny models need the exact shape again.
+  return "[auto-check] The file you just wrote FAILED its syntax check:\n" +
+    `${p}\n${err}\n` +
+    "Rewrite the ENTIRE file correctly NOW with write_file (bare JSON tool call), fixing every reported error. Do not explain.";
+}
 
 const TOOL_RE = /\x00TOOL:([^:]*):([^:]*):(.*?)\x00/g;
 const USAGE_RE = /\x00USAGE:(.*?)\x00/;
@@ -570,6 +605,7 @@ class Agent {
     let full = "";
     let rounds = 0;
     let nudges = 0;
+    this._autoChecks = 0;
     let lastSig = null;
     let loopBreaks = 0;
     const doneBits = [];
@@ -633,6 +669,15 @@ class Agent {
             doneBits.push(result.output.split("\n")[0].slice(0, 90));
             const ev = codeEvent(tc, pre);
             if (ev) full += (full.trim() ? "\n\n" : "") + codeMarkdown(ev);
+            // Auto syntax check: broken AI-written code gets one bounce-back.
+            if (ev && (tc.name === "write_file" || tc.name === "edit_file")
+                && this._autoChecks < 2) {
+              const err = syntaxCheckError(ev.path);
+              if (err) {
+                this._autoChecks++;
+                this.history.push({ role: "user", content: autoCheckMsg(ev.path, err), ts: Date.now() / 1000 });
+              }
+            }
           }
         }
       } else {
@@ -682,6 +727,7 @@ class Agent {
     this.history.push({ role: "user", content: userInput, ts: Date.now() / 1000 });
     let rounds = 0;
     let nudges = 0;
+    this._autoChecks = 0;
     const tStart = Date.now();
     let ttft = 0;
     let nTokens = 0;
@@ -790,6 +836,16 @@ class Agent {
             }
             const ev = codeEvent(tc, pre);
             if (ev) yield "\n" + codeMarkdown(ev) + "\n";
+            // Auto syntax check: broken AI-written code gets one bounce-back.
+            if (ev && (tc.name === "write_file" || tc.name === "edit_file")
+                && this._autoChecks < 2) {
+              const err = syntaxCheckError(ev.path);
+              if (err) {
+                this._autoChecks++;
+                yield `\n[auto-check] ${path.basename(ev.path)} has a syntax error - fixing...\n`;
+                this.history.push({ role: "user", content: autoCheckMsg(ev.path, err), ts: Date.now() / 1000 });
+              }
+            }
           }
         }
       } else {
@@ -909,4 +965,5 @@ class Agent {
   }
 }
 
-module.exports = { Agent, extractJsonTools, repairToolJson, wantsAction, repliedWithoutActing, stalledHalfway };
+module.exports = { Agent, extractJsonTools, repairToolJson, wantsAction,
+  repliedWithoutActing, stalledHalfway, syntaxCheckError, autoCheckMsg };

package/src/render.js

@@ -275,12 +275,10 @@ class SmoothPrinter {
 
   _degrade(skipFlush) {
     this.degraded = true;
-    if (!skipFlush && this.buf) {
-      // Hand back the un-styled tail so nothing is lost visually.
-      this.out.write("\n" + this.buf.slice(-4000) + "\n");
-    } else {
-      this.out.write("\n");
-    }
+    // Do NOT reprint the buffer here - part of it is already on screen
+    // and repainting would duplicate it. Just mark the seam and let the
+    // remaining chunks stream raw.
+    this.out.write("\n");
     this.buf = "";
   }
 
@@ -288,7 +286,9 @@ class SmoothPrinter {
     if (this.closed) return;
     this.closed = true;
     if (this.timer) { clearTimeout(this.timer); this.timer = null; }
-    if (!this.tty || this.degraded) return;
+    if (!this.tty) return;
+    // Always un-hide the cursor, degraded or not.
+    if (this.degraded) { this.out.write(`${ESC}[?25h`); return; }
     this._erase();
     const width = this.widthOf();
     this.out.write(renderMarkdown(this.buf, width) + "\n");