ai 6.0.0-beta.77 → 6.0.0-beta.79

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/CHANGELOG.md CHANGED
@@ -1,5 +1,24 @@
1
1
  # ai
2
2
 
3
+ ## 6.0.0-beta.79
4
+
5
+ ### Patch Changes
6
+
7
+ - d1bdadb: Added experimental_rerank support
8
+ - Updated dependencies [d1bdadb]
9
+ - @ai-sdk/provider@3.0.0-beta.12
10
+ - @ai-sdk/gateway@2.0.0-beta.43
11
+ - @ai-sdk/provider-utils@4.0.0-beta.26
12
+
13
+ ## 6.0.0-beta.78
14
+
15
+ ### Patch Changes
16
+
17
+ - Updated dependencies [4c44a5b]
18
+ - @ai-sdk/provider@3.0.0-beta.11
19
+ - @ai-sdk/gateway@2.0.0-beta.42
20
+ - @ai-sdk/provider-utils@4.0.0-beta.25
21
+
3
22
  ## 6.0.0-beta.77
4
23
 
5
24
  ### Patch Changes
package/dist/index.d.mts CHANGED
@@ -3,7 +3,7 @@ import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
3
3
  import { Tool, InferToolInput, InferToolOutput, AssistantModelMessage, ToolModelMessage, ReasoningPart, FlexibleSchema, InferSchema, ModelMessage, SystemModelMessage, UserModelMessage, ProviderOptions, IdGenerator, ToolCall, MaybePromiseLike, TextPart, FilePart, Resolvable, FetchFunction, DataContent } from '@ai-sdk/provider-utils';
4
4
  export { AssistantContent, AssistantModelMessage, DataContent, FilePart, FlexibleSchema, IdGenerator, ImagePart, InferSchema, InferToolInput, InferToolOutput, ModelMessage, Schema, SystemModelMessage, TextPart, Tool, ToolApprovalRequest, ToolApprovalResponse, ToolCallOptions, ToolCallPart, ToolContent, ToolExecuteFunction, ToolModelMessage, ToolResultPart, UserContent, UserModelMessage, asSchema, createIdGenerator, dynamicTool, generateId, jsonSchema, parseJsonEventStream, tool, zodSchema } from '@ai-sdk/provider-utils';
5
5
  import * as _ai_sdk_provider from '@ai-sdk/provider';
6
- import { EmbeddingModelV3, EmbeddingModelV2, EmbeddingModelV3Embedding, ImageModelV3, ImageModelV3CallWarning, ImageModelV3ProviderMetadata, JSONValue as JSONValue$1, LanguageModelV3, LanguageModelV2, LanguageModelV3FinishReason, LanguageModelV3CallWarning, LanguageModelV3Source, LanguageModelV3Middleware, SharedV3ProviderMetadata, SpeechModelV3, SpeechModelV2, SpeechModelV3CallWarning, TranscriptionModelV3, TranscriptionModelV2, TranscriptionModelV3CallWarning, LanguageModelV3Usage, LanguageModelV3CallOptions, AISDKError, LanguageModelV3ToolCall, JSONSchema7, JSONParseError, TypeValidationError, SharedV3Warning, ProviderV3, ProviderV2, NoSuchModelError } from '@ai-sdk/provider';
6
+ import { EmbeddingModelV3, EmbeddingModelV2, EmbeddingModelV3Embedding, ImageModelV3, ImageModelV3CallWarning, ImageModelV3ProviderMetadata, JSONValue as JSONValue$1, LanguageModelV3, LanguageModelV2, LanguageModelV3FinishReason, LanguageModelV3CallWarning, LanguageModelV3Source, LanguageModelV3Middleware, RerankingModelV3, SharedV3ProviderMetadata, SpeechModelV3, SpeechModelV2, SpeechModelV3CallWarning, TranscriptionModelV3, TranscriptionModelV2, TranscriptionModelV3CallWarning, LanguageModelV3Usage, LanguageModelV3CallOptions, AISDKError, LanguageModelV3ToolCall, JSONSchema7, JSONParseError, TypeValidationError, JSONObject, SharedV3Warning, ProviderV3, ProviderV2, NoSuchModelError } from '@ai-sdk/provider';
7
7
  export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, JSONSchema7, LoadAPIKeyError, LoadSettingError, NoContentGeneratedError, NoSuchModelError, TooManyEmbeddingValuesForCallError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
8
8
  import { ServerResponse } from 'node:http';
9
9
  import { AttributeValue, Tracer } from '@opentelemetry/api';
@@ -116,6 +116,11 @@ type LanguageModelResponseMetadata = {
116
116
  headers?: Record<string, string>;
117
117
  };
118
118
 
119
+ /**
120
+ * Reranking model that is used by the AI SDK Core functions.
121
+ */
122
+ type RerankingModel = RerankingModelV3;
123
+
119
124
  /**
120
125
  * Provider for language, text embedding, and image models.
121
126
  */
@@ -151,6 +156,17 @@ type Provider = {
151
156
  @returns {ImageModel} The image model associated with the id
152
157
  */
153
158
  imageModel(modelId: string): ImageModel;
159
+ /**
160
+ Returns the reranking model with the given id.
161
+ The model id is then passed to the provider function to get the model.
162
+
163
+ @param {string} id - The id of the model to return.
164
+
165
+ @return {RerankingModel<VALUE>} The reranking model associated with the id
166
+
167
+ @throws {NoSuchModelError} If no such model exists.
168
+ */
169
+ rerankingModel(modelId: string): RerankingModel;
154
170
  };
155
171
 
156
172
  /**
@@ -4523,7 +4539,7 @@ interface SpeechResult {
4523
4539
  /**
4524
4540
  Provider metadata from the provider.
4525
4541
  */
4526
- readonly providerMetadata: Record<string, Record<string, JSONValue$1>>;
4542
+ readonly providerMetadata: Record<string, JSONObject>;
4527
4543
  }
4528
4544
 
4529
4545
  /**
@@ -4694,23 +4710,26 @@ declare function wrapProvider({ provider, languageModelMiddleware, }: {
4694
4710
  * @param {Record<string, ImageModel>} [options.imageModels] - A record of image models, where keys are model IDs and values are ImageModel instances.
4695
4711
  * @param {Record<string, TranscriptionModel>} [options.transcriptionModels] - A record of transcription models, where keys are model IDs and values are TranscriptionModel instances.
4696
4712
  * @param {Record<string, SpeechModel>} [options.speechModels] - A record of speech models, where keys are model IDs and values are SpeechModel instances.
4713
+ * @param {Record<string, RerankingModel<string>>} [options.rerankingModels] - A record of reranking models, where keys are model IDs and values are RerankingModel<string> instances.
4697
4714
  * @param {Provider} [options.fallbackProvider] - An optional fallback provider to use when a requested model is not found in the custom provider.
4698
4715
  * @returns {Provider} A Provider object with languageModel, textEmbeddingModel, imageModel, transcriptionModel, and speechModel methods.
4699
4716
  *
4700
4717
  * @throws {NoSuchModelError} Throws when a requested model is not found and no fallback provider is available.
4701
4718
  */
4702
- declare function customProvider<LANGUAGE_MODELS extends Record<string, LanguageModelV3>, EMBEDDING_MODELS extends Record<string, EmbeddingModelV3<string>>, IMAGE_MODELS extends Record<string, ImageModelV3>, TRANSCRIPTION_MODELS extends Record<string, TranscriptionModelV3>, SPEECH_MODELS extends Record<string, SpeechModelV3>>({ languageModels, textEmbeddingModels, imageModels, transcriptionModels, speechModels, fallbackProvider, }: {
4719
+ declare function customProvider<LANGUAGE_MODELS extends Record<string, LanguageModelV3>, EMBEDDING_MODELS extends Record<string, EmbeddingModelV3<string>>, IMAGE_MODELS extends Record<string, ImageModelV3>, TRANSCRIPTION_MODELS extends Record<string, TranscriptionModelV3>, SPEECH_MODELS extends Record<string, SpeechModelV3>, RERANKING_MODELS extends Record<string, RerankingModelV3>>({ languageModels, textEmbeddingModels, imageModels, transcriptionModels, speechModels, rerankingModels, fallbackProvider: fallbackProviderArg, }: {
4703
4720
  languageModels?: LANGUAGE_MODELS;
4704
4721
  textEmbeddingModels?: EMBEDDING_MODELS;
4705
4722
  imageModels?: IMAGE_MODELS;
4706
4723
  transcriptionModels?: TRANSCRIPTION_MODELS;
4707
4724
  speechModels?: SPEECH_MODELS;
4725
+ rerankingModels?: RERANKING_MODELS;
4708
4726
  fallbackProvider?: ProviderV3 | ProviderV2;
4709
4727
  }): ProviderV3 & {
4710
4728
  languageModel(modelId: ExtractModelId<LANGUAGE_MODELS>): LanguageModelV3;
4711
4729
  textEmbeddingModel(modelId: ExtractModelId<EMBEDDING_MODELS>): EmbeddingModelV3<string>;
4712
4730
  imageModel(modelId: ExtractModelId<IMAGE_MODELS>): ImageModelV3;
4713
4731
  transcriptionModel(modelId: ExtractModelId<TRANSCRIPTION_MODELS>): TranscriptionModelV3;
4732
+ rerankingModel(modelId: ExtractModelId<RERANKING_MODELS>): RerankingModelV3;
4714
4733
  speechModel(modelId: ExtractModelId<SPEECH_MODELS>): SpeechModelV3;
4715
4734
  };
4716
4735
  /**
@@ -4726,7 +4745,7 @@ declare class NoSuchProviderError extends NoSuchModelError {
4726
4745
  readonly availableProviders: string[];
4727
4746
  constructor({ modelId, modelType, providerId, availableProviders, message, }: {
4728
4747
  modelId: string;
4729
- modelType: 'languageModel' | 'textEmbeddingModel' | 'imageModel' | 'transcriptionModel' | 'speechModel';
4748
+ modelType: 'languageModel' | 'textEmbeddingModel' | 'imageModel' | 'transcriptionModel' | 'speechModel' | 'rerankingModel';
4730
4749
  providerId: string;
4731
4750
  availableProviders: string[];
4732
4751
  message?: string;
@@ -4746,6 +4765,8 @@ interface ProviderRegistryProvider<PROVIDERS extends Record<string, ProviderV3>
4746
4765
  transcriptionModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${string}` : never): TranscriptionModelV3;
4747
4766
  speechModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${ExtractLiteralUnion<Parameters<NonNullable<PROVIDERS[KEY]['speechModel']>>[0]>}` : never): SpeechModelV3;
4748
4767
  speechModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${string}` : never): SpeechModelV3;
4768
+ rerankingModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${ExtractLiteralUnion<Parameters<NonNullable<PROVIDERS[KEY]['rerankingModel']>>[0]>}` : never): RerankingModelV3;
4769
+ rerankingModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${string}` : never): RerankingModelV3;
4749
4770
  }
4750
4771
  /**
4751
4772
  * Creates a registry for the given providers with optional middleware functionality.
@@ -4768,6 +4789,125 @@ declare function createProviderRegistry<PROVIDERS extends Record<string, Provide
4768
4789
  */
4769
4790
  declare const experimental_createProviderRegistry: typeof createProviderRegistry;
4770
4791
 
4792
+ /**
4793
+ * The result of a `rerank` call.
4794
+ * It contains the original documents, the reranked documents, and additional information.
4795
+ */
4796
+ interface RerankResult<VALUE> {
4797
+ /**
4798
+ * The original documents that were reranked.
4799
+ */
4800
+ readonly originalDocuments: Array<VALUE>;
4801
+ /**
4802
+ * Reranked documents.
4803
+ *
4804
+ * Sorted by relevance score in descending order.
4805
+ *
4806
+ * Can be less than the original documents if there was a topK limit.
4807
+ */
4808
+ readonly rerankedDocuments: Array<VALUE>;
4809
+ /**
4810
+ * The ranking is a list of objects with the original index,
4811
+ * relevance score, and the reranked document.
4812
+ *
4813
+ * Sorted by relevance score in descending order.
4814
+ *
4815
+ * Can be less than the original documents if there was a topK limit.
4816
+ */
4817
+ readonly ranking: Array<{
4818
+ originalIndex: number;
4819
+ relevanceScore: number;
4820
+ document: VALUE;
4821
+ }>;
4822
+ /**
4823
+ * Optional provider-specific metadata.
4824
+ */
4825
+ readonly providerMetadata?: ProviderMetadata;
4826
+ /**
4827
+ * Optional raw response data.
4828
+ */
4829
+ readonly response: {
4830
+ /**
4831
+ * ID for the generated response if the provider sends one.
4832
+ */
4833
+ id?: string;
4834
+ /**
4835
+ * Timestamp of the generated response.
4836
+ */
4837
+ timestamp: Date;
4838
+ /**
4839
+ * The ID of the model that was used to generate the response.
4840
+ */
4841
+ modelId: string;
4842
+ /**
4843
+ * Response headers.
4844
+ */
4845
+ headers?: Record<string, string>;
4846
+ /**
4847
+ * The response body.
4848
+ */
4849
+ body?: unknown;
4850
+ };
4851
+ }
4852
+
4853
+ /**
4854
+ Rerank documents using an reranking model. The type of the value is defined by the reranking model.
4855
+
4856
+ @param model - The Reranking model to use.
4857
+ @param documents - The documents that should be reranking.
4858
+ @param query - The query is a string that represents the query to rerank the documents against.
4859
+ @param topN - Top n documents to rerank.
4860
+
4861
+ @param maxRetries - Maximum number of retries. Set to 0 to disable retries. Default: 2.
4862
+ @param abortSignal - An optional abort signal that can be used to cancel the call.
4863
+ @param headers - Additional HTTP headers to be sent with the request. Only applicable for HTTP-based providers.
4864
+
4865
+ @returns A result object that contains the reranked documents, the reranked indices, and additional information.
4866
+ */
4867
+ declare function rerank<VALUE extends JSONObject | string>({ model, documents, query, topN, maxRetries: maxRetriesArg, abortSignal, headers, providerOptions, experimental_telemetry: telemetry, }: {
4868
+ /**
4869
+ The reranking model to use.
4870
+ */
4871
+ model: RerankingModel;
4872
+ /**
4873
+ * The documents that should be reranked.
4874
+ */
4875
+ documents: Array<VALUE>;
4876
+ /**
4877
+ The query is a string that represents the query to rerank the documents against.
4878
+ */
4879
+ query: string;
4880
+ /**
4881
+ * Number of top documents to return.
4882
+ */
4883
+ topN?: number;
4884
+ /**
4885
+ Maximum number of retries per reranking model call. Set to 0 to disable retries.
4886
+
4887
+ @default 2
4888
+ */
4889
+ maxRetries?: number;
4890
+ /**
4891
+ Abort signal.
4892
+ */
4893
+ abortSignal?: AbortSignal;
4894
+ /**
4895
+ Additional headers to include in the request.
4896
+ Only applicable for HTTP-based providers.
4897
+ */
4898
+ headers?: Record<string, string>;
4899
+ /**
4900
+ * Optional telemetry configuration (experimental).
4901
+ */
4902
+ experimental_telemetry?: TelemetrySettings;
4903
+ /**
4904
+ Additional provider-specific options. They are passed through
4905
+ to the provider from the AI SDK and enable provider-specific
4906
+ functionality that can be fully encapsulated in the provider.
4907
+ */
4908
+ providerOptions?: ProviderOptions;
4909
+ }): Promise<RerankResult<VALUE>>;
4910
+
4771
4911
  declare function createTextStreamResponse({ status, statusText, headers, textStream, }: ResponseInit & {
4772
4912
  textStream: ReadableStream<string>;
4773
4913
  }): Response;
@@ -4825,7 +4965,7 @@ interface TranscriptionResult {
4825
4965
  /**
4826
4966
  Provider metadata from the provider.
4827
4967
  */
4828
- readonly providerMetadata: Record<string, Record<string, JSONValue$1>>;
4968
+ readonly providerMetadata: Record<string, JSONObject>;
4829
4969
  }
4830
4970
 
4831
4971
  /**
@@ -4902,4 +5042,4 @@ declare global {
4902
5042
  var AI_SDK_LOG_WARNINGS: LogWarningsFunction | undefined | false;
4903
5043
  }
4904
5044
 
4905
- export { AbstractChat, Agent, AsyncIterableStream, CallSettings, CallWarning, ChatAddToolApproveResponseFunction, ChatInit, ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback, ChatOnToolCallCallback, ChatRequestOptions, ChatState, ChatStatus, ChatTransport, ChunkDetector, CompletionRequestOptions, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreToolMessage, CoreUserMessage, CreateUIMessage, DataUIPart, DeepPartial, DefaultChatTransport, DownloadError, DynamicToolCall, DynamicToolError, DynamicToolResult, DynamicToolUIPart, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, ErrorHandler, ToolLoopAgent as Experimental_Agent, ToolLoopAgentSettings as Experimental_AgentSettings, DownloadFunction as Experimental_DownloadFunction, GenerateImageResult as Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, InferAgentUIMessage as Experimental_InferAgentUIMessage, LogWarningsFunction as Experimental_LogWarningsFunction, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, Warning as Experimental_Warning, FileUIPart, FinishReason, GenerateObjectResult, GenerateTextOnFinishCallback, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, HttpChatTransport, HttpChatTransportInitOptions, ImageModel, ImageGenerationWarning as ImageModelCallWarning, ImageModelProviderMetadata, ImageModelResponseMetadata, InferAgentUIMessage, InferGenerateOutput, InferStreamOutput, InferUIDataParts, InferUIMessageChunk, InferUITool, InferUITools, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolInputError, JSONValue, JsonToSseTransformStream, LanguageModel, LanguageModelMiddleware, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, MessageConversionError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputGeneratedError, NoOutputSpecifiedError, NoSpeechGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, PrepareReconnectToStreamRequest, PrepareSendMessagesRequest, PrepareStepFunction, PrepareStepResult, Prompt, Provider, ProviderMetadata, ProviderRegistryProvider, ReasoningOutput, ReasoningUIPart, RepairTextFunction, RetryError, SafeValidateUIMessagesResult, SerialJobExecutor, SourceDocumentUIPart, SourceUrlUIPart, SpeechModel, SpeechModelResponseMetadata, SpeechWarning, StaticToolCall, StaticToolError, StaticToolOutputDenied, StaticToolResult, StepResult, StepStartUIPart, StopCondition, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextStreamChatTransport, TextStreamPart, TextUIPart, ToolApprovalRequestOutput, ToolCallRepairError, ToolCallRepairFunction, ToolChoice, ToolLoopAgent, ToolLoopAgentOnFinishCallback, ToolLoopAgentOnStepFinishCallback, ToolLoopAgentSettings, ToolSet, ToolUIPart, TranscriptionModel, TranscriptionModelResponseMetadata, TranscriptionWarning, TypedToolCall, TypedToolError, TypedToolOutputDenied, TypedToolResult, UIDataPartSchemas, UIDataTypes, UIMessage, UIMessageChunk, UIMessagePart, UIMessageStreamOnFinishCallback, UIMessageStreamOptions, UIMessageStreamWriter, UITool, UIToolInvocation, UITools, UI_MESSAGE_STREAM_HEADERS, UnsupportedModelVersionError, UseCompletionOptions, assistantModelMessageSchema, callCompletionApi, consumeStream, convertFileListToFileUIParts, convertToCoreMessages, convertToModelMessages, coreAssistantMessageSchema, coreMessageSchema, coreSystemMessageSchema, coreToolMessageSchema, coreUserMessageSchema, cosineSimilarity, createAgentUIStream, createAgentUIStreamResponse, createProviderRegistry, createTextStreamResponse, createUIMessageStream, createUIMessageStreamResponse, customProvider, defaultSettingsMiddleware, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, generateImage as experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractReasoningMiddleware, generateObject, generateText, getTextFromDataUrl, getToolName, getToolOrDynamicToolName, hasToolCall, isDataUIPart, isDeepEqualData, isFileUIPart, isReasoningUIPart, isTextUIPart, isToolOrDynamicToolUIPart, isToolUIPart, lastAssistantMessageIsCompleteWithApprovalResponses, lastAssistantMessageIsCompleteWithToolCalls, modelMessageSchema, parsePartialJson, pipeAgentUIStreamToResponse, pipeTextStreamToResponse, pipeUIMessageStreamToResponse, pruneMessages, readUIMessageStream, safeValidateUIMessages, simulateReadableStream, simulateStreamingMiddleware, smoothStream, stepCountIs, streamObject, streamText, systemModelMessageSchema, toolModelMessageSchema, uiMessageChunkSchema, userModelMessageSchema, validateUIMessages, wrapLanguageModel, wrapProvider };
5045
+ export { AbstractChat, Agent, AsyncIterableStream, CallSettings, CallWarning, ChatAddToolApproveResponseFunction, ChatInit, ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback, ChatOnToolCallCallback, ChatRequestOptions, ChatState, ChatStatus, ChatTransport, ChunkDetector, CompletionRequestOptions, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreToolMessage, CoreUserMessage, CreateUIMessage, DataUIPart, DeepPartial, DefaultChatTransport, DownloadError, DynamicToolCall, DynamicToolError, DynamicToolResult, DynamicToolUIPart, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, ErrorHandler, ToolLoopAgent as Experimental_Agent, ToolLoopAgentSettings as Experimental_AgentSettings, DownloadFunction as Experimental_DownloadFunction, GenerateImageResult as Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, InferAgentUIMessage as Experimental_InferAgentUIMessage, LogWarningsFunction as Experimental_LogWarningsFunction, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, Warning as Experimental_Warning, FileUIPart, FinishReason, GenerateObjectResult, GenerateTextOnFinishCallback, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, HttpChatTransport, HttpChatTransportInitOptions, ImageModel, ImageGenerationWarning as ImageModelCallWarning, ImageModelProviderMetadata, ImageModelResponseMetadata, InferAgentUIMessage, InferGenerateOutput, InferStreamOutput, InferUIDataParts, InferUIMessageChunk, InferUITool, InferUITools, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolInputError, JSONValue, JsonToSseTransformStream, LanguageModel, LanguageModelMiddleware, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, MessageConversionError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputGeneratedError, NoOutputSpecifiedError, NoSpeechGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, PrepareReconnectToStreamRequest, PrepareSendMessagesRequest, PrepareStepFunction, PrepareStepResult, Prompt, Provider, ProviderMetadata, ProviderRegistryProvider, ReasoningOutput, ReasoningUIPart, RepairTextFunction, RerankResult, RerankingModel, RetryError, SafeValidateUIMessagesResult, SerialJobExecutor, SourceDocumentUIPart, SourceUrlUIPart, SpeechModel, SpeechModelResponseMetadata, SpeechWarning, StaticToolCall, StaticToolError, StaticToolOutputDenied, StaticToolResult, StepResult, StepStartUIPart, StopCondition, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextStreamChatTransport, TextStreamPart, TextUIPart, ToolApprovalRequestOutput, ToolCallRepairError, ToolCallRepairFunction, ToolChoice, ToolLoopAgent, ToolLoopAgentOnFinishCallback, ToolLoopAgentOnStepFinishCallback, ToolLoopAgentSettings, ToolSet, ToolUIPart, TranscriptionModel, TranscriptionModelResponseMetadata, TranscriptionWarning, TypedToolCall, TypedToolError, TypedToolOutputDenied, TypedToolResult, UIDataPartSchemas, UIDataTypes, UIMessage, UIMessageChunk, UIMessagePart, UIMessageStreamOnFinishCallback, UIMessageStreamOptions, UIMessageStreamWriter, UITool, UIToolInvocation, UITools, UI_MESSAGE_STREAM_HEADERS, UnsupportedModelVersionError, UseCompletionOptions, assistantModelMessageSchema, callCompletionApi, consumeStream, convertFileListToFileUIParts, convertToCoreMessages, convertToModelMessages, coreAssistantMessageSchema, coreMessageSchema, coreSystemMessageSchema, coreToolMessageSchema, coreUserMessageSchema, cosineSimilarity, createAgentUIStream, createAgentUIStreamResponse, createProviderRegistry, createTextStreamResponse, createUIMessageStream, createUIMessageStreamResponse, customProvider, defaultSettingsMiddleware, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, generateImage as experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractReasoningMiddleware, generateObject, generateText, getTextFromDataUrl, getToolName, getToolOrDynamicToolName, hasToolCall, isDataUIPart, isDeepEqualData, isFileUIPart, isReasoningUIPart, isTextUIPart, isToolOrDynamicToolUIPart, isToolUIPart, lastAssistantMessageIsCompleteWithApprovalResponses, lastAssistantMessageIsCompleteWithToolCalls, modelMessageSchema, parsePartialJson, pipeAgentUIStreamToResponse, pipeTextStreamToResponse, pipeUIMessageStreamToResponse, pruneMessages, readUIMessageStream, rerank, safeValidateUIMessages, simulateReadableStream, simulateStreamingMiddleware, smoothStream, stepCountIs, streamObject, streamText, systemModelMessageSchema, toolModelMessageSchema, uiMessageChunkSchema, userModelMessageSchema, validateUIMessages, wrapLanguageModel, wrapProvider };
package/dist/index.d.ts CHANGED
@@ -3,7 +3,7 @@ import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
3
3
  import { Tool, InferToolInput, InferToolOutput, AssistantModelMessage, ToolModelMessage, ReasoningPart, FlexibleSchema, InferSchema, ModelMessage, SystemModelMessage, UserModelMessage, ProviderOptions, IdGenerator, ToolCall, MaybePromiseLike, TextPart, FilePart, Resolvable, FetchFunction, DataContent } from '@ai-sdk/provider-utils';
4
4
  export { AssistantContent, AssistantModelMessage, DataContent, FilePart, FlexibleSchema, IdGenerator, ImagePart, InferSchema, InferToolInput, InferToolOutput, ModelMessage, Schema, SystemModelMessage, TextPart, Tool, ToolApprovalRequest, ToolApprovalResponse, ToolCallOptions, ToolCallPart, ToolContent, ToolExecuteFunction, ToolModelMessage, ToolResultPart, UserContent, UserModelMessage, asSchema, createIdGenerator, dynamicTool, generateId, jsonSchema, parseJsonEventStream, tool, zodSchema } from '@ai-sdk/provider-utils';
5
5
  import * as _ai_sdk_provider from '@ai-sdk/provider';
6
- import { EmbeddingModelV3, EmbeddingModelV2, EmbeddingModelV3Embedding, ImageModelV3, ImageModelV3CallWarning, ImageModelV3ProviderMetadata, JSONValue as JSONValue$1, LanguageModelV3, LanguageModelV2, LanguageModelV3FinishReason, LanguageModelV3CallWarning, LanguageModelV3Source, LanguageModelV3Middleware, SharedV3ProviderMetadata, SpeechModelV3, SpeechModelV2, SpeechModelV3CallWarning, TranscriptionModelV3, TranscriptionModelV2, TranscriptionModelV3CallWarning, LanguageModelV3Usage, LanguageModelV3CallOptions, AISDKError, LanguageModelV3ToolCall, JSONSchema7, JSONParseError, TypeValidationError, SharedV3Warning, ProviderV3, ProviderV2, NoSuchModelError } from '@ai-sdk/provider';
6
+ import { EmbeddingModelV3, EmbeddingModelV2, EmbeddingModelV3Embedding, ImageModelV3, ImageModelV3CallWarning, ImageModelV3ProviderMetadata, JSONValue as JSONValue$1, LanguageModelV3, LanguageModelV2, LanguageModelV3FinishReason, LanguageModelV3CallWarning, LanguageModelV3Source, LanguageModelV3Middleware, RerankingModelV3, SharedV3ProviderMetadata, SpeechModelV3, SpeechModelV2, SpeechModelV3CallWarning, TranscriptionModelV3, TranscriptionModelV2, TranscriptionModelV3CallWarning, LanguageModelV3Usage, LanguageModelV3CallOptions, AISDKError, LanguageModelV3ToolCall, JSONSchema7, JSONParseError, TypeValidationError, JSONObject, SharedV3Warning, ProviderV3, ProviderV2, NoSuchModelError } from '@ai-sdk/provider';
7
7
  export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, JSONSchema7, LoadAPIKeyError, LoadSettingError, NoContentGeneratedError, NoSuchModelError, TooManyEmbeddingValuesForCallError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
8
8
  import { ServerResponse } from 'node:http';
9
9
  import { AttributeValue, Tracer } from '@opentelemetry/api';
@@ -116,6 +116,11 @@ type LanguageModelResponseMetadata = {
116
116
  headers?: Record<string, string>;
117
117
  };
118
118
 
119
+ /**
120
+ * Reranking model that is used by the AI SDK Core functions.
121
+ */
122
+ type RerankingModel = RerankingModelV3;
123
+
119
124
  /**
120
125
  * Provider for language, text embedding, and image models.
121
126
  */
@@ -151,6 +156,17 @@ type Provider = {
151
156
  @returns {ImageModel} The image model associated with the id
152
157
  */
153
158
  imageModel(modelId: string): ImageModel;
159
+ /**
160
+ Returns the reranking model with the given id.
161
+ The model id is then passed to the provider function to get the model.
162
+
163
+ @param {string} id - The id of the model to return.
164
+
165
+ @return {RerankingModel<VALUE>} The reranking model associated with the id
166
+
167
+ @throws {NoSuchModelError} If no such model exists.
168
+ */
169
+ rerankingModel(modelId: string): RerankingModel;
154
170
  };
155
171
 
156
172
  /**
@@ -4523,7 +4539,7 @@ interface SpeechResult {
4523
4539
  /**
4524
4540
  Provider metadata from the provider.
4525
4541
  */
4526
- readonly providerMetadata: Record<string, Record<string, JSONValue$1>>;
4542
+ readonly providerMetadata: Record<string, JSONObject>;
4527
4543
  }
4528
4544
 
4529
4545
  /**
@@ -4694,23 +4710,26 @@ declare function wrapProvider({ provider, languageModelMiddleware, }: {
4694
4710
  * @param {Record<string, ImageModel>} [options.imageModels] - A record of image models, where keys are model IDs and values are ImageModel instances.
4695
4711
  * @param {Record<string, TranscriptionModel>} [options.transcriptionModels] - A record of transcription models, where keys are model IDs and values are TranscriptionModel instances.
4696
4712
  * @param {Record<string, SpeechModel>} [options.speechModels] - A record of speech models, where keys are model IDs and values are SpeechModel instances.
4713
+ * @param {Record<string, RerankingModel<string>>} [options.rerankingModels] - A record of reranking models, where keys are model IDs and values are RerankingModel<string> instances.
4697
4714
  * @param {Provider} [options.fallbackProvider] - An optional fallback provider to use when a requested model is not found in the custom provider.
4698
4715
  * @returns {Provider} A Provider object with languageModel, textEmbeddingModel, imageModel, transcriptionModel, and speechModel methods.
4699
4716
  *
4700
4717
  * @throws {NoSuchModelError} Throws when a requested model is not found and no fallback provider is available.
4701
4718
  */
4702
- declare function customProvider<LANGUAGE_MODELS extends Record<string, LanguageModelV3>, EMBEDDING_MODELS extends Record<string, EmbeddingModelV3<string>>, IMAGE_MODELS extends Record<string, ImageModelV3>, TRANSCRIPTION_MODELS extends Record<string, TranscriptionModelV3>, SPEECH_MODELS extends Record<string, SpeechModelV3>>({ languageModels, textEmbeddingModels, imageModels, transcriptionModels, speechModels, fallbackProvider, }: {
4719
+ declare function customProvider<LANGUAGE_MODELS extends Record<string, LanguageModelV3>, EMBEDDING_MODELS extends Record<string, EmbeddingModelV3<string>>, IMAGE_MODELS extends Record<string, ImageModelV3>, TRANSCRIPTION_MODELS extends Record<string, TranscriptionModelV3>, SPEECH_MODELS extends Record<string, SpeechModelV3>, RERANKING_MODELS extends Record<string, RerankingModelV3>>({ languageModels, textEmbeddingModels, imageModels, transcriptionModels, speechModels, rerankingModels, fallbackProvider: fallbackProviderArg, }: {
4703
4720
  languageModels?: LANGUAGE_MODELS;
4704
4721
  textEmbeddingModels?: EMBEDDING_MODELS;
4705
4722
  imageModels?: IMAGE_MODELS;
4706
4723
  transcriptionModels?: TRANSCRIPTION_MODELS;
4707
4724
  speechModels?: SPEECH_MODELS;
4725
+ rerankingModels?: RERANKING_MODELS;
4708
4726
  fallbackProvider?: ProviderV3 | ProviderV2;
4709
4727
  }): ProviderV3 & {
4710
4728
  languageModel(modelId: ExtractModelId<LANGUAGE_MODELS>): LanguageModelV3;
4711
4729
  textEmbeddingModel(modelId: ExtractModelId<EMBEDDING_MODELS>): EmbeddingModelV3<string>;
4712
4730
  imageModel(modelId: ExtractModelId<IMAGE_MODELS>): ImageModelV3;
4713
4731
  transcriptionModel(modelId: ExtractModelId<TRANSCRIPTION_MODELS>): TranscriptionModelV3;
4732
+ rerankingModel(modelId: ExtractModelId<RERANKING_MODELS>): RerankingModelV3;
4714
4733
  speechModel(modelId: ExtractModelId<SPEECH_MODELS>): SpeechModelV3;
4715
4734
  };
4716
4735
  /**
@@ -4726,7 +4745,7 @@ declare class NoSuchProviderError extends NoSuchModelError {
4726
4745
  readonly availableProviders: string[];
4727
4746
  constructor({ modelId, modelType, providerId, availableProviders, message, }: {
4728
4747
  modelId: string;
4729
- modelType: 'languageModel' | 'textEmbeddingModel' | 'imageModel' | 'transcriptionModel' | 'speechModel';
4748
+ modelType: 'languageModel' | 'textEmbeddingModel' | 'imageModel' | 'transcriptionModel' | 'speechModel' | 'rerankingModel';
4730
4749
  providerId: string;
4731
4750
  availableProviders: string[];
4732
4751
  message?: string;
@@ -4746,6 +4765,8 @@ interface ProviderRegistryProvider<PROVIDERS extends Record<string, ProviderV3>
4746
4765
  transcriptionModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${string}` : never): TranscriptionModelV3;
4747
4766
  speechModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${ExtractLiteralUnion<Parameters<NonNullable<PROVIDERS[KEY]['speechModel']>>[0]>}` : never): SpeechModelV3;
4748
4767
  speechModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${string}` : never): SpeechModelV3;
4768
+ rerankingModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${ExtractLiteralUnion<Parameters<NonNullable<PROVIDERS[KEY]['rerankingModel']>>[0]>}` : never): RerankingModelV3;
4769
+ rerankingModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${string}` : never): RerankingModelV3;
4749
4770
  }
4750
4771
  /**
4751
4772
  * Creates a registry for the given providers with optional middleware functionality.
@@ -4768,6 +4789,125 @@ declare function createProviderRegistry<PROVIDERS extends Record<string, Provide
4768
4789
  */
4769
4790
  declare const experimental_createProviderRegistry: typeof createProviderRegistry;
4770
4791
 
4792
+ /**
4793
+ * The result of a `rerank` call.
4794
+ * It contains the original documents, the reranked documents, and additional information.
4795
+ */
4796
+ interface RerankResult<VALUE> {
4797
+ /**
4798
+ * The original documents that were reranked.
4799
+ */
4800
+ readonly originalDocuments: Array<VALUE>;
4801
+ /**
4802
+ * Reranked documents.
4803
+ *
4804
+ * Sorted by relevance score in descending order.
4805
+ *
4806
+ * Can be less than the original documents if there was a topK limit.
4807
+ */
4808
+ readonly rerankedDocuments: Array<VALUE>;
4809
+ /**
4810
+ * The ranking is a list of objects with the original index,
4811
+ * relevance score, and the reranked document.
4812
+ *
4813
+ * Sorted by relevance score in descending order.
4814
+ *
4815
+ * Can be less than the original documents if there was a topK limit.
4816
+ */
4817
+ readonly ranking: Array<{
4818
+ originalIndex: number;
4819
+ relevanceScore: number;
4820
+ document: VALUE;
4821
+ }>;
4822
+ /**
4823
+ * Optional provider-specific metadata.
4824
+ */
4825
+ readonly providerMetadata?: ProviderMetadata;
4826
+ /**
4827
+ * Optional raw response data.
4828
+ */
4829
+ readonly response: {
4830
+ /**
4831
+ * ID for the generated response if the provider sends one.
4832
+ */
4833
+ id?: string;
4834
+ /**
4835
+ * Timestamp of the generated response.
4836
+ */
4837
+ timestamp: Date;
4838
+ /**
4839
+ * The ID of the model that was used to generate the response.
4840
+ */
4841
+ modelId: string;
4842
+ /**
4843
+ * Response headers.
4844
+ */
4845
+ headers?: Record<string, string>;
4846
+ /**
4847
+ * The response body.
4848
+ */
4849
+ body?: unknown;
4850
+ };
4851
+ }
4852
+
4853
+ /**
4854
+ Rerank documents using an reranking model. The type of the value is defined by the reranking model.
4855
+
4856
+ @param model - The Reranking model to use.
4857
+ @param documents - The documents that should be reranking.
4858
+ @param query - The query is a string that represents the query to rerank the documents against.
4859
+ @param topN - Top n documents to rerank.
4860
+
4861
+ @param maxRetries - Maximum number of retries. Set to 0 to disable retries. Default: 2.
4862
+ @param abortSignal - An optional abort signal that can be used to cancel the call.
4863
+ @param headers - Additional HTTP headers to be sent with the request. Only applicable for HTTP-based providers.
4864
+
4865
+ @returns A result object that contains the reranked documents, the reranked indices, and additional information.
4866
+ */
4867
+ declare function rerank<VALUE extends JSONObject | string>({ model, documents, query, topN, maxRetries: maxRetriesArg, abortSignal, headers, providerOptions, experimental_telemetry: telemetry, }: {
4868
+ /**
4869
+ The reranking model to use.
4870
+ */
4871
+ model: RerankingModel;
4872
+ /**
4873
+ * The documents that should be reranked.
4874
+ */
4875
+ documents: Array<VALUE>;
4876
+ /**
4877
+ The query is a string that represents the query to rerank the documents against.
4878
+ */
4879
+ query: string;
4880
+ /**
4881
+ * Number of top documents to return.
4882
+ */
4883
+ topN?: number;
4884
+ /**
4885
+ Maximum number of retries per reranking model call. Set to 0 to disable retries.
4886
+
4887
+ @default 2
4888
+ */
4889
+ maxRetries?: number;
4890
+ /**
4891
+ Abort signal.
4892
+ */
4893
+ abortSignal?: AbortSignal;
4894
+ /**
4895
+ Additional headers to include in the request.
4896
+ Only applicable for HTTP-based providers.
4897
+ */
4898
+ headers?: Record<string, string>;
4899
+ /**
4900
+ * Optional telemetry configuration (experimental).
4901
+ */
4902
+ experimental_telemetry?: TelemetrySettings;
4903
+ /**
4904
+ Additional provider-specific options. They are passed through
4905
+ to the provider from the AI SDK and enable provider-specific
4906
+ functionality that can be fully encapsulated in the provider.
4907
+ */
4908
+ providerOptions?: ProviderOptions;
4909
+ }): Promise<RerankResult<VALUE>>;
4910
+
4771
4911
  declare function createTextStreamResponse({ status, statusText, headers, textStream, }: ResponseInit & {
4772
4912
  textStream: ReadableStream<string>;
4773
4913
  }): Response;
@@ -4825,7 +4965,7 @@ interface TranscriptionResult {
4825
4965
  /**
4826
4966
  Provider metadata from the provider.
4827
4967
  */
4828
- readonly providerMetadata: Record<string, Record<string, JSONValue$1>>;
4968
+ readonly providerMetadata: Record<string, JSONObject>;
4829
4969
  }
4830
4970
 
4831
4971
  /**
@@ -4902,4 +5042,4 @@ declare global {
4902
5042
  var AI_SDK_LOG_WARNINGS: LogWarningsFunction | undefined | false;
4903
5043
  }
4904
5044
 
4905
- export { AbstractChat, Agent, AsyncIterableStream, CallSettings, CallWarning, ChatAddToolApproveResponseFunction, ChatInit, ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback, ChatOnToolCallCallback, ChatRequestOptions, ChatState, ChatStatus, ChatTransport, ChunkDetector, CompletionRequestOptions, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreToolMessage, CoreUserMessage, CreateUIMessage, DataUIPart, DeepPartial, DefaultChatTransport, DownloadError, DynamicToolCall, DynamicToolError, DynamicToolResult, DynamicToolUIPart, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, ErrorHandler, ToolLoopAgent as Experimental_Agent, ToolLoopAgentSettings as Experimental_AgentSettings, DownloadFunction as Experimental_DownloadFunction, GenerateImageResult as Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, InferAgentUIMessage as Experimental_InferAgentUIMessage, LogWarningsFunction as Experimental_LogWarningsFunction, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, Warning as Experimental_Warning, FileUIPart, FinishReason, GenerateObjectResult, GenerateTextOnFinishCallback, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, HttpChatTransport, HttpChatTransportInitOptions, ImageModel, ImageGenerationWarning as ImageModelCallWarning, ImageModelProviderMetadata, ImageModelResponseMetadata, InferAgentUIMessage, InferGenerateOutput, InferStreamOutput, InferUIDataParts, InferUIMessageChunk, InferUITool, InferUITools, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolInputError, JSONValue, JsonToSseTransformStream, LanguageModel, LanguageModelMiddleware, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, MessageConversionError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputGeneratedError, NoOutputSpecifiedError, NoSpeechGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, PrepareReconnectToStreamRequest, PrepareSendMessagesRequest, PrepareStepFunction, PrepareStepResult, Prompt, Provider, ProviderMetadata, ProviderRegistryProvider, ReasoningOutput, ReasoningUIPart, RepairTextFunction, RetryError, SafeValidateUIMessagesResult, SerialJobExecutor, SourceDocumentUIPart, SourceUrlUIPart, SpeechModel, SpeechModelResponseMetadata, SpeechWarning, StaticToolCall, StaticToolError, StaticToolOutputDenied, StaticToolResult, StepResult, StepStartUIPart, StopCondition, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextStreamChatTransport, TextStreamPart, TextUIPart, ToolApprovalRequestOutput, ToolCallRepairError, ToolCallRepairFunction, ToolChoice, ToolLoopAgent, ToolLoopAgentOnFinishCallback, ToolLoopAgentOnStepFinishCallback, ToolLoopAgentSettings, ToolSet, ToolUIPart, TranscriptionModel, TranscriptionModelResponseMetadata, TranscriptionWarning, TypedToolCall, TypedToolError, TypedToolOutputDenied, TypedToolResult, UIDataPartSchemas, UIDataTypes, UIMessage, UIMessageChunk, UIMessagePart, UIMessageStreamOnFinishCallback, UIMessageStreamOptions, UIMessageStreamWriter, UITool, UIToolInvocation, UITools, UI_MESSAGE_STREAM_HEADERS, UnsupportedModelVersionError, UseCompletionOptions, assistantModelMessageSchema, callCompletionApi, consumeStream, convertFileListToFileUIParts, convertToCoreMessages, convertToModelMessages, coreAssistantMessageSchema, coreMessageSchema, coreSystemMessageSchema, coreToolMessageSchema, coreUserMessageSchema, cosineSimilarity, createAgentUIStream, createAgentUIStreamResponse, createProviderRegistry, createTextStreamResponse, createUIMessageStream, createUIMessageStreamResponse, customProvider, defaultSettingsMiddleware, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, generateImage as experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractReasoningMiddleware, generateObject, generateText, getTextFromDataUrl, getToolName, getToolOrDynamicToolName, hasToolCall, isDataUIPart, isDeepEqualData, isFileUIPart, isReasoningUIPart, isTextUIPart, isToolOrDynamicToolUIPart, isToolUIPart, lastAssistantMessageIsCompleteWithApprovalResponses, lastAssistantMessageIsCompleteWithToolCalls, modelMessageSchema, parsePartialJson, pipeAgentUIStreamToResponse, pipeTextStreamToResponse, pipeUIMessageStreamToResponse, pruneMessages, readUIMessageStream, safeValidateUIMessages, simulateReadableStream, simulateStreamingMiddleware, smoothStream, stepCountIs, streamObject, streamText, systemModelMessageSchema, toolModelMessageSchema, uiMessageChunkSchema, userModelMessageSchema, validateUIMessages, wrapLanguageModel, wrapProvider };
5045
+ export { AbstractChat, Agent, AsyncIterableStream, CallSettings, CallWarning, ChatAddToolApproveResponseFunction, ChatInit, ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback, ChatOnToolCallCallback, ChatRequestOptions, ChatState, ChatStatus, ChatTransport, ChunkDetector, CompletionRequestOptions, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreToolMessage, CoreUserMessage, CreateUIMessage, DataUIPart, DeepPartial, DefaultChatTransport, DownloadError, DynamicToolCall, DynamicToolError, DynamicToolResult, DynamicToolUIPart, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, ErrorHandler, ToolLoopAgent as Experimental_Agent, ToolLoopAgentSettings as Experimental_AgentSettings, DownloadFunction as Experimental_DownloadFunction, GenerateImageResult as Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, InferAgentUIMessage as Experimental_InferAgentUIMessage, LogWarningsFunction as Experimental_LogWarningsFunction, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, Warning as Experimental_Warning, FileUIPart, FinishReason, GenerateObjectResult, GenerateTextOnFinishCallback, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, HttpChatTransport, HttpChatTransportInitOptions, ImageModel, ImageGenerationWarning as ImageModelCallWarning, ImageModelProviderMetadata, ImageModelResponseMetadata, InferAgentUIMessage, InferGenerateOutput, InferStreamOutput, InferUIDataParts, InferUIMessageChunk, InferUITool, InferUITools, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolInputError, JSONValue, JsonToSseTransformStream, LanguageModel, LanguageModelMiddleware, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, MessageConversionError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputGeneratedError, NoOutputSpecifiedError, NoSpeechGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, PrepareReconnectToStreamRequest, PrepareSendMessagesRequest, PrepareStepFunction, PrepareStepResult, Prompt, Provider, ProviderMetadata, ProviderRegistryProvider, ReasoningOutput, ReasoningUIPart, RepairTextFunction, RerankResult, RerankingModel, RetryError, SafeValidateUIMessagesResult, SerialJobExecutor, SourceDocumentUIPart, SourceUrlUIPart, SpeechModel, SpeechModelResponseMetadata, SpeechWarning, StaticToolCall, StaticToolError, StaticToolOutputDenied, StaticToolResult, StepResult, StepStartUIPart, StopCondition, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextStreamChatTransport, TextStreamPart, TextUIPart, ToolApprovalRequestOutput, ToolCallRepairError, ToolCallRepairFunction, ToolChoice, ToolLoopAgent, ToolLoopAgentOnFinishCallback, ToolLoopAgentOnStepFinishCallback, ToolLoopAgentSettings, ToolSet, ToolUIPart, TranscriptionModel, TranscriptionModelResponseMetadata, TranscriptionWarning, TypedToolCall, TypedToolError, TypedToolOutputDenied, TypedToolResult, UIDataPartSchemas, UIDataTypes, UIMessage, UIMessageChunk, UIMessagePart, UIMessageStreamOnFinishCallback, UIMessageStreamOptions, UIMessageStreamWriter, UITool, UIToolInvocation, UITools, UI_MESSAGE_STREAM_HEADERS, UnsupportedModelVersionError, UseCompletionOptions, assistantModelMessageSchema, callCompletionApi, consumeStream, convertFileListToFileUIParts, convertToCoreMessages, convertToModelMessages, coreAssistantMessageSchema, coreMessageSchema, coreSystemMessageSchema, coreToolMessageSchema, coreUserMessageSchema, cosineSimilarity, createAgentUIStream, createAgentUIStreamResponse, createProviderRegistry, createTextStreamResponse, createUIMessageStream, createUIMessageStreamResponse, customProvider, defaultSettingsMiddleware, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, generateImage as experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractReasoningMiddleware, generateObject, generateText, getTextFromDataUrl, getToolName, getToolOrDynamicToolName, hasToolCall, isDataUIPart, isDeepEqualData, isFileUIPart, isReasoningUIPart, isTextUIPart, isToolOrDynamicToolUIPart, isToolUIPart, lastAssistantMessageIsCompleteWithApprovalResponses, lastAssistantMessageIsCompleteWithToolCalls, modelMessageSchema, parsePartialJson, pipeAgentUIStreamToResponse, pipeTextStreamToResponse, pipeUIMessageStreamToResponse, pruneMessages, readUIMessageStream, rerank, safeValidateUIMessages, simulateReadableStream, simulateStreamingMiddleware, smoothStream, stepCountIs, streamObject, streamText, systemModelMessageSchema, toolModelMessageSchema, uiMessageChunkSchema, userModelMessageSchema, validateUIMessages, wrapLanguageModel, wrapProvider };