aether-code 0.43.0 → 0.43.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aether-code",
-  "version": "0.43.0",
+  "version": "0.43.2",
   "description": "Uncensored AI coding agent for your terminal — Claude Code alternative with MCP support. Reads code, writes files, runs commands. Drives IDA Pro, Roblox Studio, Wireshark, Blender, and any MCP server. No refusal layer.",
   "homepage": "https://trynoguard.com",
   "repository": {

package/src/agent.js

@@ -13,6 +13,9 @@ import { scanProject, detectVerifyCommand, detectBuildCommand } from "./project-
 import { c, divider, turn, toolLabel, toolSummary, makeTokenStripper, errorLine, startSpinner } from "./render.js";
 
 const DEFAULT_MAX_TURNS = 25;
+const DEFAULT_MAX_CREDITS = 500;
+const MAX_TOTAL_TOOL_CALLS = 40;
+const MAX_CONSECUTIVE_TOOL_ONLY = 6;
 
 // Rolling history window (messages) sent upstream per turn. The session keeps
 // the full transcript locally, but we only forward the anchor (first message,
@@ -86,6 +89,7 @@ export async function runAgent({
   autoYes = false,
   unsafePaths = false,
   maxTurns = DEFAULT_MAX_TURNS,
+  maxCredits = DEFAULT_MAX_CREDITS,
   model = null, // null = server default (gemma). Premium models gated server-side.
   onTokens = () => {},
   // Optional MCPManager. When provided, its tools are merged into the agent's
@@ -141,6 +145,8 @@ export async function runAgent({
   let appliedNothingNudges = 0;
   let verifyRounds = 0;
   let buildRounds = 0;
+  let consecutiveToolOnlyTurns = 0;
+  let totalToolCalls = 0;
 
   for (let i = 0; i < maxTurns; i++) {
     // No turn header and no leading blank here — each step (assistant text and
@@ -215,13 +221,17 @@ export async function runAgent({
       process.stdout.write(tail);
     }
     if (lastWasText) process.stdout.write("\n");
-    totalCredits += res.creditsCharged ?? 0;
+    const turnCredits = res.creditsCharged ?? 0;
+    totalCredits += turnCredits;
     totalIn += res.usage?.prompt_tokens ?? 0;
     totalOut += res.usage?.completion_tokens ?? 0;
     if (typeof res.balanceAfter === "number") lastBalance = res.balanceAfter;
     onTokens({ totalCredits, totalIn, totalOut, balance: lastBalance });
-    // Per-turn cost line removed for a cleaner look — the session summary at the
-    // end carries the totals.
+
+    // Per-turn cost — visible so users catch runaway loops early
+    const inK = ((res.usage?.prompt_tokens ?? 0) / 1000).toFixed(1);
+    const outTok = res.usage?.completion_tokens ?? 0;
+    console.log(c.gray(`  ${inK}k→${outTok} tokens · ${turnCredits} credits (total: ${totalCredits})`));
 
     // Push assistant message into history
     messages.push({
@@ -231,6 +241,31 @@ export async function runAgent({
     });
 
     const toolCalls = res.message.tool_calls ?? [];
+    const hasTextContent = !!(res.message.content && res.message.content.trim());
+
+    if (toolCalls.length > 0 && !hasTextContent) {
+      consecutiveToolOnlyTurns++;
+    } else if (hasTextContent) {
+      consecutiveToolOnlyTurns = 0;
+    }
+
+    // Credit cap — hard stop to prevent runaway cost
+    if (maxCredits > 0 && totalCredits >= maxCredits) {
+      console.log(c.red(`\n[Credit budget exhausted: ${totalCredits} credits used, limit was ${maxCredits}. Stopping.]`));
+      return { ok: true, totalCredits, totalIn, totalOut, balance: lastBalance, messages };
+    }
+
+    // Tool call count cap
+    if (toolCalls.length > 0 && totalToolCalls + toolCalls.length > MAX_TOTAL_TOOL_CALLS) {
+      console.log(c.red(`\n[Tool call limit reached (${MAX_TOTAL_TOOL_CALLS}). Stopping.]`));
+      return { ok: true, totalCredits, totalIn, totalOut, balance: lastBalance, messages };
+    }
+
+    // Consecutive tool-only turns cap (model stuck in a tool loop with no text)
+    if (consecutiveToolOnlyTurns >= MAX_CONSECUTIVE_TOOL_ONLY) {
+      console.log(c.red(`\n[Too many consecutive tool-only turns (${MAX_CONSECUTIVE_TOOL_ONLY}). Stopping.]`));
+      return { ok: true, totalCredits, totalIn, totalOut, balance: lastBalance, messages };
+    }
     if (toolCalls.length === 0) {
       // Completeness guard: on a modification task, if the model read files it
       // never edited, it likely either described the fix instead of applying it
@@ -343,6 +378,7 @@ export async function runAgent({
 
     // Execute each tool call. Show the actual args (now that we have them
     // fully assembled) and run.
+    totalToolCalls += toolCalls.length;
     for (const call of toolCalls) {
       let args = {};
       try { args = JSON.parse(call.function.arguments || "{}"); } catch { /* leave empty */ }