ai-retry 0.4.0 → 0.4.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/dist/index.d.ts CHANGED
@@ -1,4 +1,6 @@
1
1
  import { EmbeddingModelV2, EmbeddingModelV2CallOptions, EmbeddingModelV2Embed, LanguageModelV2, LanguageModelV2Generate, LanguageModelV2Stream, Retries, RetryAttempt, RetryContext, RetryErrorAttempt, RetryModel, RetryResultAttempt, Retryable, RetryableModelOptions } from "./types-BrJaHkFh.js";
2
+ import * as _ai_sdk_provider0 from "@ai-sdk/provider";
3
+ import { LanguageModelV2StreamPart } from "@ai-sdk/provider";
2
4
 
3
5
  //#region src/create-retryable-model.d.ts
4
6
  declare function createRetryable<MODEL extends LanguageModelV2>(options: RetryableModelOptions<MODEL>): LanguageModelV2;
@@ -10,4 +12,56 @@ declare function createRetryable<MODEL extends EmbeddingModelV2>(options: Retrya
10
12
  */
11
13
  declare const getModelKey: (model: LanguageModelV2 | EmbeddingModelV2) => string;
12
14
  //#endregion
13
- export { EmbeddingModelV2, EmbeddingModelV2CallOptions, EmbeddingModelV2Embed, LanguageModelV2, LanguageModelV2Generate, LanguageModelV2Stream, Retries, RetryAttempt, RetryContext, RetryErrorAttempt, RetryModel, RetryResultAttempt, Retryable, RetryableModelOptions, createRetryable, getModelKey };
15
+ //#region src/utils.d.ts
16
+ declare const isObject: (value: unknown) => value is Record<string, unknown>;
17
+ declare const isString: (value: unknown) => value is string;
18
+ declare const isStreamResult: (result: LanguageModelV2Generate | LanguageModelV2Stream) => result is LanguageModelV2Stream;
19
+ declare const isGenerateResult: (result: LanguageModelV2Generate | LanguageModelV2Stream) => result is LanguageModelV2Generate;
20
+ /**
21
+ * Type guard to check if a retry attempt is an error attempt
22
+ */
23
+ declare function isErrorAttempt(attempt: RetryAttempt<any>): attempt is RetryErrorAttempt<any>;
24
+ /**
25
+ * Type guard to check if a retry attempt is a result attempt
26
+ */
27
+ declare function isResultAttempt(attempt: RetryAttempt<any>): attempt is RetryResultAttempt;
28
+ /**
29
+ * Check if a stream part is a content part (e.g., text delta, reasoning delta, source, tool call, tool result).
30
+ * These types are also emitted by `onChunk` callbacks.
31
+ * @see https://github.com/vercel/ai/blob/1fe4bd4144bff927f5319d9d206e782a73979ccb/packages/ai/src/generate-text/stream-text.ts#L686-L697
32
+ */
33
+ declare const isStreamContentPart: (part: LanguageModelV2StreamPart) => part is _ai_sdk_provider0.LanguageModelV2Source | _ai_sdk_provider0.LanguageModelV2ToolCall | {
34
+ type: "tool-result";
35
+ toolCallId: string;
36
+ toolName: string;
37
+ result: unknown;
38
+ isError?: boolean;
39
+ providerExecuted?: boolean;
40
+ providerMetadata?: _ai_sdk_provider0.SharedV2ProviderMetadata;
41
+ } | {
42
+ type: "text-delta";
43
+ id: string;
44
+ providerMetadata?: _ai_sdk_provider0.SharedV2ProviderMetadata;
45
+ delta: string;
46
+ } | {
47
+ type: "reasoning-delta";
48
+ id: string;
49
+ providerMetadata?: _ai_sdk_provider0.SharedV2ProviderMetadata;
50
+ delta: string;
51
+ } | {
52
+ type: "tool-input-start";
53
+ id: string;
54
+ toolName: string;
55
+ providerMetadata?: _ai_sdk_provider0.SharedV2ProviderMetadata;
56
+ providerExecuted?: boolean;
57
+ } | {
58
+ type: "tool-input-delta";
59
+ id: string;
60
+ delta: string;
61
+ providerMetadata?: _ai_sdk_provider0.SharedV2ProviderMetadata;
62
+ } | {
63
+ type: "raw";
64
+ rawValue: unknown;
65
+ };
66
+ //#endregion
67
+ export { EmbeddingModelV2, EmbeddingModelV2CallOptions, EmbeddingModelV2Embed, LanguageModelV2, LanguageModelV2Generate, LanguageModelV2Stream, Retries, RetryAttempt, RetryContext, RetryErrorAttempt, RetryModel, RetryResultAttempt, Retryable, RetryableModelOptions, createRetryable, getModelKey, isErrorAttempt, isGenerateResult, isObject, isResultAttempt, isStreamContentPart, isStreamResult, isString };
package/dist/index.js CHANGED
@@ -1,4 +1,4 @@
1
- import { getModelKey, isErrorAttempt, isGenerateResult, isResultAttempt, isStreamContentPart } from "./utils-DNoBKkQe.js";
1
+ import { getModelKey, isErrorAttempt, isGenerateResult, isObject, isResultAttempt, isStreamContentPart, isStreamResult, isString } from "./utils-Dojn0elD.js";
2
2
  import { delay } from "@ai-sdk/provider-utils";
3
3
  import { getErrorMessage } from "@ai-sdk/provider";
4
4
  import { RetryError } from "ai";
@@ -406,4 +406,4 @@ function createRetryable(options) {
406
406
  }
407
407
 
408
408
  //#endregion
409
- export { createRetryable, getModelKey };
409
+ export { createRetryable, getModelKey, isErrorAttempt, isGenerateResult, isObject, isResultAttempt, isStreamContentPart, isStreamResult, isString };
@@ -1,4 +1,4 @@
1
- import { getModelKey, isErrorAttempt, isObject, isResultAttempt, isString } from "../utils-DNoBKkQe.js";
1
+ import { getModelKey, isErrorAttempt, isObject, isResultAttempt, isString } from "../utils-Dojn0elD.js";
2
2
  import { isAbortError } from "@ai-sdk/provider-utils";
3
3
  import { APICallError } from "ai";
4
4
 
@@ -10,6 +10,7 @@ const getModelKey = (model) => {
10
10
  //#region src/utils.ts
11
11
  const isObject = (value) => typeof value === "object" && value !== null;
12
12
  const isString = (value) => typeof value === "string";
13
+ const isStreamResult = (result) => "stream" in result;
13
14
  const isGenerateResult = (result) => "content" in result;
14
15
  /**
15
16
  * Type guard to check if a retry attempt is an error attempt
@@ -33,4 +34,4 @@ const isStreamContentPart = (part) => {
33
34
  };
34
35
 
35
36
  //#endregion
36
- export { getModelKey, isErrorAttempt, isGenerateResult, isObject, isResultAttempt, isStreamContentPart, isString };
37
+ export { getModelKey, isErrorAttempt, isGenerateResult, isObject, isResultAttempt, isStreamContentPart, isStreamResult, isString };
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "ai-retry",
3
- "version": "0.4.0",
3
+ "version": "0.4.1",
4
4
  "description": "AI SDK Retry",
5
5
  "main": "./dist/index.js",
6
6
  "module": "./dist/index.js",