agent-accelerator 0.5.1 → 0.5.3

package/src/utils/retry.ts

@@ -29,7 +29,21 @@ export function isTransientError(error: unknown): boolean {
   return false;
 }
 
-function readRetryAfterMs(error: unknown): number | undefined {
+/**
+ * Whether a FAILED RUN (the provider layer already retried the call) is worth trying again: the failure is transient
+ * (HTTP 408/425/429/5xx, network codes, `transient`/`retryable` flags, `incomplete_stream`, a request/stream timeout).
+ * Never for aborts, bad configuration or input (401/403/400/404, `ConfigError`, `ValidationError`, `SecurityPolicyError`,
+ * `ThinkingLevelError`) and never for budgets or limits, which a retry would only reset or hit again.
+ */
+export function isRetryableRun(error: unknown): boolean {
+  if (!error || isAbortError(error)) return false;
+  const name = (error as { name?: unknown }).name;
+  if (name === "TimeoutError") return true; // the loop's own request / first-byte / idle timeouts
+  if (["BudgetExceededError", "ConfigError", "ValidationError", "SecurityPolicyError", "SessionLoadError", "ThinkingLevelError", "ToolError"].includes(String(name))) return false;
+  return isTransientError(error);
+}
+
+export function readRetryAfterMs(error: unknown): number | undefined {
   const err = error as Record<string, unknown>;
   // Preferred: milliseconds attached by the adapter from the Retry-After header.
   const direct = err["retryAfterMs"];
@@ -43,7 +57,8 @@ function readRetryAfterMs(error: unknown): number | undefined {
   return Math.min(60_000, secs * 1000);
 }
 
-function sleep(ms: number, signal?: AbortSignal): Promise<void> {
+// Waits `ms`, rejecting with an AbortError as soon as `signal` aborts; in: retry backoff and Google background polling, out: the awaiting caller.
+export function sleep(ms: number, signal?: AbortSignal): Promise<void> {
   if (ms <= 0) return Promise.resolve();
   return new Promise((resolve, reject) => {
     const timer = setTimeout(() => {