ai 5.0.0-beta.21 → 5.0.0-beta.22

package/dist/index.d.ts

@@ -4,9 +4,8 @@ import { AttributeValue, Tracer } from '@opentelemetry/api';
 import { EmbeddingModelV2, EmbeddingModelV2Embedding, ImageModelV2, ImageModelV2CallWarning, ImageModelV2ProviderMetadata, JSONValue as JSONValue$1, LanguageModelV2, LanguageModelV2FinishReason, LanguageModelV2CallWarning, LanguageModelV2Source, SharedV2ProviderMetadata, SpeechModelV2, SpeechModelV2CallWarning, TranscriptionModelV2, TranscriptionModelV2CallWarning, LanguageModelV2Usage, LanguageModelV2CallOptions, AISDKError, LanguageModelV2ToolCall, JSONSchema7, JSONParseError, TypeValidationError, LanguageModelV2Middleware, ProviderV2, NoSuchModelError, JSONObject } from '@ai-sdk/provider';
 export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, JSONSchema7, LoadAPIKeyError, NoContentGeneratedError, NoSuchModelError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
 import * as z3 from 'zod/v3';
-import * as z4$1 from 'zod/v4';
+import * as z4 from 'zod/v4';
 import { z } from 'zod/v4';
-import * as z4 from 'zod/v4/core';
 import { ServerResponse } from 'node:http';
 import { ServerResponse as ServerResponse$1 } from 'http';
 
@@ -669,7 +668,7 @@ It always recurses into arrays.
 
 Adopted from [type-fest](https://github.com/sindresorhus/type-fest/tree/main) PartialDeep.
  */
-type DeepPartial<T> = T extends z3.ZodTypeAny ? DeepPartialInternal<z3.infer<T>> : T extends z4.$ZodType ? DeepPartialInternal<z4.infer<T>> : DeepPartialInternal<T>;
+type DeepPartial<T> = T extends z3.ZodTypeAny ? DeepPartialInternal<z3.infer<T>> : T extends z4.core.$ZodType ? DeepPartialInternal<z4.infer<T>> : DeepPartialInternal<T>;
 type DeepPartialInternal<T> = T extends null | undefined | string | number | boolean | symbol | bigint | void | Date | RegExp | ((...arguments_: any[]) => unknown) | (new (...arguments_: any[]) => unknown) ? T : T extends Map<infer KeyType, infer ValueType> ? PartialMap<KeyType, ValueType> : T extends Set<infer ItemType> ? PartialSet<ItemType> : T extends ReadonlyMap<infer KeyType, infer ValueType> ? PartialReadonlyMap<KeyType, ValueType> : T extends ReadonlySet<infer ItemType> ? PartialReadonlySet<ItemType> : T extends object ? T extends ReadonlyArray<infer ItemType> ? ItemType[] extends T ? readonly ItemType[] extends T ? ReadonlyArray<DeepPartialInternal<ItemType | undefined>> : Array<DeepPartialInternal<ItemType | undefined>> : PartialObject<T> : PartialObject<T> : unknown;
 type PartialMap<KeyType, ValueType> = {} & Map<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
 type PartialSet<T> = {} & Set<DeepPartialInternal<T>>;
@@ -697,7 +696,7 @@ interface Output<OUTPUT, PARTIAL> {
 }
 declare const text: () => Output<string, string>;
 declare const object: <OUTPUT>({ schema: inputSchema, }: {
-    schema: z4$1.ZodType<OUTPUT, any> | z3.Schema<OUTPUT, z3.ZodTypeDef, any> | Schema<OUTPUT>;
+    schema: z4.core.$ZodType<OUTPUT, any> | z3.Schema<OUTPUT, z3.ZodTypeDef, any> | Schema<OUTPUT>;
 }) => Output<OUTPUT, DeepPartial<OUTPUT>>;
 
 type output_Output<OUTPUT, PARTIAL> = Output<OUTPUT, PARTIAL>;
@@ -777,7 +776,7 @@ declare const systemModelMessageSchema: z.ZodType<SystemModelMessage>;
 /**
 @deprecated Use `systemModelMessageSchema` instead.
  */
-declare const coreSystemMessageSchema: z.ZodType<SystemModelMessage, unknown>;
+declare const coreSystemMessageSchema: z.ZodType<SystemModelMessage, unknown, z.core.$ZodTypeInternals<SystemModelMessage, unknown>>;
 /**
 @deprecated Use `UserModelMessage` instead.
  */
@@ -786,7 +785,7 @@ declare const userModelMessageSchema: z.ZodType<UserModelMessage>;
 /**
 @deprecated Use `userModelMessageSchema` instead.
  */
-declare const coreUserMessageSchema: z.ZodType<UserModelMessage, unknown>;
+declare const coreUserMessageSchema: z.ZodType<UserModelMessage, unknown, z.core.$ZodTypeInternals<UserModelMessage, unknown>>;
 /**
 @deprecated Use `AssistantModelMessage` instead.
  */
@@ -795,7 +794,7 @@ declare const assistantModelMessageSchema: z.ZodType<AssistantModelMessage>;
 /**
 @deprecated Use `assistantModelMessageSchema` instead.
  */
-declare const coreAssistantMessageSchema: z.ZodType<AssistantModelMessage, unknown>;
+declare const coreAssistantMessageSchema: z.ZodType<AssistantModelMessage, unknown, z.core.$ZodTypeInternals<AssistantModelMessage, unknown>>;
 /**
 @deprecated Use `ToolModelMessage` instead.
  */
@@ -804,7 +803,7 @@ declare const toolModelMessageSchema: z.ZodType<ToolModelMessage>;
 /**
 @deprecated Use `toolModelMessageSchema` instead.
  */
-declare const coreToolMessageSchema: z.ZodType<ToolModelMessage, unknown>;
+declare const coreToolMessageSchema: z.ZodType<ToolModelMessage, unknown, z.core.$ZodTypeInternals<ToolModelMessage, unknown>>;
 /**
 @deprecated Use `ModelMessage` instead.
    */
@@ -2571,7 +2570,7 @@ functionality that can be fully encapsulated in the provider.
 @returns
 A result object that contains the generated object, the finish reason, the token usage, and additional information.
  */
-declare function generateObject<SCHEMA extends z3.Schema | z4$1.ZodType | Schema = z4$1.ZodType<JSONValue$1>, OUTPUT extends 'object' | 'array' | 'enum' | 'no-schema' = InferSchema<SCHEMA> extends string ? 'enum' : 'object', RESULT = OUTPUT extends 'array' ? Array<InferSchema<SCHEMA>> : InferSchema<SCHEMA>>(options: Omit<CallSettings, 'stopSequences'> & Prompt & (OUTPUT extends 'enum' ? {
+declare function generateObject<SCHEMA extends z3.Schema | z4.core.$ZodType | Schema = z4.core.$ZodType<JSONValue$1>, OUTPUT extends 'object' | 'array' | 'enum' | 'no-schema' = InferSchema<SCHEMA> extends string ? 'enum' : 'object', RESULT = OUTPUT extends 'array' ? Array<InferSchema<SCHEMA>> : InferSchema<SCHEMA>>(options: Omit<CallSettings, 'stopSequences'> & Prompt & (OUTPUT extends 'enum' ? {
     /**
 The enum values that the model should use.
   */
@@ -2888,7 +2887,7 @@ functionality that can be fully encapsulated in the provider.
 @returns
 A result object for accessing the partial object stream and additional information.
  */
-declare function streamObject<SCHEMA extends z3.Schema | z4$1.ZodType | Schema = z4$1.ZodType<JSONValue$1>, OUTPUT extends 'object' | 'array' | 'enum' | 'no-schema' = InferSchema<SCHEMA> extends string ? 'enum' : 'object', RESULT = OUTPUT extends 'array' ? Array<InferSchema<SCHEMA>> : InferSchema<SCHEMA>>(options: Omit<CallSettings, 'stopSequences'> & Prompt & (OUTPUT extends 'enum' ? {
+declare function streamObject<SCHEMA extends z3.Schema | z4.core.$ZodType | Schema = z4.core.$ZodType<JSONValue$1>, OUTPUT extends 'object' | 'array' | 'enum' | 'no-schema' = InferSchema<SCHEMA> extends string ? 'enum' : 'object', RESULT = OUTPUT extends 'array' ? Array<InferSchema<SCHEMA>> : InferSchema<SCHEMA>>(options: Omit<CallSettings, 'stopSequences'> & Prompt & (OUTPUT extends 'enum' ? {
     /**
 The enum values that the model should use.
   */
@@ -3121,7 +3120,7 @@ declare function simulateStreamingMiddleware(): LanguageModelV2Middleware;
  * @param options.model - The original LanguageModelV2 instance to be wrapped.
  * @param options.middleware - The middleware to be applied to the language model. When multiple middlewares are provided, the first middleware will transform the input first, and the last middleware will be wrapped directly around the model.
  * @param options.modelId - Optional custom model ID to override the original model's ID.
- * @param options.providerId - Optional custom provider ID to override the original model's provider.
+ * @param options.providerId - Optional custom provider ID to override the original model's provider ID.
  * @returns A new LanguageModelV2 instance with middleware applied.
  */
 declare const wrapLanguageModel: ({ model, middleware: middlewareArg, modelId, providerId, }: {
@@ -3131,6 +3130,22 @@ declare const wrapLanguageModel: ({ model, middleware: middlewareArg, modelId, p
     providerId?: string;
 }) => LanguageModelV2;
 
+/**
+ * Wraps a ProviderV2 instance with middleware functionality.
+ * This function allows you to apply middleware to all language models
+ * from the provider, enabling you to transform parameters, wrap generate
+ * operations, and wrap stream operations for every language model.
+ *
+ * @param options - Configuration options for wrapping the provider.
+ * @param options.provider - The original ProviderV2 instance to be wrapped.
+ * @param options.languageModelMiddleware - The middleware to be applied to all language models from the provider. When multiple middlewares are provided, the first middleware will transform the input first, and the last middleware will be wrapped directly around the model.
+ * @returns A new ProviderV2 instance with middleware applied to all language models.
+ */
+declare function wrapProvider({ provider, languageModelMiddleware, }: {
+    provider: ProviderV2;
+    languageModelMiddleware: LanguageModelV2Middleware | LanguageModelV2Middleware[];
+}): ProviderV2;
+
 /**
  * Creates a custom provider with specified language models, text embedding models, image models, transcription models, speech models, and an optional fallback provider.
  *
@@ -3194,10 +3209,20 @@ interface ProviderRegistryProvider<PROVIDERS extends Record<string, ProviderV2>
     speechModel<KEY extends keyof PROVIDERS>(id: KEY extends string ? `${KEY & string}${SEPARATOR}${string}` : never): SpeechModelV2;
 }
 /**
- * Creates a registry for the given providers.
+ * Creates a registry for the given providers with optional middleware functionality.
+ * This function allows you to register multiple providers and optionally apply middleware
+ * to all language models from the registry, enabling you to transform parameters, wrap generate
+ * operations, and wrap stream operations for every language model accessed through the registry.
+ *
+ * @param providers - A record of provider instances to be registered in the registry.
+ * @param options - Configuration options for the provider registry.
+ * @param options.separator - The separator used between provider ID and model ID in the combined identifier. Defaults to ':'.
+ * @param options.languageModelMiddleware - Optional middleware to be applied to all language models from the registry. When multiple middlewares are provided, the first middleware will transform the input first, and the last middleware will be wrapped directly around the model.
+ * @returns A new ProviderRegistryProvider instance that provides access to all registered providers with optional middleware applied to language models.
  */
-declare function createProviderRegistry<PROVIDERS extends Record<string, ProviderV2>, SEPARATOR extends string = ':'>(providers: PROVIDERS, { separator, }?: {
+declare function createProviderRegistry<PROVIDERS extends Record<string, ProviderV2>, SEPARATOR extends string = ':'>(providers: PROVIDERS, { separator, languageModelMiddleware, }?: {
     separator?: SEPARATOR;
+    languageModelMiddleware?: LanguageModelV2Middleware | LanguageModelV2Middleware[];
 }): ProviderRegistryProvider<PROVIDERS, SEPARATOR>;
 /**
  * @deprecated Use `createProviderRegistry` instead.
@@ -4007,4 +4032,4 @@ declare global {
     var AI_SDK_DEFAULT_PROVIDER: ProviderV2 | undefined;
 }
 
-export { AbstractChat, CallSettings, CallWarning, ChatInit, ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback, ChatOnToolCallCallback, ChatRequestOptions, ChatState, ChatStatus, ChatTransport, ChunkDetector, CompletionRequestOptions, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreToolMessage, CoreUserMessage, CreateUIMessage, DataUIPart, DeepPartial, DefaultChatTransport, DownloadError, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, ErrorHandler, Agent as Experimental_Agent, AgentSettings as Experimental_AgentSettings, GenerateImageResult as Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, FileUIPart, FinishReason, GenerateObjectResult, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, HttpChatTransport, HttpChatTransportInitOptions, ImageModel, ImageGenerationWarning as ImageModelCallWarning, ImageModelProviderMetadata, ImageModelResponseMetadata, InferUIDataParts, InferUIMessageChunk, InferUITool, InferUITools, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolInputError, JSONRPCError, JSONRPCMessage, JSONRPCNotification, JSONRPCRequest, JSONRPCResponse, JSONValue, JsonToSseTransformStream, LanguageModel, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, MCPClientError, MCPTransport, MessageConversionError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputSpecifiedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, PrepareReconnectToStreamRequest, PrepareSendMessagesRequest, PrepareStepFunction, PrepareStepResult, Prompt, Provider, ProviderMetadata, ProviderRegistryProvider, ReasoningUIPart, RepairTextFunction, RetryError, SerialJobExecutor, SourceDocumentUIPart, SourceUrlUIPart, SpeechModel, SpeechModelResponseMetadata, SpeechWarning, StepResult, StepStartUIPart, StopCondition, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextStreamChatTransport, TextStreamPart, TextUIPart, ToolCallRepairError, ToolCallRepairFunction, ToolCallUnion, ToolChoice, ToolErrorUnion, ToolResultUnion, ToolSet, ToolUIPart, TranscriptionModel, TranscriptionModelResponseMetadata, TranscriptionWarning, UIDataPartSchemas, UIDataTypes, UIMessage, UIMessageChunk, UIMessagePart, UIMessageStreamOptions, UIMessageStreamWriter, UITool, UITools, UI_MESSAGE_STREAM_HEADERS, UnsupportedModelVersionError, UseCompletionOptions, assistantModelMessageSchema, callCompletionApi, convertFileListToFileUIParts, convertToCoreMessages, convertToModelMessages, coreAssistantMessageSchema, coreMessageSchema, coreSystemMessageSchema, coreToolMessageSchema, coreUserMessageSchema, cosineSimilarity, createProviderRegistry, createTextStreamResponse, createUIMessageStream, createUIMessageStreamResponse, customProvider, defaultSettingsMiddleware, embed, embedMany, createMCPClient as experimental_createMCPClient, experimental_createProviderRegistry, experimental_customProvider, generateImage as experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractReasoningMiddleware, generateObject, generateText, getTextFromDataUrl, getToolName, hasToolCall, isDeepEqualData, isToolUIPart, modelMessageSchema, parsePartialJson, pipeTextStreamToResponse, pipeUIMessageStreamToResponse, readUIMessageStream, simulateReadableStream, simulateStreamingMiddleware, smoothStream, stepCountIs, streamObject, streamText, systemModelMessageSchema, toolModelMessageSchema, userModelMessageSchema, wrapLanguageModel };
+export { AbstractChat, AsyncIterableStream, CallSettings, CallWarning, ChatInit, ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback, ChatOnToolCallCallback, ChatRequestOptions, ChatState, ChatStatus, ChatTransport, ChunkDetector, CompletionRequestOptions, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreToolMessage, CoreUserMessage, CreateUIMessage, DataUIPart, DeepPartial, DefaultChatTransport, DownloadError, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, ErrorHandler, Agent as Experimental_Agent, AgentSettings as Experimental_AgentSettings, GenerateImageResult as Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, FileUIPart, FinishReason, GenerateObjectResult, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, HttpChatTransport, HttpChatTransportInitOptions, ImageModel, ImageGenerationWarning as ImageModelCallWarning, ImageModelProviderMetadata, ImageModelResponseMetadata, InferUIDataParts, InferUIMessageChunk, InferUITool, InferUITools, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolInputError, JSONRPCError, JSONRPCMessage, JSONRPCNotification, JSONRPCRequest, JSONRPCResponse, JSONValue, JsonToSseTransformStream, LanguageModel, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, MCPClientError, MCPTransport, MessageConversionError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputSpecifiedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, PrepareReconnectToStreamRequest, PrepareSendMessagesRequest, PrepareStepFunction, PrepareStepResult, Prompt, Provider, ProviderMetadata, ProviderRegistryProvider, ReasoningUIPart, RepairTextFunction, RetryError, SerialJobExecutor, SourceDocumentUIPart, SourceUrlUIPart, SpeechModel, SpeechModelResponseMetadata, SpeechWarning, StepResult, StepStartUIPart, StopCondition, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextStreamChatTransport, TextStreamPart, TextUIPart, ToolCallRepairError, ToolCallRepairFunction, ToolCallUnion, ToolChoice, ToolErrorUnion, ToolResultUnion, ToolSet, ToolUIPart, TranscriptionModel, TranscriptionModelResponseMetadata, TranscriptionWarning, UIDataPartSchemas, UIDataTypes, UIMessage, UIMessageChunk, UIMessagePart, UIMessageStreamOptions, UIMessageStreamWriter, UITool, UITools, UI_MESSAGE_STREAM_HEADERS, UnsupportedModelVersionError, UseCompletionOptions, assistantModelMessageSchema, callCompletionApi, convertFileListToFileUIParts, convertToCoreMessages, convertToModelMessages, coreAssistantMessageSchema, coreMessageSchema, coreSystemMessageSchema, coreToolMessageSchema, coreUserMessageSchema, cosineSimilarity, createProviderRegistry, createTextStreamResponse, createUIMessageStream, createUIMessageStreamResponse, customProvider, defaultSettingsMiddleware, embed, embedMany, createMCPClient as experimental_createMCPClient, experimental_createProviderRegistry, experimental_customProvider, generateImage as experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractReasoningMiddleware, generateObject, generateText, getTextFromDataUrl, getToolName, hasToolCall, isDeepEqualData, isToolUIPart, modelMessageSchema, parsePartialJson, pipeTextStreamToResponse, pipeUIMessageStreamToResponse, readUIMessageStream, simulateReadableStream, simulateStreamingMiddleware, smoothStream, stepCountIs, streamObject, streamText, systemModelMessageSchema, toolModelMessageSchema, userModelMessageSchema, wrapLanguageModel, wrapProvider };

package/dist/index.js

@@ -108,7 +108,8 @@ __export(src_exports, {
   tool: () => import_provider_utils23.tool,
   toolModelMessageSchema: () => toolModelMessageSchema,
   userModelMessageSchema: () => userModelMessageSchema,
-  wrapLanguageModel: () => wrapLanguageModel
+  wrapLanguageModel: () => wrapLanguageModel,
+  wrapProvider: () => wrapProvider
 });
 module.exports = __toCommonJS(src_exports);
 var import_provider_utils29 = require("@ai-sdk/provider-utils");
@@ -3150,24 +3151,24 @@ function processUIMessageStream({
 }) {
   return stream.pipeThrough(
     new TransformStream({
-      async transform(part, controller) {
+      async transform(chunk, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
           var _a16, _b, _c, _d;
           function updateToolInvocationPart(options) {
             var _a17;
-            const part2 = state.message.parts.find(
-              (part3) => isToolUIPart(part3) && part3.toolCallId === options.toolCallId
+            const part = state.message.parts.find(
+              (part2) => isToolUIPart(part2) && part2.toolCallId === options.toolCallId
             );
             const anyOptions = options;
-            const anyPart = part2;
-            if (part2 != null) {
-              part2.state = options.state;
+            const anyPart = part;
+            if (part != null) {
+              part.state = options.state;
               anyPart.input = anyOptions.input;
               anyPart.output = anyOptions.output;
               anyPart.errorText = anyOptions.errorText;
-              anyPart.providerExecuted = (_a17 = anyOptions.providerExecuted) != null ? _a17 : part2.providerExecuted;
-              if (anyOptions.providerMetadata != null && part2.state === "input-available") {
-                part2.callProviderMetadata = anyOptions.providerMetadata;
+              anyPart.providerExecuted = (_a17 = anyOptions.providerExecuted) != null ? _a17 : part.providerExecuted;
+              if (anyOptions.providerMetadata != null && part.state === "input-available") {
+                part.callProviderMetadata = anyOptions.providerMetadata;
               }
             } else {
               state.message.parts.push({
@@ -3194,31 +3195,31 @@ function processUIMessageStream({
               state.message.metadata = mergedMetadata;
             }
           }
-          switch (part.type) {
+          switch (chunk.type) {
             case "text-start": {
               const textPart = {
                 type: "text",
                 text: "",
-                providerMetadata: part.providerMetadata,
+                providerMetadata: chunk.providerMetadata,
                 state: "streaming"
               };
-              state.activeTextParts[part.id] = textPart;
+              state.activeTextParts[chunk.id] = textPart;
               state.message.parts.push(textPart);
               write();
               break;
             }
             case "text-delta": {
-              const textPart = state.activeTextParts[part.id];
-              textPart.text += part.delta;
-              textPart.providerMetadata = (_a16 = part.providerMetadata) != null ? _a16 : textPart.providerMetadata;
+              const textPart = state.activeTextParts[chunk.id];
+              textPart.text += chunk.delta;
+              textPart.providerMetadata = (_a16 = chunk.providerMetadata) != null ? _a16 : textPart.providerMetadata;
               write();
               break;
             }
             case "text-end": {
-              const textPart = state.activeTextParts[part.id];
+              const textPart = state.activeTextParts[chunk.id];
               textPart.state = "done";
-              textPart.providerMetadata = (_b = part.providerMetadata) != null ? _b : textPart.providerMetadata;
-              delete state.activeTextParts[part.id];
+              textPart.providerMetadata = (_b = chunk.providerMetadata) != null ? _b : textPart.providerMetadata;
+              delete state.activeTextParts[chunk.id];
               write();
               break;
             }
@@ -3226,34 +3227,34 @@ function processUIMessageStream({
               const reasoningPart = {
                 type: "reasoning",
                 text: "",
-                providerMetadata: part.providerMetadata,
+                providerMetadata: chunk.providerMetadata,
                 state: "streaming"
               };
-              state.activeReasoningParts[part.id] = reasoningPart;
+              state.activeReasoningParts[chunk.id] = reasoningPart;
               state.message.parts.push(reasoningPart);
               write();
               break;
             }
             case "reasoning-delta": {
-              const reasoningPart = state.activeReasoningParts[part.id];
-              reasoningPart.text += part.delta;
-              reasoningPart.providerMetadata = (_c = part.providerMetadata) != null ? _c : reasoningPart.providerMetadata;
+              const reasoningPart = state.activeReasoningParts[chunk.id];
+              reasoningPart.text += chunk.delta;
+              reasoningPart.providerMetadata = (_c = chunk.providerMetadata) != null ? _c : reasoningPart.providerMetadata;
               write();
               break;
             }
             case "reasoning-end": {
-              const reasoningPart = state.activeReasoningParts[part.id];
-              reasoningPart.providerMetadata = (_d = part.providerMetadata) != null ? _d : reasoningPart.providerMetadata;
+              const reasoningPart = state.activeReasoningParts[chunk.id];
+              reasoningPart.providerMetadata = (_d = chunk.providerMetadata) != null ? _d : reasoningPart.providerMetadata;
               reasoningPart.state = "done";
-              delete state.activeReasoningParts[part.id];
+              delete state.activeReasoningParts[chunk.id];
               write();
               break;
             }
             case "file": {
               state.message.parts.push({
                 type: "file",
-                mediaType: part.mediaType,
-                url: part.url
+                mediaType: chunk.mediaType,
+                url: chunk.url
               });
               write();
               break;
@@ -3261,10 +3262,10 @@ function processUIMessageStream({
             case "source-url": {
               state.message.parts.push({
                 type: "source-url",
-                sourceId: part.sourceId,
-                url: part.url,
-                title: part.title,
-                providerMetadata: part.providerMetadata
+                sourceId: chunk.sourceId,
+                url: chunk.url,
+                title: chunk.title,
+                providerMetadata: chunk.providerMetadata
               });
               write();
               break;
@@ -3272,40 +3273,40 @@ function processUIMessageStream({
             case "source-document": {
               state.message.parts.push({
                 type: "source-document",
-                sourceId: part.sourceId,
-                mediaType: part.mediaType,
-                title: part.title,
-                filename: part.filename,
-                providerMetadata: part.providerMetadata
+                sourceId: chunk.sourceId,
+                mediaType: chunk.mediaType,
+                title: chunk.title,
+                filename: chunk.filename,
+                providerMetadata: chunk.providerMetadata
               });
               write();
               break;
             }
             case "tool-input-start": {
               const toolInvocations = state.message.parts.filter(isToolUIPart);
-              state.partialToolCalls[part.toolCallId] = {
+              state.partialToolCalls[chunk.toolCallId] = {
                 text: "",
-                toolName: part.toolName,
+                toolName: chunk.toolName,
                 index: toolInvocations.length
               };
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
-                toolName: part.toolName,
+                toolCallId: chunk.toolCallId,
+                toolName: chunk.toolName,
                 state: "input-streaming",
                 input: void 0,
-                providerExecuted: part.providerExecuted
+                providerExecuted: chunk.providerExecuted
               });
               write();
               break;
             }
             case "tool-input-delta": {
-              const partialToolCall = state.partialToolCalls[part.toolCallId];
-              partialToolCall.text += part.inputTextDelta;
+              const partialToolCall = state.partialToolCalls[chunk.toolCallId];
+              partialToolCall.text += chunk.inputTextDelta;
               const { value: partialArgs } = await parsePartialJson(
                 partialToolCall.text
               );
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
+                toolCallId: chunk.toolCallId,
                 toolName: partialToolCall.toolName,
                 state: "input-streaming",
                 input: partialArgs
@@ -3315,24 +3316,24 @@ function processUIMessageStream({
             }
             case "tool-input-available": {
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
-                toolName: part.toolName,
+                toolCallId: chunk.toolCallId,
+                toolName: chunk.toolName,
                 state: "input-available",
-                input: part.input,
-                providerExecuted: part.providerExecuted,
-                providerMetadata: part.providerMetadata
+                input: chunk.input,
+                providerExecuted: chunk.providerExecuted,
+                providerMetadata: chunk.providerMetadata
               });
               write();
-              if (onToolCall && !part.providerExecuted) {
+              if (onToolCall && !chunk.providerExecuted) {
                 const result = await onToolCall({
-                  toolCall: part
+                  toolCall: chunk
                 });
                 if (result != null) {
                   updateToolInvocationPart({
-                    toolCallId: part.toolCallId,
-                    toolName: part.toolName,
+                    toolCallId: chunk.toolCallId,
+                    toolName: chunk.toolName,
                     state: "output-available",
-                    input: part.input,
+                    input: chunk.input,
                     output: result
                   });
                   write();
@@ -3346,7 +3347,7 @@ function processUIMessageStream({
                 throw new Error("tool_result must be preceded by a tool_call");
               }
               const toolInvocationIndex = toolInvocations.findIndex(
-                (invocation) => invocation.toolCallId === part.toolCallId
+                (invocation) => invocation.toolCallId === chunk.toolCallId
               );
               if (toolInvocationIndex === -1) {
                 throw new Error(
@@ -3357,12 +3358,12 @@ function processUIMessageStream({
                 toolInvocations[toolInvocationIndex]
               );
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
+                toolCallId: chunk.toolCallId,
                 toolName,
                 state: "output-available",
                 input: toolInvocations[toolInvocationIndex].input,
-                output: part.output,
-                providerExecuted: part.providerExecuted
+                output: chunk.output,
+                providerExecuted: chunk.providerExecuted
               });
               write();
               break;
@@ -3373,7 +3374,7 @@ function processUIMessageStream({
                 throw new Error("tool_result must be preceded by a tool_call");
               }
               const toolInvocationIndex = toolInvocations.findIndex(
-                (invocation) => invocation.toolCallId === part.toolCallId
+                (invocation) => invocation.toolCallId === chunk.toolCallId
               );
               if (toolInvocationIndex === -1) {
                 throw new Error(
@@ -3384,12 +3385,12 @@ function processUIMessageStream({
                 toolInvocations[toolInvocationIndex]
               );
               updateToolInvocationPart({
-                toolCallId: part.toolCallId,
+                toolCallId: chunk.toolCallId,
                 toolName,
                 state: "output-error",
                 input: toolInvocations[toolInvocationIndex].input,
-                errorText: part.errorText,
-                providerExecuted: part.providerExecuted
+                errorText: chunk.errorText,
+                providerExecuted: chunk.providerExecuted
               });
               write();
               break;
@@ -3404,62 +3405,65 @@ function processUIMessageStream({
               break;
             }
             case "start": {
-              if (part.messageId != null) {
-                state.message.id = part.messageId;
+              if (chunk.messageId != null) {
+                state.message.id = chunk.messageId;
               }
-              await updateMessageMetadata(part.messageMetadata);
-              if (part.messageId != null || part.messageMetadata != null) {
+              await updateMessageMetadata(chunk.messageMetadata);
+              if (chunk.messageId != null || chunk.messageMetadata != null) {
                 write();
               }
               break;
             }
             case "finish": {
-              await updateMessageMetadata(part.messageMetadata);
-              if (part.messageMetadata != null) {
+              await updateMessageMetadata(chunk.messageMetadata);
+              if (chunk.messageMetadata != null) {
                 write();
               }
               break;
             }
             case "message-metadata": {
-              await updateMessageMetadata(part.messageMetadata);
-              if (part.messageMetadata != null) {
+              await updateMessageMetadata(chunk.messageMetadata);
+              if (chunk.messageMetadata != null) {
                 write();
               }
               break;
             }
             case "error": {
-              onError == null ? void 0 : onError(new Error(part.errorText));
+              onError == null ? void 0 : onError(new Error(chunk.errorText));
               break;
             }
             default: {
-              if (isDataUIMessageChunk(part)) {
-                const dataPart = part;
-                if (dataPart.transient) {
-                  onData == null ? void 0 : onData(dataPart);
+              if (isDataUIMessageChunk(chunk)) {
+                if ((dataPartSchemas == null ? void 0 : dataPartSchemas[chunk.type]) != null) {
+                  await (0, import_provider_utils11.validateTypes)({
+                    value: chunk.data,
+                    schema: dataPartSchemas[chunk.type]
+                  });
+                }
+                const dataChunk = chunk;
+                if (dataChunk.transient) {
+                  onData == null ? void 0 : onData(dataChunk);
                   break;
                 }
-                const existingPart = dataPart.id != null ? state.message.parts.find(
-                  (partArg) => dataPart.type === partArg.type && dataPart.id === partArg.id
+                const existingUIPart = dataChunk.id != null ? state.message.parts.find(
+                  (chunkArg) => dataChunk.type === chunkArg.type && dataChunk.id === chunkArg.id
                 ) : void 0;
-                if (existingPart != null) {
-                  existingPart.data = isObject(existingPart.data) && isObject(dataPart.data) ? mergeObjects(existingPart.data, dataPart.data) : dataPart.data;
+                if (existingUIPart != null) {
+                  existingUIPart.data = dataChunk.data;
                 } else {
-                  state.message.parts.push(dataPart);
+                  state.message.parts.push(dataChunk);
                 }
-                onData == null ? void 0 : onData(dataPart);
+                onData == null ? void 0 : onData(dataChunk);
                 write();
               }
             }
           }
-          controller.enqueue(part);
+          controller.enqueue(chunk);
         });
       }
     })
   );
 }
-function isObject(value) {
-  return typeof value === "object" && value !== null;
-}
 
 // src/ui-message-stream/handle-ui-message-stream-finish.ts
 function handleUIMessageStreamFinish({
@@ -4882,10 +4886,10 @@ var DefaultStreamTextResult = class {
     sendFinish = true,
     onError = import_provider22.getErrorMessage
   } = {}) {
-    const responseMessageId = getResponseUIMessageId({
+    const responseMessageId = generateMessageId != null ? getResponseUIMessageId({
       originalMessages,
-      responseMessageId: this.generateId
-    });
+      responseMessageId: generateMessageId
+    }) : void 0;
     const baseStream = this.fullStream.pipeThrough(
       new TransformStream({
         transform: async (part, controller) => {
@@ -5039,8 +5043,8 @@ var DefaultStreamTextResult = class {
               if (sendStart) {
                 controller.enqueue({
                   type: "start",
-                  messageId: responseMessageId,
-                  messageMetadata: messageMetadataValue
+                  ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {},
+                  ...responseMessageId != null ? { messageId: responseMessageId } : {}
                 });
               }
               break;
@@ -5049,7 +5053,7 @@ var DefaultStreamTextResult = class {
               if (sendFinish) {
                 controller.enqueue({
                   type: "finish",
-                  messageMetadata: messageMetadataValue
+                  ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {}
                 });
               }
               break;
@@ -7456,6 +7460,28 @@ var doWrap = ({
   };
 };
 
+// src/middleware/wrap-provider.ts
+function wrapProvider({
+  provider,
+  languageModelMiddleware
+}) {
+  const wrappedProvider = {
+    languageModel(modelId) {
+      let model = provider.languageModel(modelId);
+      model = wrapLanguageModel({
+        model,
+        middleware: languageModelMiddleware
+      });
+      return model;
+    },
+    textEmbeddingModel: provider.textEmbeddingModel,
+    imageModel: provider.imageModel,
+    transcriptionModel: provider.transcriptionModel,
+    speechModel: provider.speechModel
+  };
+  return wrappedProvider;
+}
+
 // src/registry/custom-provider.ts
 var import_provider27 = require("@ai-sdk/provider");
 function customProvider({
@@ -7544,10 +7570,12 @@ _a15 = symbol15;
 // src/registry/provider-registry.ts
 var import_provider29 = require("@ai-sdk/provider");
 function createProviderRegistry(providers, {
-  separator = ":"
+  separator = ":",
+  languageModelMiddleware
 } = {}) {
   const registry = new DefaultProviderRegistry({
-    separator
+    separator,
+    languageModelMiddleware
   });
   for (const [id, provider] of Object.entries(providers)) {
     registry.registerProvider({ id, provider });
@@ -7556,9 +7584,13 @@ function createProviderRegistry(providers, {
 }
 var experimental_createProviderRegistry = createProviderRegistry;
 var DefaultProviderRegistry = class {
-  constructor({ separator }) {
+  constructor({
+    separator,
+    languageModelMiddleware
+  }) {
     this.providers = {};
     this.separator = separator;
+    this.languageModelMiddleware = languageModelMiddleware;
   }
   registerProvider({
     id,
@@ -7592,13 +7624,19 @@ var DefaultProviderRegistry = class {
   languageModel(id) {
     var _a16, _b;
     const [providerId, modelId] = this.splitId(id, "languageModel");
-    const model = (_b = (_a16 = this.getProvider(providerId, "languageModel")).languageModel) == null ? void 0 : _b.call(
+    let model = (_b = (_a16 = this.getProvider(providerId, "languageModel")).languageModel) == null ? void 0 : _b.call(
       _a16,
       modelId
     );
     if (model == null) {
       throw new import_provider29.NoSuchModelError({ modelId: id, modelType: "languageModel" });
     }
+    if (this.languageModelMiddleware != null) {
+      model = wrapLanguageModel({
+        model,
+        middleware: this.languageModelMiddleware
+      });
+    }
     return model;
   }
   textEmbeddingModel(id) {
@@ -9337,6 +9375,7 @@ function readUIMessageStream({
   tool,
   toolModelMessageSchema,
   userModelMessageSchema,
-  wrapLanguageModel
+  wrapLanguageModel,
+  wrapProvider
 });
 //# sourceMappingURL=index.js.map