ai-retry 0.9.0 → 0.10.1

package/README.md

@@ -391,6 +391,32 @@ By default, if a [`retry-after-ms`](https://learn.microsoft.com/en-us/azure/ai-f
 
 ### Options
 
+#### Disabling Retries
+
+You can disable retries entirely, which is useful for testing or specific environments. When disabled, the base model will execute directly without any retry logic.
+
+```typescript
+const retryableModel = createRetryable({
+  model: openai('gpt-4'), // Base model
+  retries: [/* ... */],
+  disabled: true, // Retries are completely disabled
+});
+
+// Or disable based on environment
+const retryableModel = createRetryable({
+  model: openai('gpt-4'), // Base model
+  retries: [/* ... */],
+  disabled: process.env.NODE_ENV === 'test', // Disable in test environment
+});
+
+// Or use a function for dynamic control
+const retryableModel = createRetryable({
+  model: openai('gpt-4'), // Base model
+  retries: [/* ... */],
+  disabled: () => !featureFlags.isEnabled('ai-retries'), // Check feature flag
+});
+```
+
 #### Retry Delays
 
 You can delay retries with an optional exponential backoff. The delay respects abort signals, so requests can still be cancelled during the delay period.
@@ -548,11 +574,19 @@ Creates a retryable model that works with both language models and embedding mod
 interface RetryableModelOptions<MODEL extends LanguageModelV2 | EmbeddingModelV2> {
   model: MODEL;
   retries: Array<Retryable<MODEL> | MODEL>;
+  disabled?: boolean | (() => boolean);
   onError?: (context: RetryContext<MODEL>) => void;
   onRetry?: (context: RetryContext<MODEL>) => void;
 }
 ```
 
+**Options:**
+- `model`: The base model to use for the initial request.
+- `retries`: Array of retryables (functions, models, or retry objects) to attempt on failure.
+- `disabled`: Disable all retry logic. Can be a boolean or function returning boolean. Default: `false` (retries enabled).
+- `onError`: Callback invoked when an error occurs.
+- `onRetry`: Callback invoked before attempting a retry.
+
 #### `Retryable`
 
 A `Retryable` is a function that receives a `RetryContext` with the current error or result and model and all previous attempts.

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { a as LanguageModelV2Generate, c as Retry, d as RetryErrorAttempt, f as RetryResultAttempt, h as RetryableOptions, i as LanguageModelV2, l as RetryAttempt, m as RetryableModelOptions, n as EmbeddingModelV2CallOptions, o as LanguageModelV2Stream, p as Retryable, r as EmbeddingModelV2Embed, s as Retries, t as EmbeddingModelV2, u as RetryContext } from "./types-BuPozWMn.js";
+import { a as LanguageModelV2Generate, c as Retry, d as RetryErrorAttempt, f as RetryResultAttempt, h as RetryableOptions, i as LanguageModelV2, l as RetryAttempt, m as RetryableModelOptions, n as EmbeddingModelV2CallOptions, o as LanguageModelV2Stream, p as Retryable, r as EmbeddingModelV2Embed, s as Retries, t as EmbeddingModelV2, u as RetryContext } from "./types-CphTxZ_b.js";
 import * as _ai_sdk_provider0 from "@ai-sdk/provider";
 import { LanguageModelV2 as LanguageModelV2$1, LanguageModelV2StreamPart } from "@ai-sdk/provider";
 

package/dist/index.js

@@ -105,6 +105,13 @@ var RetryableEmbeddingModel = class {
 		this.currentModel = options.model;
 	}
 	/**
+	* Check if retries are disabled
+	*/
+	isDisabled() {
+		if (this.options.disabled === void 0) return false;
+		return typeof this.options.disabled === "function" ? this.options.disabled() : this.options.disabled;
+	}
+	/**
 	* Execute a function with retry logic for handling errors
 	*/
 	async withRetry(input) {
@@ -197,6 +204,10 @@ var RetryableEmbeddingModel = class {
 		* Always start with the original model
 		*/
 		this.currentModel = this.baseModel;
+		/**
+		* If retries are disabled, bypass retry machinery entirely
+		*/
+		if (this.isDisabled()) return this.currentModel.doEmbed(options);
 		const { result } = await this.withRetry({
 			fn: async (currentRetry) => {
 				const callOptions = {
@@ -234,6 +245,13 @@ var RetryableLanguageModel = class {
 		this.currentModel = options.model;
 	}
 	/**
+	* Check if retries are disabled
+	*/
+	isDisabled() {
+		if (this.options.disabled === void 0) return false;
+		return typeof this.options.disabled === "function" ? this.options.disabled() : this.options.disabled;
+	}
+	/**
 	* Execute a function with retry logic for handling errors
 	*/
 	async withRetry(input) {
@@ -371,6 +389,10 @@ var RetryableLanguageModel = class {
 		* Always start with the original model
 		*/
 		this.currentModel = this.baseModel;
+		/**
+		* If retries are disabled, bypass retry machinery entirely
+		*/
+		if (this.isDisabled()) return this.currentModel.doGenerate(options);
 		const { result } = await this.withRetry({
 			fn: async (currentRetry) => {
 				const callOptions = {
@@ -390,6 +412,10 @@ var RetryableLanguageModel = class {
 		*/
 		this.currentModel = this.baseModel;
 		/**
+		* If retries are disabled, bypass retry machinery entirely
+		*/
+		if (this.isDisabled()) return this.currentModel.doStream(options);
+		/**
 		* Perform the initial call to doStream with retry logic to handle errors before any data is streamed.
 		*/
 		let { result, attempts } = await this.withRetry({

package/dist/retryables/index.d.ts

@@ -1,4 +1,4 @@
-import { h as RetryableOptions, i as LanguageModelV2, p as Retryable, t as EmbeddingModelV2 } from "../types-BuPozWMn.js";
+import { h as RetryableOptions, i as LanguageModelV2, p as Retryable, t as EmbeddingModelV2 } from "../types-CphTxZ_b.js";
 
 //#region src/retryables/content-filter-triggered.d.ts
 

package/dist/{types-BuPozWMn.d.ts → types-CphTxZ_b.d.ts}

@@ -9,6 +9,7 @@ type EmbeddingModelV2$1<VALUE = any> = EmbeddingModelV2<VALUE>;
 interface RetryableModelOptions<MODEL extends LanguageModelV2$1 | EmbeddingModelV2$1> {
   model: MODEL;
   retries: Retries<MODEL>;
+  disabled?: boolean | (() => boolean);
   onError?: (context: RetryContext<MODEL>) => void;
   onRetry?: (context: RetryContext<MODEL>) => void;
 }

package/package.json

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