ai-retry 1.0.0-beta.0 → 1.0.0-beta.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/README.md CHANGED
@@ -24,6 +24,9 @@ Install the beta version of `ai-retry` to support AI SDK v6.
24
24
  > [!NOTE]
25
25
  > AI SDK v6 is still in beta, so break changes are expected.
26
26
 
27
+ > [!NOTE]
28
+ > AI SDK v6 support is available in the [v6 branch](https://github.com/zirkelc/ai-retry/tree/v6).
29
+
27
30
  ```bash
28
31
  npm install ai-retry@beta
29
32
  ```
@@ -94,6 +97,41 @@ const result = await embed({
94
97
  console.log(result.embedding);
95
98
  ```
96
99
 
100
+ #### Vercel AI Gateway
101
+
102
+ You can use `ai-retry` with Vercel AI Gateway by providing the model as a string. Internally, the model will be resolved with the default `gateway` [provider instance](https://ai-sdk.dev/providers/ai-sdk-providers/ai-gateway#provider-instance) from AI SDK.
103
+
104
+ ```typescript
105
+ import { gateway } from 'ai';
106
+ import { createRetryable } from 'ai-retry';
107
+
108
+ const retryableModel = createRetryable({
109
+ model: 'openai/gpt-5',
110
+ retries: [
111
+ 'anthropic/claude-sonnet-4'
112
+ ]
113
+ });
114
+
115
+ // Is the same as:
116
+ const retryableModel = createRetryable({
117
+ model: gateway('openai/gpt-5'),
118
+ retries: [
119
+ gateway('anthropic/claude-sonnet-4')
120
+ ]
121
+ });
122
+ ```
123
+
124
+ By default, the `gateway` provider resolves model strings as language models. If you want to use an embedding model, you need to use the `textEmbeddingModel` method.
125
+
126
+ ```typescript
127
+ import { gateway } from 'ai';
128
+ import { createRetryable } from 'ai-retry';
129
+
130
+ const retryableModel = createRetryable({
131
+ model: gateway.textEmbeddingModel('openai/text-embedding-3-large'),
132
+ });
133
+ ```
134
+
97
135
  ### Retryables
98
136
 
99
137
  The objects passed to the `retries` are called retryables and control the retry behavior. We can distinguish between two types of retryables:
package/dist/index.d.mts CHANGED
@@ -1,7 +1,10 @@
1
- import { _ as Retryable, a as LanguageModel, c as LanguageModelStream, d as Retries, f as Retry, g as RetryResultAttempt, h as RetryErrorAttempt, i as EmbeddingModelV3, l as LanguageModelStreamPart, m as RetryContext, n as EmbeddingModelCallOptions, o as LanguageModelCallOptions, p as RetryAttempt, r as EmbeddingModelEmbed, s as LanguageModelGenerate, t as EmbeddingModel, u as LanguageModelV3, v as RetryableModelOptions, y as RetryableOptions } from "./types-D7Pm83wM.mjs";
1
+ import { S as RetryableOptions, _ as RetryContext, a as LanguageModel, b as Retryable, c as LanguageModelStream, d as ResolvableLanguageModel, f as ResolvableModel, g as RetryAttempt, h as Retry, i as GatewayLanguageModelId, l as LanguageModelStreamPart, m as Retries, n as EmbeddingModelCallOptions, o as LanguageModelCallOptions, p as ResolvedModel, r as EmbeddingModelEmbed, s as LanguageModelGenerate, t as EmbeddingModel, u as ProviderOptions, v as RetryErrorAttempt, x as RetryableModelOptions, y as RetryResultAttempt } from "./types-D7G-2JLh.mjs";
2
2
  import * as _ai_sdk_provider0 from "@ai-sdk/provider";
3
3
 
4
4
  //#region src/create-retryable-model.d.ts
5
+ declare function createRetryable<MODEL extends LanguageModel>(options: Omit<RetryableModelOptions<LanguageModel>, 'model'> & {
6
+ model: GatewayLanguageModelId;
7
+ }): LanguageModel;
5
8
  declare function createRetryable<MODEL extends LanguageModel>(options: RetryableModelOptions<MODEL>): LanguageModel;
6
9
  declare function createRetryable<MODEL extends EmbeddingModel>(options: RetryableModelOptions<MODEL>): EmbeddingModel;
7
10
  //#endregion
@@ -59,9 +62,5 @@ declare const isStreamContentPart: (part: LanguageModelStreamPart) => part is _a
59
62
  type: "raw";
60
63
  rawValue: unknown;
61
64
  };
62
- /**
63
- * Type guard to check if a value is a Retry object (has a model property with a MODEL)
64
- */
65
- declare const isRetry: <MODEL extends LanguageModel | EmbeddingModel>(value: unknown) => value is Retry<MODEL>;
66
65
  //#endregion
67
- export { EmbeddingModel, EmbeddingModelCallOptions, EmbeddingModelEmbed, EmbeddingModelV3, LanguageModel, LanguageModelCallOptions, LanguageModelGenerate, LanguageModelStream, LanguageModelStreamPart, LanguageModelV3, Retries, Retry, RetryAttempt, RetryContext, RetryErrorAttempt, RetryResultAttempt, Retryable, RetryableModelOptions, RetryableOptions, createRetryable, getModelKey, isEmbeddingModel, isErrorAttempt, isGenerateResult, isLanguageModel, isModel, isObject, isResultAttempt, isRetry, isStreamContentPart, isStreamResult, isString };
66
+ export { EmbeddingModel, EmbeddingModelCallOptions, EmbeddingModelEmbed, GatewayLanguageModelId, LanguageModel, LanguageModelCallOptions, LanguageModelGenerate, LanguageModelStream, LanguageModelStreamPart, ProviderOptions, ResolvableLanguageModel, ResolvableModel, ResolvedModel, Retries, Retry, RetryAttempt, RetryContext, RetryErrorAttempt, RetryResultAttempt, Retryable, RetryableModelOptions, RetryableOptions, createRetryable, getModelKey, isEmbeddingModel, isErrorAttempt, isGenerateResult, isLanguageModel, isModel, isObject, isResultAttempt, isStreamContentPart, isStreamResult, isString };
package/dist/index.mjs CHANGED
@@ -1,7 +1,7 @@
1
- import { a as isModel, c as isRetry, d as isString, i as isLanguageModel, l as isStreamContentPart, n as isErrorAttempt, o as isObject, r as isGenerateResult, s as isResultAttempt, t as isEmbeddingModel, u as isStreamResult } from "./utils-BnNgdhF6.mjs";
1
+ import { a as isModel, c as isStreamContentPart, i as isLanguageModel, l as isStreamResult, n as isErrorAttempt, o as isObject, r as isGenerateResult, s as isResultAttempt, t as isEmbeddingModel, u as isString } from "./utils-BXFlA2mh.mjs";
2
+ import { RetryError, gateway } from "ai";
2
3
  import { delay } from "@ai-sdk/provider-utils";
3
4
  import { getErrorMessage } from "@ai-sdk/provider";
4
- import { RetryError } from "ai";
5
5
 
6
6
  //#region src/calculate-exponential-backoff.ts
7
7
  /**
@@ -27,6 +27,15 @@ function countModelAttempts(model, attempts) {
27
27
  return attempts.filter((a) => getModelKey(a.model) === modelKey).length;
28
28
  }
29
29
 
30
+ //#endregion
31
+ //#region src/resolve-model.ts
32
+ /**
33
+ * Resolve a model string via the AI SDK Gateway to a modelinstance
34
+ */
35
+ function resolveModel(model) {
36
+ return isModel(model) ? model : gateway(model);
37
+ }
38
+
30
39
  //#endregion
31
40
  //#region src/find-retry-model.ts
32
41
  /**
@@ -45,13 +54,19 @@ async function findRetryModel(retries, context) {
45
54
  for (const retry of applicableRetries) {
46
55
  let retryModel;
47
56
  if (typeof retry === "function") retryModel = await retry(context);
48
- else if (isRetry(retry)) retryModel = retry;
57
+ else if (isObject(retry) && "model" in retry) retryModel = retry;
49
58
  else retryModel = { model: retry };
50
59
  if (retryModel) {
60
+ /**
61
+ * The model can be string or an instance.
62
+ * If it is a string, we need to resolve it to an instance.
63
+ */
64
+ const modelValue = retryModel.model;
65
+ const resolvedModel = resolveModel(modelValue);
51
66
  /**
52
67
  * The model key uniquely identifies a model instance (provider + modelId)
53
68
  */
54
- const retryModelKey = getModelKey(retryModel.model);
69
+ const retryModelKey = getModelKey(resolvedModel);
55
70
  /**
56
71
  * Find all attempts with the same model
57
72
  */
@@ -60,7 +75,10 @@ async function findRetryModel(retries, context) {
60
75
  /**
61
76
  * Check if the model can still be retried based on maxAttempts
62
77
  */
63
- if (retryAttempts.length < maxAttempts) return retryModel;
78
+ if (retryAttempts.length < maxAttempts) return {
79
+ ...retryModel,
80
+ model: resolvedModel
81
+ };
64
82
  }
65
83
  }
66
84
  }
@@ -513,9 +531,16 @@ var RetryableLanguageModel = class {
513
531
  //#endregion
514
532
  //#region src/create-retryable-model.ts
515
533
  function createRetryable(options) {
516
- if ("doEmbed" in options.model) return new RetryableEmbeddingModel(options);
517
- return new RetryableLanguageModel(options);
534
+ const model = isModel(options.model) ? options.model : gateway(options.model);
535
+ if (isEmbeddingModel(model)) return new RetryableEmbeddingModel({
536
+ ...options,
537
+ model
538
+ });
539
+ return new RetryableLanguageModel({
540
+ ...options,
541
+ model
542
+ });
518
543
  }
519
544
 
520
545
  //#endregion
521
- export { createRetryable, getModelKey, isEmbeddingModel, isErrorAttempt, isGenerateResult, isLanguageModel, isModel, isObject, isResultAttempt, isRetry, isStreamContentPart, isStreamResult, isString };
546
+ export { createRetryable, getModelKey, isEmbeddingModel, isErrorAttempt, isGenerateResult, isLanguageModel, isModel, isObject, isResultAttempt, isStreamContentPart, isStreamResult, isString };
@@ -1,17 +1,17 @@
1
- import { _ as Retryable, a as LanguageModel, t as EmbeddingModel, y as RetryableOptions } from "../types-D7Pm83wM.mjs";
1
+ import { S as RetryableOptions, b as Retryable, d as ResolvableLanguageModel, t as EmbeddingModel } from "../types-D7G-2JLh.mjs";
2
2
 
3
3
  //#region src/retryables/content-filter-triggered.d.ts
4
4
 
5
5
  /**
6
6
  * Fallback to a different model if the content filter was triggered.
7
7
  */
8
- declare function contentFilterTriggered<MODEL extends LanguageModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
8
+ declare function contentFilterTriggered<MODEL extends ResolvableLanguageModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
9
9
  //#endregion
10
10
  //#region src/retryables/request-not-retryable.d.ts
11
11
  /**
12
12
  * Fallback to a different model if the error is non-retryable.
13
13
  */
14
- declare function requestNotRetryable<MODEL extends LanguageModel | EmbeddingModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
14
+ declare function requestNotRetryable<MODEL extends ResolvableLanguageModel | EmbeddingModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
15
15
  //#endregion
16
16
  //#region src/retryables/request-timeout.d.ts
17
17
  /**
@@ -19,7 +19,7 @@ declare function requestNotRetryable<MODEL extends LanguageModel | EmbeddingMode
19
19
  * Use in combination with the `abortSignal` option.
20
20
  * If no timeout is specified, a default of 60 seconds is used.
21
21
  */
22
- declare function requestTimeout<MODEL extends LanguageModel | EmbeddingModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
22
+ declare function requestTimeout<MODEL extends ResolvableLanguageModel | EmbeddingModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
23
23
  //#endregion
24
24
  //#region src/retryables/retry-after-delay.d.ts
25
25
  /**
@@ -27,7 +27,7 @@ declare function requestTimeout<MODEL extends LanguageModel | EmbeddingModel>(mo
27
27
  * Uses the `Retry-After` or `Retry-After-Ms` headers if present.
28
28
  * Otherwise uses the specified `delay` and `backoffFactor` if provided.
29
29
  */
30
- declare function retryAfterDelay<MODEL extends LanguageModel | EmbeddingModel>(options: RetryableOptions<MODEL>): Retryable<MODEL>;
30
+ declare function retryAfterDelay<MODEL extends ResolvableLanguageModel | EmbeddingModel>(options: RetryableOptions<MODEL>): Retryable<MODEL>;
31
31
  //#endregion
32
32
  //#region src/retryables/service-overloaded.d.ts
33
33
  /**
@@ -37,13 +37,13 @@ declare function retryAfterDelay<MODEL extends LanguageModel | EmbeddingModel>(o
37
37
  * - Response with `type: "overloaded_error"`
38
38
  * - Response with a `message` containing "overloaded"
39
39
  */
40
- declare function serviceOverloaded<MODEL extends LanguageModel | EmbeddingModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
40
+ declare function serviceOverloaded<MODEL extends ResolvableLanguageModel | EmbeddingModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
41
41
  //#endregion
42
42
  //#region src/retryables/service-unavailable.d.ts
43
43
  /**
44
44
  * Fallback to a different model if the provider returns a service unavailable error.
45
45
  * This retryable handles HTTP status code 503 (Service Unavailable).
46
46
  */
47
- declare function serviceUnavailable<MODEL extends LanguageModel | EmbeddingModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
47
+ declare function serviceUnavailable<MODEL extends ResolvableLanguageModel | EmbeddingModel>(model: MODEL, options?: RetryableOptions<MODEL>): Retryable<MODEL>;
48
48
  //#endregion
49
49
  export { contentFilterTriggered, requestNotRetryable, requestTimeout, retryAfterDelay, serviceOverloaded, serviceUnavailable };
@@ -1,6 +1,6 @@
1
- import { d as isString, n as isErrorAttempt, o as isObject, s as isResultAttempt } from "../utils-BnNgdhF6.mjs";
2
- import { isAbortError } from "@ai-sdk/provider-utils";
1
+ import { n as isErrorAttempt, o as isObject, s as isResultAttempt, u as isString } from "../utils-BXFlA2mh.mjs";
3
2
  import { APICallError } from "ai";
3
+ import { isAbortError } from "@ai-sdk/provider-utils";
4
4
 
5
5
  //#region src/retryables/content-filter-triggered.ts
6
6
  /**
@@ -0,0 +1,90 @@
1
+ import { gateway } from "ai";
2
+ import { EmbeddingModelV3, LanguageModelV3, LanguageModelV3CallOptions, LanguageModelV3StreamPart, SharedV3ProviderOptions } from "@ai-sdk/provider";
3
+
4
+ //#region src/types.d.ts
5
+ type Literals<T> = T extends string ? string extends T ? never : T : never;
6
+ type LanguageModel = LanguageModelV3;
7
+ type EmbeddingModel<VALUE = any> = EmbeddingModelV3<VALUE>;
8
+ type LanguageModelCallOptions = LanguageModelV3CallOptions;
9
+ type LanguageModelStreamPart = LanguageModelV3StreamPart;
10
+ type ProviderOptions = SharedV3ProviderOptions;
11
+ type GatewayLanguageModelId = Parameters<(typeof gateway)['languageModel']>[0];
12
+ type ResolvableLanguageModel = LanguageModel | Literals<GatewayLanguageModelId>;
13
+ type ResolvableModel<MODEL extends LanguageModel | EmbeddingModel> = MODEL extends LanguageModel ? ResolvableLanguageModel : EmbeddingModel;
14
+ type ResolvedModel<MODEL extends ResolvableLanguageModel | EmbeddingModel> = MODEL extends ResolvableLanguageModel ? LanguageModel : EmbeddingModel;
15
+ /**
16
+ * Options for creating a retryable model.
17
+ */
18
+ interface RetryableModelOptions<MODEL extends LanguageModel | EmbeddingModel> {
19
+ model: MODEL;
20
+ retries: Retries<MODEL>;
21
+ disabled?: boolean | (() => boolean);
22
+ onError?: (context: RetryContext<MODEL>) => void;
23
+ onRetry?: (context: RetryContext<MODEL>) => void;
24
+ }
25
+ /**
26
+ * The context provided to Retryables with the current attempt and all previous attempts.
27
+ */
28
+ type RetryContext<MODEL extends ResolvableLanguageModel | EmbeddingModel> = {
29
+ /**
30
+ * Current attempt that caused the retry
31
+ */
32
+ current: RetryAttempt<ResolvedModel<MODEL>>;
33
+ /**
34
+ * All attempts made so far, including the current one
35
+ */
36
+ attempts: Array<RetryAttempt<ResolvedModel<MODEL>>>;
37
+ };
38
+ /**
39
+ * A retry attempt with an error
40
+ */
41
+ type RetryErrorAttempt<MODEL extends LanguageModel | EmbeddingModel> = {
42
+ type: 'error';
43
+ error: unknown;
44
+ result?: undefined;
45
+ model: MODEL;
46
+ };
47
+ /**
48
+ * A retry attempt with a successful result
49
+ */
50
+ type RetryResultAttempt = {
51
+ type: 'result';
52
+ result: LanguageModelGenerate;
53
+ error?: undefined;
54
+ model: LanguageModel;
55
+ };
56
+ /**
57
+ * A retry attempt with either an error or a result and the model used
58
+ */
59
+ type RetryAttempt<MODEL extends LanguageModel | EmbeddingModel> = RetryErrorAttempt<MODEL> | RetryResultAttempt;
60
+ /**
61
+ * A model to retry with and the maximum number of attempts for that model.
62
+ *
63
+ * The model can be:
64
+ * - The exact MODEL type (instance)
65
+ * - A gateway string literal (for LanguageModel only)
66
+ * - A ResolvableModel<MODEL> (for compatibility with plain model arrays)
67
+ *
68
+ * This flexible approach allows retryable functions to return the exact model type
69
+ * they received without type assertions, while still supporting string-based gateway models.
70
+ */
71
+ type Retry<MODEL extends ResolvableLanguageModel | EmbeddingModel> = {
72
+ model: MODEL;
73
+ maxAttempts?: number;
74
+ delay?: number;
75
+ backoffFactor?: number;
76
+ providerOptions?: ProviderOptions;
77
+ timeout?: number;
78
+ };
79
+ /**
80
+ * A function that determines whether to retry with a different model based on the current attempt and all previous attempts.
81
+ */
82
+ type Retryable<MODEL extends ResolvableLanguageModel | EmbeddingModel> = (context: RetryContext<MODEL>) => Retry<MODEL> | Promise<Retry<MODEL> | undefined> | undefined;
83
+ type Retries<MODEL extends LanguageModel | EmbeddingModel> = Array<Retryable<ResolvableModel<MODEL>> | Retry<ResolvableModel<MODEL>> | ResolvableModel<MODEL>>;
84
+ type RetryableOptions<MODEL extends ResolvableLanguageModel | EmbeddingModel> = Partial<Omit<Retry<MODEL>, 'model'>>;
85
+ type LanguageModelGenerate = Awaited<ReturnType<LanguageModel['doGenerate']>>;
86
+ type LanguageModelStream = Awaited<ReturnType<LanguageModel['doStream']>>;
87
+ type EmbeddingModelCallOptions<VALUE> = Parameters<EmbeddingModel<VALUE>['doEmbed']>[0];
88
+ type EmbeddingModelEmbed<VALUE = any> = Awaited<ReturnType<EmbeddingModel<VALUE>['doEmbed']>>;
89
+ //#endregion
90
+ export { RetryableOptions as S, RetryContext as _, LanguageModel as a, Retryable as b, LanguageModelStream as c, ResolvableLanguageModel as d, ResolvableModel as f, RetryAttempt as g, Retry as h, GatewayLanguageModelId as i, LanguageModelStreamPart as l, Retries as m, EmbeddingModelCallOptions as n, LanguageModelCallOptions as o, ResolvedModel as p, EmbeddingModelEmbed as r, LanguageModelGenerate as s, EmbeddingModel as t, ProviderOptions as u, RetryErrorAttempt as v, RetryableModelOptions as x, RetryResultAttempt as y };
@@ -2,8 +2,8 @@
2
2
  const isObject = (value) => typeof value === "object" && value !== null;
3
3
  const isString = (value) => typeof value === "string";
4
4
  const isModel = (model) => isLanguageModel(model) || isEmbeddingModel(model);
5
- const isLanguageModel = (model) => isObject(model) && "provider" in model && "modelId" in model && "specificationVersion" in model && model.specificationVersion === "v3";
6
- const isEmbeddingModel = (model) => isObject(model) && "provider" in model && "modelId" in model && "specificationVersion" in model && model.specificationVersion === "v3";
5
+ const isLanguageModel = (model) => isObject(model) && "provider" in model && "modelId" in model && "specificationVersion" in model && "doGenerate" in model && model.specificationVersion === "v3";
6
+ const isEmbeddingModel = (model) => isObject(model) && "provider" in model && "modelId" in model && "specificationVersion" in model && "doEmbed" in model && model.specificationVersion === "v3";
7
7
  const isStreamResult = (result) => "stream" in result;
8
8
  const isGenerateResult = (result) => "content" in result;
9
9
  /**
@@ -26,10 +26,6 @@ function isResultAttempt(attempt) {
26
26
  const isStreamContentPart = (part) => {
27
27
  return part.type === "text-delta" || part.type === "reasoning-delta" || part.type === "source" || part.type === "tool-call" || part.type === "tool-result" || part.type === "tool-input-start" || part.type === "tool-input-delta" || part.type === "raw";
28
28
  };
29
- /**
30
- * Type guard to check if a value is a Retry object (has a model property with a MODEL)
31
- */
32
- const isRetry = (value) => isObject(value) && "model" in value && isModel(value.model);
33
29
 
34
30
  //#endregion
35
- export { isModel as a, isRetry as c, isString as d, isLanguageModel as i, isStreamContentPart as l, isErrorAttempt as n, isObject as o, isGenerateResult as r, isResultAttempt as s, isEmbeddingModel as t, isStreamResult as u };
31
+ export { isModel as a, isStreamContentPart as c, isLanguageModel as i, isStreamResult as l, isErrorAttempt as n, isObject as o, isGenerateResult as r, isResultAttempt as s, isEmbeddingModel as t, isString as u };
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "ai-retry",
3
- "version": "1.0.0-beta.0",
3
+ "version": "1.0.0-beta.1",
4
4
  "description": "AI SDK Retry",
5
5
  "main": "./dist/index.mjs",
6
6
  "module": "./dist/index.mjs",
@@ -35,6 +35,7 @@
35
35
  "devDependencies": {
36
36
  "@ai-sdk/anthropic": "3.0.0-beta.59",
37
37
  "@ai-sdk/azure": "3.0.0-beta.68",
38
+ "@ai-sdk/gateway": "2.0.0-beta.66",
38
39
  "@ai-sdk/groq": "3.0.0-beta.36",
39
40
  "@ai-sdk/openai": "3.0.0-beta.66",
40
41
  "@ai-sdk/test-server": "1.0.0-beta.1",
@@ -49,8 +50,7 @@
49
50
  "publint": "^0.3.15",
50
51
  "tsdown": "^0.16.7",
51
52
  "tsx": "^4.20.6",
52
- "typescript": "^5.9.3",
53
- "unbuild": "^3.6.1",
53
+ "typescript": "^5.9.2",
54
54
  "vitest": "^4.0.14",
55
55
  "zod": "^4.1.13"
56
56
  },
@@ -59,9 +59,9 @@
59
59
  "@ai-sdk/provider-utils": "4.0.0-beta.34"
60
60
  },
61
61
  "scripts": {
62
- "publish:alpha": "pnpm version prerelease --preid alpha && pnpm publish --tag alpha",
63
62
  "build": "tsdown",
64
63
  "test": "vitest",
64
+ "typecheck": "tsc --noEmit",
65
65
  "lint": "biome check . --write"
66
66
  }
67
67
  }
@@ -1,77 +0,0 @@
1
- import { ProviderOptions } from "@ai-sdk/provider-utils";
2
- import { EmbeddingModelV3, LanguageModelV3, LanguageModelV3CallOptions, LanguageModelV3StreamPart } from "@ai-sdk/provider";
3
-
4
- //#region src/types.d.ts
5
- type EmbeddingModelV3$1<VALUE = any> = EmbeddingModelV3<VALUE>;
6
- type EmbeddingModel<VALUE = any> = EmbeddingModelV3<VALUE>;
7
- type LanguageModel = LanguageModelV3;
8
- type LanguageModelCallOptions = LanguageModelV3CallOptions;
9
- type LanguageModelStreamPart = LanguageModelV3StreamPart;
10
- /**
11
- * Options for creating a retryable model.
12
- */
13
- interface RetryableModelOptions<MODEL extends LanguageModel | EmbeddingModel<any>> {
14
- model: MODEL;
15
- retries: Retries<MODEL>;
16
- disabled?: boolean | (() => boolean);
17
- onError?: (context: RetryContext<MODEL>) => void;
18
- onRetry?: (context: RetryContext<MODEL>) => void;
19
- }
20
- /**
21
- * The context provided to Retryables with the current attempt and all previous attempts.
22
- */
23
- type RetryContext<MODEL extends LanguageModel | EmbeddingModel<any>> = {
24
- /**
25
- * Current attempt that caused the retry
26
- */
27
- current: RetryAttempt<MODEL>;
28
- /**
29
- * All attempts made so far, including the current one
30
- */
31
- attempts: Array<RetryAttempt<MODEL>>;
32
- };
33
- /**
34
- * A retry attempt with an error
35
- */
36
- type RetryErrorAttempt<MODEL extends LanguageModel | EmbeddingModel<any>> = {
37
- type: 'error';
38
- error: unknown;
39
- result?: undefined;
40
- model: MODEL;
41
- };
42
- /**
43
- * A retry attempt with a successful result
44
- */
45
- type RetryResultAttempt = {
46
- type: 'result';
47
- result: LanguageModelGenerate;
48
- error?: undefined;
49
- model: LanguageModel;
50
- };
51
- /**
52
- * A retry attempt with either an error or a result and the model used
53
- */
54
- type RetryAttempt<MODEL extends LanguageModel | EmbeddingModel<any>> = RetryErrorAttempt<MODEL> | RetryResultAttempt;
55
- /**
56
- * A model to retry with and the maximum number of attempts for that model.
57
- */
58
- type Retry<MODEL extends LanguageModel | EmbeddingModel<any>> = {
59
- model: MODEL;
60
- maxAttempts?: number;
61
- delay?: number;
62
- backoffFactor?: number;
63
- providerOptions?: ProviderOptions;
64
- timeout?: number;
65
- };
66
- /**
67
- * A function that determines whether to retry with a different model based on the current attempt and all previous attempts.
68
- */
69
- type Retryable<MODEL extends LanguageModel | EmbeddingModel<any>> = (context: RetryContext<MODEL>) => Retry<MODEL> | Promise<Retry<MODEL>> | undefined;
70
- type Retries<MODEL extends LanguageModel | EmbeddingModel<any>> = Array<Retryable<MODEL> | Retry<MODEL> | MODEL>;
71
- type RetryableOptions<MODEL extends LanguageModel | EmbeddingModel<any>> = Partial<Omit<Retry<MODEL>, 'model'>>;
72
- type LanguageModelGenerate = Awaited<ReturnType<LanguageModel['doGenerate']>>;
73
- type LanguageModelStream = Awaited<ReturnType<LanguageModel['doStream']>>;
74
- type EmbeddingModelCallOptions<VALUE> = Parameters<EmbeddingModel<VALUE>['doEmbed']>[0];
75
- type EmbeddingModelEmbed<VALUE> = Awaited<ReturnType<EmbeddingModel<VALUE>['doEmbed']>>;
76
- //#endregion
77
- export { Retryable as _, LanguageModel as a, LanguageModelStream as c, Retries as d, Retry as f, RetryResultAttempt as g, RetryErrorAttempt as h, EmbeddingModelV3$1 as i, LanguageModelStreamPart as l, RetryContext as m, EmbeddingModelCallOptions as n, LanguageModelCallOptions as o, RetryAttempt as p, EmbeddingModelEmbed as r, LanguageModelGenerate as s, EmbeddingModel as t, LanguageModelV3 as u, RetryableModelOptions as v, RetryableOptions as y };