ai-retry 0.5.0 → 0.5.1

package/README.md

@@ -170,11 +170,10 @@ const retryable = createRetryable({
 
 #### Retry After Delay
 
-Handle retryable errors with delays and respect `retry-after` headers from rate-limited responses. This is useful for handling 429 (Too Many Requests) and 503 (Service Unavailable) errors.
-
-> [!NOTE] 
-> If the response contains a [`retry-after`](https://developer.mozilla.org/en-US/docs/Web/HTTP/Reference/Headers/Retry-After) header, it will be prioritized over the configured delay.
+If an error is retryable, such as 429 (Too Many Requests) or 503 (Service Unavailable) errors, it will be retried after a delay. 
+The delay and exponential backoff can be configured. If the response contains a [`retry-after`](https://developer.mozilla.org/en-US/docs/Web/HTTP/Reference/Headers/Retry-After) header, it will be prioritized over the configured delay.
 
+Note that this retryable does not accept a model parameter, it will always retry the model from the latest failed attempt.
 
 ```typescript
 import { retryAfterDelay } from 'ai-retry/retryables';
@@ -187,14 +186,14 @@ const retryableModel = createRetryable({
 
     // Or retry with exponential backoff (2s, 4s, 8s)
     retryAfterDelay({ delay: 2000, backoffFactor: 2, maxAttempts: 3 }),
-    
-    // Or switch to a different model after delay
-    retryAfterDelay(openai('gpt-4-mini'), { delay: 1000 }),
+
+    // Or retry only if the response contains a retry-after header
+    retryAfterDelay({ maxAttempts: 3 }),
   ],
 });
 ```
 
-By default, if a [`retry-after-ms`](https://learn.microsoft.com/en-us/azure/ai-foundry/openai/how-to/provisioned-get-started#what-should--i-do-when-i-receive-a-429-response) or `retry-after` header is present in the response, it will be prioritized over the configured delay. The delay from the header will be capped at 60 seconds for safety. If no headers are present, the configured delay or exponential backoff will be used.
+By default, if a [`retry-after-ms`](https://learn.microsoft.com/en-us/azure/ai-foundry/openai/how-to/provisioned-get-started#what-should--i-do-when-i-receive-a-429-response) or [`retry-after`](https://developer.mozilla.org/en-US/docs/Web/HTTP/Reference/Headers/Retry-After) header is present in the response, it will be prioritized over the configured delay. The delay from the header will be capped at 60 seconds for safety.
 
 #### Fallbacks
 
@@ -370,7 +369,7 @@ There are several built-in retryables:
 - [`contentFilterTriggered`](./src/retryables/content-filter-triggered.ts): Content filter was triggered based on the prompt or completion.
 - [`requestTimeout`](./src/retryables/request-timeout.ts): Request timeout occurred.
 - [`requestNotRetryable`](./src/retryables/request-not-retryable.ts): Request failed with a non-retryable error.
-- [`retryAfterDelay`](./src/retryables/retry-after-delay.ts): Retry with exponential backoff and respect `retry-after` headers for rate limiting.
+- [`retryAfterDelay`](./src/retryables/retry-after-delay.ts): Retry with delay and exponential backoff and respect `retry-after` headers.
 - [`serviceOverloaded`](./src/retryables/service-overloaded.ts): Response with status code 529 (service overloaded).
   - Use this retryable to handle Anthropic's overloaded errors.
 

package/dist/retryables/index.d.ts

@@ -23,11 +23,10 @@ declare function requestTimeout<MODEL extends LanguageModelV2 | EmbeddingModelV2
 //#endregion
 //#region src/retryables/retry-after-delay.d.ts
 /**
- * Retry with the same or a different model if the error is retryable with a delay.
+ * Retry the current failed attempt with the same model, if the error is retryable.
  * Uses the `Retry-After` or `Retry-After-Ms` headers if present.
- * Otherwise uses the specified `delay` with exponential backoff if `backoffFactor` is provided.
+ * Otherwise uses the specified `delay` and `backoffFactor` if provided.
  */
-declare function retryAfterDelay<MODEL extends LanguageModelV2 | EmbeddingModelV2>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
 declare function retryAfterDelay<MODEL extends LanguageModelV2 | EmbeddingModelV2>(options: RetryableOptions<MODEL>): Retryable<MODEL>;
 //#endregion
 //#region src/retryables/service-overloaded.d.ts

package/dist/retryables/index.js

@@ -1,4 +1,4 @@
-import { a as isModelV2, n as isErrorAttempt, o as isObject, s as isResultAttempt, u as isString } from "../utils-BlCGaP0E.js";
+import { n as isErrorAttempt, o as isObject, s as isResultAttempt, u as isString } from "../utils-BlCGaP0E.js";
 import { isAbortError } from "@ai-sdk/provider-utils";
 import { APICallError } from "ai";
 
@@ -69,23 +69,6 @@ function requestTimeout(model, options) {
 	};
 }
 
-//#endregion
-//#region src/internal/resolve-retryable-options.ts
-/**
-* Helper to resolve `RetryableOptions` from either a model and/or options object.
-* Used to support function overloads in retryables:
-* - `retryable(model)`
-* - `retryable(model, options)`
-* - `retryable(options)`
-*/
-function resolveRetryableOptions(modelOrOptions, options) {
-	if (isModelV2(modelOrOptions)) return {
-		...options,
-		model: modelOrOptions
-	};
-	return modelOrOptions;
-}
-
 //#endregion
 //#region src/parse-retry-headers.ts
 function parseRetryHeaders(headers) {
@@ -108,23 +91,28 @@ function parseRetryHeaders(headers) {
 //#endregion
 //#region src/retryables/retry-after-delay.ts
 const MAX_RETRY_AFTER_MS = 6e4;
-function retryAfterDelay(modelOrOptions, options) {
-	const resolvedOptions = resolveRetryableOptions(modelOrOptions, options);
+/**
+* Retry the current failed attempt with the same model, if the error is retryable.
+* Uses the `Retry-After` or `Retry-After-Ms` headers if present.
+* Otherwise uses the specified `delay` and `backoffFactor` if provided.
+*/
+function retryAfterDelay(options) {
 	return (context) => {
 		const { current } = context;
 		if (isErrorAttempt(current)) {
 			const { error } = current;
 			if (APICallError.isInstance(error) && error.isRetryable === true) {
-				const model = resolvedOptions.model ?? current.model;
+				const model = current.model;
 				const headerDelay = parseRetryHeaders(error.responseHeaders);
 				if (headerDelay !== null) return {
 					model,
-					...resolvedOptions,
-					delay: Math.min(headerDelay, MAX_RETRY_AFTER_MS)
+					...options,
+					delay: Math.min(headerDelay, MAX_RETRY_AFTER_MS),
+					backoffFactor: 1
 				};
 				return {
 					model,
-					...resolvedOptions
+					...options
 				};
 			}
 		}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-retry",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "AI SDK Retry",
   "main": "./dist/index.js",
   "module": "./dist/index.js",