@corbat-tech/coco 1.5.0 → 1.6.0

package/dist/cli/index.js

@@ -19,6 +19,7 @@ import { fileURLToPath, URL as URL$1 } from 'url';
 import JSON5 from 'json5';
 import Anthropic from '@anthropic-ai/sdk';
 import OpenAI from 'openai';
+import { jsonrepair } from 'jsonrepair';
 import * as http from 'http';
 import { execFile, exec, execSync, execFileSync, spawn } from 'child_process';
 import { promisify } from 'util';
@@ -981,6 +982,71 @@ var init_allow_path_prompt = __esm({
     init_allowed_paths();
   }
 });
+
+// src/tools/utils/heartbeat.ts
+var heartbeat_exports = {};
+__export(heartbeat_exports, {
+  CommandHeartbeat: () => CommandHeartbeat
+});
+var CommandHeartbeat;
+var init_heartbeat = __esm({
+  "src/tools/utils/heartbeat.ts"() {
+    CommandHeartbeat = class {
+      startTime = 0;
+      lastActivityTime = 0;
+      updateInterval = null;
+      warnThreshold = 30;
+      // seconds
+      updateIntervalSeconds = 10;
+      // seconds
+      callbacks;
+      constructor(callbacks = {}) {
+        this.callbacks = callbacks;
+      }
+      /**
+       * Start monitoring - begins periodic updates and silence warnings
+       */
+      start() {
+        this.startTime = Date.now();
+        this.lastActivityTime = Date.now();
+        this.updateInterval = setInterval(() => {
+          const stats = this.getStats();
+          this.callbacks.onUpdate?.(stats);
+          if (stats.silentSeconds >= this.warnThreshold) {
+            this.callbacks.onWarn?.(`\u26A0\uFE0F  Command silent for ${stats.silentSeconds}s`);
+          }
+        }, this.updateIntervalSeconds * 1e3);
+      }
+      /**
+       * Register activity - call this when command produces output
+       * Resets the silence timer
+       */
+      activity() {
+        this.lastActivityTime = Date.now();
+      }
+      /**
+       * Stop monitoring - clears periodic interval
+       * Should be called in finally{} block to ensure cleanup
+       */
+      stop() {
+        if (this.updateInterval) {
+          clearInterval(this.updateInterval);
+          this.updateInterval = null;
+        }
+      }
+      /**
+       * Get current heartbeat statistics
+       */
+      getStats() {
+        const now = Date.now();
+        return {
+          elapsedSeconds: Math.floor((now - this.startTime) / 1e3),
+          silentSeconds: Math.floor((now - this.lastActivityTime) / 1e3)
+        };
+      }
+    };
+  }
+});
 function isEnvVarSafe(name) {
   if (SAFE_ENV_VARS.has(name)) {
     return true;
@@ -1117,23 +1183,50 @@ Examples:
         }
         const startTime = performance.now();
         const timeoutMs = timeout ?? DEFAULT_TIMEOUT_MS;
+        const { CommandHeartbeat: CommandHeartbeat2 } = await Promise.resolve().then(() => (init_heartbeat(), heartbeat_exports));
+        const heartbeat = new CommandHeartbeat2({
+          onUpdate: (stats) => {
+            if (stats.elapsedSeconds > 10) {
+              process.stderr.write(`\r\u23F1\uFE0F  ${stats.elapsedSeconds}s elapsed`);
+            }
+          },
+          onWarn: (message) => {
+            process.stderr.write(`
+${message}
+`);
+          }
+        });
         try {
+          heartbeat.start();
           const options = {
             cwd: cwd ?? process.cwd(),
             timeout: timeoutMs,
             env: { ...process.env, ...env2 },
             shell: true,
             reject: false,
+            buffer: false,
+            // Enable streaming
             maxBuffer: MAX_OUTPUT_SIZE
           };
-          const result = await execa(command, options);
+          const subprocess = execa(command, options);
+          let stdoutBuffer = "";
+          let stderrBuffer = "";
+          subprocess.stdout?.on("data", (chunk) => {
+            const text11 = chunk.toString();
+            stdoutBuffer += text11;
+            process.stdout.write(text11);
+            heartbeat.activity();
+          });
+          subprocess.stderr?.on("data", (chunk) => {
+            const text11 = chunk.toString();
+            stderrBuffer += text11;
+            process.stderr.write(text11);
+            heartbeat.activity();
+          });
+          const result = await subprocess;
           return {
-            stdout: truncateOutput(
-              typeof result.stdout === "string" ? result.stdout : String(result.stdout ?? "")
-            ),
-            stderr: truncateOutput(
-              typeof result.stderr === "string" ? result.stderr : String(result.stderr ?? "")
-            ),
+            stdout: truncateOutput(stdoutBuffer),
+            stderr: truncateOutput(stderrBuffer),
             exitCode: result.exitCode ?? 0,
             duration: performance.now() - startTime
           };
@@ -1148,6 +1241,9 @@ Examples:
             `Command execution failed: ${error instanceof Error ? error.message : String(error)}`,
             { tool: "bash_exec", cause: error instanceof Error ? error : void 0 }
           );
+        } finally {
+          heartbeat.stop();
+          process.stderr.write("\r                                        \r");
         }
       }
     });
@@ -6191,8 +6287,10 @@ var OpenAIProvider = class {
       const timeoutInterval = setInterval(checkTimeout, 5e3);
       try {
         for await (const chunk of stream) {
-          lastActivityTime = Date.now();
           const delta = chunk.choices[0]?.delta;
+          if (delta?.content || delta?.tool_calls) {
+            lastActivityTime = Date.now();
+          }
           if (delta?.content) {
             yield { type: "text", text: delta.content };
           }
@@ -6238,10 +6336,20 @@ var OpenAIProvider = class {
           let input = {};
           try {
             input = builder.arguments ? JSON.parse(builder.arguments) : {};
-          } catch {
+          } catch (error) {
             console.warn(
-              `[OpenAI] Failed to parse tool call arguments: ${builder.arguments?.slice(0, 100)}`
+              `[OpenAI] Failed to parse tool call arguments for ${builder.name}: ${builder.arguments?.slice(0, 100)}`
             );
+            try {
+              if (builder.arguments) {
+                const repaired = jsonrepair(builder.arguments);
+                input = JSON.parse(repaired);
+                console.log(`[OpenAI] \u2713 Successfully repaired JSON for ${builder.name}`);
+              }
+            } catch (repairError) {
+              console.error(`[OpenAI] Cannot repair JSON for ${builder.name}, using empty object`);
+              console.error(`[OpenAI] Original error:`, error);
+            }
           }
           yield {
             type: "tool_use_end",
@@ -10601,14 +10709,42 @@ function generateToolCatalog(registry) {
 }
 var COCO_SYSTEM_PROMPT = `You are Corbat-Coco, an autonomous coding assistant with an extensive toolkit.
 
-## Your Approach: Tool-Aware Problem Solving
+## YOUR PRIMARY DIRECTIVE: EXECUTE, DON'T TALK ABOUT EXECUTING
+
+\u{1F6A8} **CRITICAL - READ THIS FIRST** \u{1F6A8}
+YOU ARE AN EXECUTION AGENT, NOT A CONVERSATIONAL ASSISTANT.
+
+**WRONG BEHAVIOR (Never do this):**
+\u274C "I'll create a file called hello.js with a function..."
+\u274C "I created hello.js with the following code..."
+\u274C "Here's what the file would look like..."
+\u274C Showing code blocks without calling write_file tool
+
+**CORRECT BEHAVIOR (Always do this):**
+\u2705 Immediately call write_file tool with the code
+\u2705 Then say "Created hello.js with greeting function"
+\u2705 TOOLS FIRST, then brief confirmation
+
+**Core Principle: USE TOOLS, DON'T DESCRIBE**
+\u26A0\uFE0F CRITICAL: You MUST use your tools to perform actions. NEVER just describe what you would do or claim you did something without actually calling a tool.
 
-When the user asks you to do something, follow this process:
-1. **Analyze the request**: Understand what the user needs
-2. **Scan your tools**: Review which of your available tools can help accomplish the task
-3. **Plan your approach**: Decide which tools to use and in what order
-4. **Execute**: Use the appropriate tools, combining multiple when needed
-5. **Verify**: Check your work (read files after editing, run tests after changes)
+**Tool Calling is MANDATORY:**
+- User says "create a file" \u2192 CALL write_file tool FIRST (don't show code, don't explain, just CALL THE TOOL)
+- User says "search the web" \u2192 CALL web_search tool FIRST (don't describe what you would search for)
+- User says "run tests" \u2192 CALL bash_exec tool FIRST (don't say you ran them, actually run them)
+- EVERY action requires a TOOL CALL. Text responses are ONLY for brief confirmations AFTER tools execute.
+
+**Execution Process:**
+1. **Analyze**: Understand what the user wants (in your head, don't output this)
+2. **Execute**: IMMEDIATELY CALL THE APPROPRIATE TOOLS (this is mandatory, not optional)
+3. **Respond**: Brief confirmation of what was done (AFTER tools executed)
+
+**Critical Rules:**
+- User says "create X with Y" \u2192 Immediately call write_file/edit_file tool, no discussion
+- If a task needs data you don't have, fetch it with web_search/web_fetch FIRST, THEN complete the task with other tools
+- Never ask "should I do this?" or "do you want me to...?" - JUST DO IT (with tools)
+- If you don't call tools, you didn't do the task - showing code is NOT the same as creating files
+- NEVER show code blocks as examples - ALWAYS write them to files with tools
 
 **IMPORTANT**: You have many tools beyond basic file/bash/git. Before answering "I can't do that", check if any of your tools can help. For example:
 - Need information from the internet? Use **web_search** and **web_fetch**
@@ -10626,13 +10762,19 @@ When the user asks you to do something, follow this process:
 {TOOL_CATALOG}
 
 ## Guidelines
-- Be helpful and direct
-- If a task requires multiple steps, execute them one by one
-- Always verify your work by reading files after editing or running tests after changes
-- You can use multiple tools together for complex tasks
-- When uncertain which tool to use, check the full list above
+- **Be action-oriented**: Execute tasks immediately without asking for confirmation
+- **Multi-step tasks**: Chain tools together to complete the full request
+- **Always verify**: Read files after editing, run tests after changes
+- **Don't present options**: If the user says "create X", create it with reasonable defaults
+- **Don't ask "should I..."**: The user already told you what to do by making the request
+- **Combine tools**: Use web_search + write_file, bash + read_file, etc. to complete tasks fully
 - **Never** add "Co-Authored-By", "Generated by", or any AI attribution to commits, code comments, documentation, or PR descriptions. All output must read as if written by the developer.
 
+**Example Flows:**
+- "Create an HTML with weather data" \u2192 web_search for weather \u2192 write_file with HTML \u2192 DONE
+- "Add tests for function X" \u2192 read_file to see X \u2192 write_file with tests \u2192 bash to run tests \u2192 DONE
+- "Fix the bug in Y" \u2192 read_file to understand \u2192 edit_file to fix \u2192 bash to test \u2192 DONE
+
 ## File Access
 File operations are restricted to the project directory by default.
 When you need to access a path outside the project, use the **authorize_path** tool first \u2014 it will ask the user for permission interactively. Once authorized, proceed with the file operation.
@@ -24325,7 +24467,21 @@ Examples:
     const projectDir = cwd ?? process.cwd();
     const startTime = performance.now();
     const timeoutMs = timeout ?? DEFAULT_TIMEOUT_MS3;
+    const { CommandHeartbeat: CommandHeartbeat2 } = await Promise.resolve().then(() => (init_heartbeat(), heartbeat_exports));
+    const heartbeat = new CommandHeartbeat2({
+      onUpdate: (stats) => {
+        if (stats.elapsedSeconds > 10) {
+          process.stderr.write(`\r\u23F1\uFE0F  ${stats.elapsedSeconds}s elapsed`);
+        }
+      },
+      onWarn: (message) => {
+        process.stderr.write(`
+${message}
+`);
+      }
+    });
     try {
+      heartbeat.start();
       const pm = packageManager ?? await detectPackageManager(projectDir);
       const cmdArgs = ["run", script];
       if (args && args.length > 0) {
@@ -24336,13 +24492,30 @@ Examples:
         timeout: timeoutMs,
         env: { ...process.env, ...env2 },
         reject: false,
+        buffer: false,
+        // Enable streaming
         maxBuffer: MAX_OUTPUT_SIZE2
       };
-      const result = await execa(pm, cmdArgs, options);
+      const subprocess = execa(pm, cmdArgs, options);
+      let stdoutBuffer = "";
+      let stderrBuffer = "";
+      subprocess.stdout?.on("data", (chunk) => {
+        const text11 = chunk.toString();
+        stdoutBuffer += text11;
+        process.stdout.write(text11);
+        heartbeat.activity();
+      });
+      subprocess.stderr?.on("data", (chunk) => {
+        const text11 = chunk.toString();
+        stderrBuffer += text11;
+        process.stderr.write(text11);
+        heartbeat.activity();
+      });
+      const result = await subprocess;
       return {
         success: result.exitCode === 0,
-        stdout: truncateOutput2(String(result.stdout ?? "")),
-        stderr: truncateOutput2(String(result.stderr ?? "")),
+        stdout: truncateOutput2(stdoutBuffer),
+        stderr: truncateOutput2(stderrBuffer),
         exitCode: result.exitCode ?? 0,
         duration: performance.now() - startTime,
         packageManager: pm
@@ -24358,6 +24531,9 @@ Examples:
         `Failed to run script '${script}': ${error instanceof Error ? error.message : String(error)}`,
         { tool: "run_script", cause: error instanceof Error ? error : void 0 }
       );
+    } finally {
+      heartbeat.stop();
+      process.stderr.write("\r                                        \r");
     }
   }
 });
@@ -24383,7 +24559,21 @@ Examples:
     const projectDir = cwd ?? process.cwd();
     const startTime = performance.now();
     const timeoutMs = timeout ?? DEFAULT_TIMEOUT_MS3;
+    const { CommandHeartbeat: CommandHeartbeat2 } = await Promise.resolve().then(() => (init_heartbeat(), heartbeat_exports));
+    const heartbeat = new CommandHeartbeat2({
+      onUpdate: (stats) => {
+        if (stats.elapsedSeconds > 10) {
+          process.stderr.write(`\r\u23F1\uFE0F  ${stats.elapsedSeconds}s elapsed`);
+        }
+      },
+      onWarn: (message) => {
+        process.stderr.write(`
+${message}
+`);
+      }
+    });
     try {
+      heartbeat.start();
       const pm = packageManager ?? await detectPackageManager(projectDir);
       let cmdArgs;
       if (packages && packages.length > 0) {
@@ -24423,13 +24613,30 @@ Examples:
         cwd: projectDir,
         timeout: timeoutMs,
         reject: false,
+        buffer: false,
+        // Enable streaming
         maxBuffer: MAX_OUTPUT_SIZE2
       };
-      const result = await execa(pm, cmdArgs, options);
+      const subprocess = execa(pm, cmdArgs, options);
+      let stdoutBuffer = "";
+      let stderrBuffer = "";
+      subprocess.stdout?.on("data", (chunk) => {
+        const text11 = chunk.toString();
+        stdoutBuffer += text11;
+        process.stdout.write(text11);
+        heartbeat.activity();
+      });
+      subprocess.stderr?.on("data", (chunk) => {
+        const text11 = chunk.toString();
+        stderrBuffer += text11;
+        process.stderr.write(text11);
+        heartbeat.activity();
+      });
+      const result = await subprocess;
       return {
         success: result.exitCode === 0,
-        stdout: truncateOutput2(String(result.stdout ?? "")),
-        stderr: truncateOutput2(String(result.stderr ?? "")),
+        stdout: truncateOutput2(stdoutBuffer),
+        stderr: truncateOutput2(stderrBuffer),
         exitCode: result.exitCode ?? 0,
         duration: performance.now() - startTime,
         packageManager: pm
@@ -24445,6 +24652,9 @@ Examples:
         `Failed to install dependencies: ${error instanceof Error ? error.message : String(error)}`,
         { tool: "install_deps", cause: error instanceof Error ? error : void 0 }
       );
+    } finally {
+      heartbeat.stop();
+      process.stderr.write("\r                                        \r");
     }
   }
 });
@@ -24469,12 +24679,26 @@ Examples:
     const projectDir = cwd ?? process.cwd();
     const startTime = performance.now();
     const timeoutMs = timeout ?? DEFAULT_TIMEOUT_MS3;
+    const { CommandHeartbeat: CommandHeartbeat2 } = await Promise.resolve().then(() => (init_heartbeat(), heartbeat_exports));
+    const heartbeat = new CommandHeartbeat2({
+      onUpdate: (stats) => {
+        if (stats.elapsedSeconds > 10) {
+          process.stderr.write(`\r\u23F1\uFE0F  ${stats.elapsedSeconds}s elapsed`);
+        }
+      },
+      onWarn: (message) => {
+        process.stderr.write(`
+${message}
+`);
+      }
+    });
     try {
       try {
         await fs23__default.access(path20__default.join(projectDir, "Makefile"));
       } catch {
         throw new ToolError("No Makefile found in directory", { tool: "make" });
       }
+      heartbeat.start();
       const cmdArgs = [];
       if (target) {
         cmdArgs.push(...target.split(/\s+/));
@@ -24487,13 +24711,30 @@ Examples:
         timeout: timeoutMs,
         env: { ...process.env, ...env2 },
         reject: false,
+        buffer: false,
+        // Enable streaming
         maxBuffer: MAX_OUTPUT_SIZE2
       };
-      const result = await execa("make", cmdArgs, options);
+      const subprocess = execa("make", cmdArgs, options);
+      let stdoutBuffer = "";
+      let stderrBuffer = "";
+      subprocess.stdout?.on("data", (chunk) => {
+        const text11 = chunk.toString();
+        stdoutBuffer += text11;
+        process.stdout.write(text11);
+        heartbeat.activity();
+      });
+      subprocess.stderr?.on("data", (chunk) => {
+        const text11 = chunk.toString();
+        stderrBuffer += text11;
+        process.stderr.write(text11);
+        heartbeat.activity();
+      });
+      const result = await subprocess;
       return {
         success: result.exitCode === 0,
-        stdout: truncateOutput2(String(result.stdout ?? "")),
-        stderr: truncateOutput2(String(result.stderr ?? "")),
+        stdout: truncateOutput2(stdoutBuffer),
+        stderr: truncateOutput2(stderrBuffer),
         exitCode: result.exitCode ?? 0,
         duration: performance.now() - startTime
       };
@@ -24509,6 +24750,9 @@ Examples:
         `Make failed: ${error instanceof Error ? error.message : String(error)}`,
         { tool: "make", cause: error instanceof Error ? error : void 0 }
       );
+    } finally {
+      heartbeat.stop();
+      process.stderr.write("\r                                        \r");
     }
   }
 });
@@ -24534,7 +24778,21 @@ Examples:
     const projectDir = cwd ?? process.cwd();
     const startTime = performance.now();
     const timeoutMs = timeout ?? DEFAULT_TIMEOUT_MS3;
+    const { CommandHeartbeat: CommandHeartbeat2 } = await Promise.resolve().then(() => (init_heartbeat(), heartbeat_exports));
+    const heartbeat = new CommandHeartbeat2({
+      onUpdate: (stats) => {
+        if (stats.elapsedSeconds > 10) {
+          process.stderr.write(`\r\u23F1\uFE0F  ${stats.elapsedSeconds}s elapsed`);
+        }
+      },
+      onWarn: (message) => {
+        process.stderr.write(`
+${message}
+`);
+      }
+    });
     try {
+      heartbeat.start();
       const cmdArgs = [];
       if (project) {
         cmdArgs.push("--project", project);
@@ -24552,13 +24810,30 @@ Examples:
         cwd: projectDir,
         timeout: timeoutMs,
         reject: false,
+        buffer: false,
+        // Enable streaming
         maxBuffer: MAX_OUTPUT_SIZE2
       };
-      const result = await execa("npx", ["tsc", ...cmdArgs], options);
+      const subprocess = execa("npx", ["tsc", ...cmdArgs], options);
+      let stdoutBuffer = "";
+      let stderrBuffer = "";
+      subprocess.stdout?.on("data", (chunk) => {
+        const text11 = chunk.toString();
+        stdoutBuffer += text11;
+        process.stdout.write(text11);
+        heartbeat.activity();
+      });
+      subprocess.stderr?.on("data", (chunk) => {
+        const text11 = chunk.toString();
+        stderrBuffer += text11;
+        process.stderr.write(text11);
+        heartbeat.activity();
+      });
+      const result = await subprocess;
       return {
         success: result.exitCode === 0,
-        stdout: truncateOutput2(String(result.stdout ?? "")),
-        stderr: truncateOutput2(String(result.stderr ?? "")),
+        stdout: truncateOutput2(stdoutBuffer),
+        stderr: truncateOutput2(stderrBuffer),
         exitCode: result.exitCode ?? 0,
         duration: performance.now() - startTime
       };
@@ -24573,6 +24848,9 @@ Examples:
         `TypeScript compile failed: ${error instanceof Error ? error.message : String(error)}`,
         { tool: "tsc", cause: error instanceof Error ? error : void 0 }
       );
+    } finally {
+      heartbeat.stop();
+      process.stderr.write("\r                                        \r");
     }
   }
 });
@@ -30442,7 +30720,11 @@ async function startRepl(options = {}) {
           clearSpinner();
           renderToolStart(result2.name, result2.input);
           renderToolEnd(result2);
-          setSpinner("Processing...");
+          if (isCocoMode()) {
+            setSpinner("Processing results & checking quality...");
+          } else {
+            setSpinner("Processing...");
+          }
         },
         onToolSkipped: (tc, reason) => {
           clearSpinner();
@@ -30455,10 +30737,18 @@ async function startRepl(options = {}) {
             if (!thinkingStartTime) return;
             const elapsed = Math.floor((Date.now() - thinkingStartTime) / 1e3);
             if (elapsed < 4) return;
-            if (elapsed < 8) setSpinner("Analyzing request...");
-            else if (elapsed < 12) setSpinner("Planning approach...");
-            else if (elapsed < 16) setSpinner("Preparing tools...");
-            else setSpinner(`Still working... (${elapsed}s)`);
+            if (isCocoMode()) {
+              if (elapsed < 8) setSpinner("Analyzing request...");
+              else if (elapsed < 15) setSpinner("Running quality checks...");
+              else if (elapsed < 25) setSpinner("Iterating for quality...");
+              else if (elapsed < 40) setSpinner("Verifying implementation...");
+              else setSpinner(`Quality iteration in progress... (${elapsed}s)`);
+            } else {
+              if (elapsed < 8) setSpinner("Analyzing request...");
+              else if (elapsed < 12) setSpinner("Planning approach...");
+              else if (elapsed < 16) setSpinner("Preparing tools...");
+              else setSpinner(`Still working... (${elapsed}s)`);
+            }
           }, 2e3);
         },
         onThinkingEnd: () => {