@prestyj/agent 5.11.0 → 5.13.0

package/dist/index.cjs

@@ -83,7 +83,13 @@ function isContextOverflow(err) {
   if (overflowStatus === 402) return false;
   if (isBillingError(err)) return false;
   const msg = err.message.toLowerCase();
-  return msg.includes("prompt is too long") || msg.includes("prompt too long") || msg.includes("input is too long") || msg.includes("context_length_exceeded") || msg.includes("context_window_exceeded") || msg.includes("maximum context length") || msg.includes("exceeds model context window") || msg.includes("exceeds the context window") || msg.includes("content_too_large") || msg.includes("request_too_large") || msg.includes("reduce the length") || msg.includes("please shorten") || msg.includes("token") && msg.includes("exceed");
+  if (msg.includes("prompt is too long") || msg.includes("prompt too long") || msg.includes("input is too long") || msg.includes("context_length_exceeded") || msg.includes("context_window_exceeded") || msg.includes("maximum context length") || msg.includes("exceeds model context window") || msg.includes("exceeds the context window") || msg.includes("content_too_large") || msg.includes("request_too_large") || msg.includes("reduce the length") || msg.includes("please shorten")) {
+    return true;
+  }
+  const rateLimited = overflowStatus === 429 || msg.includes("rate limit") || msg.includes("rate_limit") || msg.includes("too many requests");
+  const perUnitTime = msg.includes("per min") || msg.includes("/min") || msg.includes("per minute") || msg.includes("per hour") || msg.includes("per day") || msg.includes("tpm") || msg.includes("rpm");
+  if (rateLimited && perUnitTime) return false;
+  return msg.includes("token") && msg.includes("exceed");
 }
 function parseOverflowNumber(value) {
   return Number(value.replace(/[,_\s]/g, ""));
@@ -196,6 +202,17 @@ function isMalformedStream(err) {
   const msg = err.message;
   return /\bin JSON at position \d+/i.test(msg);
 }
+var TIMEOUT_NAMES = /* @__PURE__ */ new Set(["TimeoutError", "ConnectTimeoutError", "HeadersTimeoutError"]);
+var TIMEOUT_MESSAGES = [
+  /^request timed out\.?$/i,
+  /\brequest to [\w .-]+ timed out\b/i,
+  /\b(?:connection|socket|headers|stream) timed out\b/i
+];
+function isBareTimeout(e) {
+  if (typeof e.name === "string" && TIMEOUT_NAMES.has(e.name)) return true;
+  if (typeof e.message !== "string") return false;
+  return TIMEOUT_MESSAGES.some((re) => re.test(e.message));
+}
 function isTransportFailure(err) {
   const codes = /* @__PURE__ */ new Set([
     "ECONNRESET",
@@ -232,6 +249,8 @@ function isTransportFailure(err) {
     if (typeof e.message === "string") {
       for (const re of messages) if (re.test(e.message)) return true;
     }
+    const clientError = typeof e.status === "number" && e.status >= 400 && e.status < 500;
+    if (!clientError && isBareTimeout(e)) return true;
     cur = e.cause;
   }
   return false;
@@ -457,6 +476,21 @@ async function* agentLoop(messages, options) {
         diag("stream_call", { nonStreaming: useNonStreamingFallback });
         streamCallStart = Date.now();
         providerAttemptStartedAt = streamCallStart;
+        let liveApiKey = options.apiKey;
+        let liveAccountId = options.accountId;
+        let liveProjectId = options.projectId;
+        if (options.resolveCredentials) {
+          try {
+            const fresh = await options.resolveCredentials();
+            liveApiKey = fresh.apiKey;
+            if (fresh.accountId !== void 0) liveAccountId = fresh.accountId;
+            if (fresh.projectId !== void 0) liveProjectId = fresh.projectId;
+          } catch (credErr) {
+            diag("credential_refresh_failed", {
+              error: (credErr instanceof Error ? credErr.message : String(credErr)).slice(0, 200)
+            });
+          }
+        }
         const result = (0, import_ai.stream)({
           provider: options.provider,
           model: options.model,
@@ -468,12 +502,12 @@ async function* agentLoop(messages, options) {
           maxTokens: options.maxTokens,
           temperature: options.temperature,
           thinking: options.thinking,
-          apiKey: options.apiKey,
+          apiKey: liveApiKey,
           baseUrl: options.baseUrl,
           signal: streamController.signal,
-          accountId: options.accountId,
+          accountId: liveAccountId,
           transportSessionId: options.transportSessionId,
-          projectId: options.projectId,
+          projectId: liveProjectId,
           cacheRetention: options.cacheRetention,
           promptCacheKey: options.promptCacheKey,
           serviceTier: options.serviceTier,
@@ -917,6 +951,7 @@ async function* agentLoop(messages, options) {
           continue;
         }
       }
+      const emptyExhausted = !hasActionableContent;
       emptyResponseRetries = 0;
       useNonStreamingFallback = false;
       totalUsage.inputTokens += response.usage.inputTokens;
@@ -930,9 +965,11 @@ async function* agentLoop(messages, options) {
       if (response.usage.cacheWrite) {
         totalUsage.cacheWrite = (totalUsage.cacheWrite ?? 0) + response.usage.cacheWrite;
       }
-      messages.push(response.message);
-      latestProviderUsage = response.usage;
-      usageAnchorIndex = messages.length - 1;
+      if (!emptyExhausted) {
+        messages.push(response.message);
+        latestProviderUsage = response.usage;
+        usageAnchorIndex = messages.length - 1;
+      }
       const completedAt = Date.now();
       const outputTokensPerSecond = providerDurationMs > 0 && response.usage.outputTokens > 0 ? response.usage.outputTokens / (providerDurationMs / 1e3) : void 0;
       const timing = {
@@ -964,8 +1001,15 @@ async function* agentLoop(messages, options) {
       }
       consecutivePauses = 0;
       const allToolCalls = extractToolCalls(response.message.content);
-      if (response.stopReason !== "tool_use" && allToolCalls.length === 0) {
-        if (response.stopReason === "max_tokens") {
+      if (emptyExhausted || response.stopReason !== "tool_use" && allToolCalls.length === 0) {
+        if (emptyExhausted) {
+          diag("empty_response_exhausted", {
+            provider: options.provider,
+            model: options.model,
+            stopReason: response.stopReason
+          });
+          yield { type: "truncated", reason: "empty_response", continued: false };
+        } else if (response.stopReason === "max_tokens") {
           if (maxTokensContinuations < MAX_OUTPUT_CONTINUATIONS) {
             maxTokensContinuations++;
             diag("max_tokens_continuation", {
@@ -1187,6 +1231,7 @@ async function executeSingleToolCall(toolCall, options, pushEvent) {
   let resultContent;
   let details;
   let isError = false;
+  let invalidArgAttempt;
   const tool = options.toolMap.get(toolCall.name);
   if (!tool) {
     resultContent = `Unknown tool: ${toolCall.name}`;
@@ -1224,6 +1269,7 @@ async function executeSingleToolCall(toolCall, options, pushEvent) {
         const failureKey = `${toolCall.name}:${prettyError}`;
         const failureCount = (options.invalidToolArgumentCounts.get(failureKey) ?? 0) + 1;
         options.invalidToolArgumentCounts.set(failureKey, failureCount);
+        invalidArgAttempt = failureCount;
         resultContent = `Invalid arguments for tool \`${toolCall.name}\`:
 ` + prettyError + "\nRe-issue the call with each field as the correct type.";
         if (failureCount >= 3) {
@@ -1254,7 +1300,8 @@ async function executeSingleToolCall(toolCall, options, pushEvent) {
     result: toolResultPreview(resultContent),
     details,
     isError,
-    durationMs
+    durationMs,
+    ...invalidArgAttempt === void 0 ? {} : { invalidArgAttempt }
   });
   return { toolCallId: toolCall.id, content: resultContent, isError };
 }