@prestyj/agent 5.1.0 → 5.2.0

package/dist/index.d.ts

@@ -60,9 +60,21 @@ interface AgentDoneEvent {
     totalTurns: number;
     totalUsage: Usage;
 }
+/**
+ * Terminal signal emitted when the loop stops because it exhausted its turn
+ * budget (`maxTurns`) mid-task — i.e. the model still wanted to run tools but
+ * ran out of turns. Distinguishes a hard cut-off from a clean completion so
+ * callers (e.g. the subagent spawner) can tell the parent the output may be
+ * incomplete. Yielded immediately before the final `agent_done`.
+ */
+interface AgentMaxTurnsEvent {
+    type: "max_turns";
+    totalTurns: number;
+    maxTurns: number;
+}
 interface AgentRetryEvent {
     type: "retry";
-    reason: "overloaded" | "rate_limit" | "provider_error" | "empty_response" | "stream_stall" | "overflow_compact";
+    reason: "overloaded" | "rate_limit" | "provider_error" | "empty_response" | "stream_stall" | "overflow_compact" | "tool_argument_glitch";
     attempt: number;
     maxAttempts: number;
     delayMs: number;
@@ -101,7 +113,7 @@ interface AgentFollowUpMessageEvent {
     type: "follow_up_message";
     content: Message["content"];
 }
-type AgentEvent = AgentTextDeltaEvent | AgentThinkingDeltaEvent | AgentToolCallStartEvent | AgentToolCallUpdateEvent | AgentToolCallEndEvent | AgentToolCallDeltaEvent | AgentServerToolCallEvent | AgentServerToolResultEvent | AgentSteeringMessageEvent | AgentFollowUpMessageEvent | AgentRetryEvent | AgentTurnEndEvent | AgentDoneEvent | AgentErrorEvent;
+type AgentEvent = AgentTextDeltaEvent | AgentThinkingDeltaEvent | AgentToolCallStartEvent | AgentToolCallUpdateEvent | AgentToolCallEndEvent | AgentToolCallDeltaEvent | AgentServerToolCallEvent | AgentServerToolResultEvent | AgentSteeringMessageEvent | AgentFollowUpMessageEvent | AgentRetryEvent | AgentTurnEndEvent | AgentDoneEvent | AgentMaxTurnsEvent | AgentErrorEvent;
 interface AgentOptions {
     provider: StreamOptions["provider"];
     model: string;

package/dist/index.js

@@ -6,6 +6,7 @@ import { ZodError, prettifyError } from "zod";
 import {
   stream,
   EventStream,
+  EZCoderAIError,
   isHardBillingMessage
 } from "@prestyj/ai";
 var DEFAULT_MAX_TURNS = 300;
@@ -232,6 +233,7 @@ async function* agentLoop(messages, options) {
   const toolMap = new Map((options.tools ?? []).map((t) => [t.name, t]));
   const totalUsage = { inputTokens: 0, outputTokens: 0 };
   let turn = 0;
+  let hitMaxTurns = false;
   let firstTurn = true;
   let consecutivePauses = 0;
   let toolPairingRepaired = false;
@@ -242,6 +244,7 @@ async function* agentLoop(messages, options) {
   let overflowCompactionAttempts = 0;
   let toolResultTruncationAttempted = false;
   const invalidToolArgumentCounts = /* @__PURE__ */ new Map();
+  let toolArgumentAutoContinueUsed = false;
   let useNonStreamingFallback = false;
   const MAX_OVERLOAD_RETRIES = 10;
   const MAX_EMPTY_RESPONSE_RETRIES = 2;
@@ -814,8 +817,12 @@ async function* agentLoop(messages, options) {
         }
       }
       let fatalToolArgumentError = null;
-      const markFatalToolArgumentError = (error) => {
+      let fatalToolArgumentRecoverable = false;
+      let fatalToolArgumentToolName = "";
+      const markFatalToolArgumentError = (error, recoverable, toolName) => {
         fatalToolArgumentError = error;
+        fatalToolArgumentRecoverable = recoverable;
+        fatalToolArgumentToolName = toolName;
       };
       const executionOptions = {
         signal: options.signal,
@@ -831,8 +838,24 @@ async function* agentLoop(messages, options) {
       messages.push({ role: "tool", content: executionResult.toolResults });
       const toolsAborted = executionResult.aborted;
       if (fatalToolArgumentError) {
-        yield { type: "error", error: fatalToolArgumentError };
-        break;
+        if (fatalToolArgumentRecoverable && !toolArgumentAutoContinueUsed) {
+          toolArgumentAutoContinueUsed = true;
+          for (const key of invalidToolArgumentCounts.keys()) {
+            if (key.startsWith(`${fatalToolArgumentToolName}:`))
+              invalidToolArgumentCounts.delete(key);
+          }
+          yield {
+            type: "retry",
+            reason: "tool_argument_glitch",
+            attempt: 1,
+            maxAttempts: 1,
+            delayMs: 0,
+            silent: false
+          };
+        } else {
+          yield { type: "error", error: fatalToolArgumentError };
+          break;
+        }
       }
       if (toolsAborted) break;
       if (options.getSteeringMessages) {
@@ -844,6 +867,9 @@ async function* agentLoop(messages, options) {
           }
         }
       }
+      if (turn >= maxTurns) {
+        hitMaxTurns = true;
+      }
     }
   } finally {
     sanitizeOrphanedServerTools(messages);
@@ -855,6 +881,19 @@ async function* agentLoop(messages, options) {
       break;
     }
   }
+  if (hitMaxTurns) {
+    diag("max_turns_reached", {
+      turn,
+      maxTurns,
+      provider: options.provider,
+      model: options.model
+    });
+    yield {
+      type: "max_turns",
+      totalTurns: turn,
+      maxTurns
+    };
+  }
   yield {
     type: "agent_done",
     totalTurns: turn,
@@ -920,10 +959,17 @@ async function executeSingleToolCall(toolCall, options, pushEvent) {
         resultContent = `Invalid arguments for tool \`${toolCall.name}\`:
 ` + prettyError + "\nRe-issue the call with each field as the correct type.";
         if (failureCount >= 3) {
+          const recoverable = Object.keys(toolCall.args ?? {}).length === 0;
           options.markFatalToolArgumentError(
-            new Error(
-              `The model repeatedly issued invalid arguments for tool \`${toolCall.name}\`. This is usually an upstream model/tool-calling bug. Your conversation is preserved; send another message or switch models to continue.`
-            )
+            new EZCoderAIError(
+              `The model repeatedly issued invalid arguments for tool \`${toolCall.name}\`. This is usually an upstream model/tool-calling bug` + (recoverable ? " (the provider's stream returned empty tool-call arguments)" : "") + `. Your conversation is preserved; send another message or switch models to continue.`,
+              {
+                source: "provider",
+                hint: "This is the model/provider's fault, not a ezcoder bug. " + (recoverable ? "ezcoder already retried automatically once; if it recurs, send another message or switch models." : "Send another message or switch models to continue.")
+              }
+            ),
+            recoverable,
+            toolCall.name
           );
         }
       } else {