@ai-sdk/provider-utils - npm Package Compare versions

Comparing version 5.0.2 to 5.0.3

src/retry-with-exponential-backoff.ts

+import { delay } from './delay';
+import { getErrorMessage } from './get-error-message';
+import { isAbortError } from './is-abort-error';
+
+export type RetryFunction = <OUTPUT>(
+  fn: () => PromiseLike<OUTPUT>,
+) => PromiseLike<OUTPUT>;
+
+export type RetryErrorReason = 'maxRetriesExceeded' | 'errorNotRetryable';
+
+export type RetryErrorFactory = ({
+  message,
+  reason,
+  errors,
+}: {
+  message: string;
+  reason: RetryErrorReason;
+  errors: Array<unknown>;
+}) => unknown;
+
+export type RetryDelayProvider = ({
+  error,
+  exponentialBackoffDelay,
+}: {
+  error: unknown;
+  exponentialBackoffDelay: number;
+}) => number;
+
+export type ShouldRetryFunction = (
+  error: unknown,
+) => boolean | Promise<boolean>;
+
+/**
+ * Retries a failed operation with exponential backoff.
+ */
+export const retryWithExponentialBackoff =
+  ({
+    maxRetries = 2,
+    initialDelayInMs = 2000,
+    backoffFactor = 2,
+    abortSignal,
+    shouldRetry,
+    getDelayInMs = ({ exponentialBackoffDelay }) => exponentialBackoffDelay,
+    createRetryError = ({ message }) => new Error(message),
+  }: {
+    maxRetries?: number;
+    initialDelayInMs?: number;
+    backoffFactor?: number;
+    abortSignal?: AbortSignal;
+    shouldRetry: ShouldRetryFunction;
+    getDelayInMs?: RetryDelayProvider;
+    createRetryError?: RetryErrorFactory;
+  }): RetryFunction =>
+  async <OUTPUT>(f: () => PromiseLike<OUTPUT>) =>
+    retryWithExponentialBackoffInternal(f, {
+      maxRetries,
+      delayInMs: initialDelayInMs,
+      backoffFactor,
+      abortSignal,
+      shouldRetry,
+      getDelayInMs,
+      createRetryError,
+    });
+
+async function retryWithExponentialBackoffInternal<OUTPUT>(
+  f: () => PromiseLike<OUTPUT>,
+  {
+    maxRetries,
+    delayInMs,
+    backoffFactor,
+    abortSignal,
+    shouldRetry,
+    getDelayInMs,
+    createRetryError,
+  }: {
+    maxRetries: number;
+    delayInMs: number;
+    backoffFactor: number;
+    abortSignal: AbortSignal | undefined;
+    shouldRetry: ShouldRetryFunction;
+    getDelayInMs: RetryDelayProvider;
+    createRetryError: RetryErrorFactory;
+  },
+  errors: unknown[] = [],
+): Promise<OUTPUT> {
+  try {
+    return await f();
+  } catch (error) {
+    if (isAbortError(error)) {
+      throw error; // don't retry when the request was aborted
+    }
+
+    if (maxRetries === 0) {
+      throw error; // don't wrap the error when retries are disabled
+    }
+
+    const errorMessage = getErrorMessage(error);
+    const newErrors = [...errors, error];
+    const tryNumber = newErrors.length;
+
+    if (tryNumber > maxRetries) {
+      throw createRetryError({
+        message: `Failed after ${tryNumber} attempts. Last error: ${errorMessage}`,
+        reason: 'maxRetriesExceeded',
+        errors: newErrors,
+      });
+    }
+
+    if ((await shouldRetry(error)) && tryNumber <= maxRetries) {
+      await delay(
+        getDelayInMs({
+          error,
+          exponentialBackoffDelay: delayInMs,
+        }),
+        { abortSignal },
+      );
+
+      return retryWithExponentialBackoffInternal(
+        f,
+        {
+          maxRetries,
+          delayInMs: backoffFactor * delayInMs,
+          backoffFactor,
+          abortSignal,
+          shouldRetry,
+          getDelayInMs,
+          createRetryError,
+        },
+        newErrors,
+      );
+    }
+
+    if (tryNumber === 1) {
+      throw error; // don't wrap the error when a non-retryable error occurs on the first try
+    }
+
+    throw createRetryError({
+      message: `Failed after ${tryNumber} attempts with non-retryable error: '${errorMessage}'`,
+      reason: 'errorNotRetryable',
+      errors: newErrors,
+    });
+  }
+}

CHANGELOG.md

 # @ai-sdk/provider-utils
 
+## 5.0.3
+
+### Patch Changes
+
+- 8c616f0: feat(mcp): add maxRetries option for failed mcp tool calls
+
 ## 5.0.2
@@ -4,0 +10,0 @@

package.json

 {
   "name": "@ai-sdk/provider-utils",
-  "version": "5.0.2",
+  "version": "5.0.3",
   "type": "module",
@@ -5,0 +5,0 @@ "license": "Apache-2.0",

src/index.ts

@@ -73,2 +73,3 @@ export { asArray } from './as-array';
 export { resolveProviderReference } from './resolve-provider-reference';
+export * from './retry-with-exponential-backoff';
 export * from './response-handler';
@@ -75,0 +76,0 @@ export {

No alert changes

Improved metrics

Total package byte prevSize

756460

1.6%

Number of package files

153

0.66%

Lines of code

13070

1.82%

No dependency changes