ai 5.0.0-beta.20 → 5.0.0-beta.22

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");
@@ -1874,7 +1875,11 @@ function toResponseMessages({
   ).filter((part) => part.type !== "text" || part.text.length > 0).map((part) => {
     switch (part.type) {
       case "text":
-        return part;
+        return {
+          type: "text",
+          text: part.text,
+          providerOptions: part.providerMetadata
+        };
       case "reasoning":
         return {
           type: "reasoning",
@@ -2616,16 +2621,19 @@ var import_v47 = require("zod/v4");
 var uiMessageChunkSchema = import_v47.z.union([
   import_v47.z.strictObject({
     type: import_v47.z.literal("text-start"),
-    id: import_v47.z.string()
+    id: import_v47.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("text-delta"),
     id: import_v47.z.string(),
-    delta: import_v47.z.string()
+    delta: import_v47.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("text-end"),
-    id: import_v47.z.string()
+    id: import_v47.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("error"),
@@ -2647,7 +2655,8 @@ var uiMessageChunkSchema = import_v47.z.union([
     toolCallId: import_v47.z.string(),
     toolName: import_v47.z.string(),
     input: import_v47.z.unknown(),
-    providerExecuted: import_v47.z.boolean().optional()
+    providerExecuted: import_v47.z.boolean().optional(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("tool-output-available"),
@@ -2664,23 +2673,23 @@ var uiMessageChunkSchema = import_v47.z.union([
   import_v47.z.strictObject({
     type: import_v47.z.literal("reasoning"),
     text: import_v47.z.string(),
-    providerMetadata: import_v47.z.record(import_v47.z.string(), import_v47.z.any()).optional()
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("reasoning-start"),
     id: import_v47.z.string(),
-    providerMetadata: import_v47.z.record(import_v47.z.string(), import_v47.z.any()).optional()
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("reasoning-delta"),
     id: import_v47.z.string(),
     delta: import_v47.z.string(),
-    providerMetadata: import_v47.z.record(import_v47.z.string(), import_v47.z.any()).optional()
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("reasoning-end"),
     id: import_v47.z.string(),
-    providerMetadata: import_v47.z.record(import_v47.z.string(), import_v47.z.any()).optional()
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("reasoning-part-finish")
@@ -2690,8 +2699,7 @@ var uiMessageChunkSchema = import_v47.z.union([
     sourceId: import_v47.z.string(),
     url: import_v47.z.string(),
     title: import_v47.z.string().optional(),
-    providerMetadata: import_v47.z.any().optional()
-    // Use z.any() for generic metadata
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("source-document"),
@@ -2699,13 +2707,13 @@ var uiMessageChunkSchema = import_v47.z.union([
     mediaType: import_v47.z.string(),
     title: import_v47.z.string(),
     filename: import_v47.z.string().optional(),
-    providerMetadata: import_v47.z.any().optional()
-    // Use z.any() for generic metadata
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("file"),
     url: import_v47.z.string(),
-    mediaType: import_v47.z.string()
+    mediaType: import_v47.z.string(),
+    providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
     type: import_v47.z.string().startsWith("data-"),
@@ -3143,22 +3151,25 @@ function processUIMessageStream({
 }) {
   return stream.pipeThrough(
     new TransformStream({
-      async transform(part, controller) {
+      async transform(chunk, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
-          var _a16, _b;
+          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;
+              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({
                 type: `tool-${options.toolName}`,
@@ -3167,7 +3178,8 @@ function processUIMessageStream({
                 input: anyOptions.input,
                 output: anyOptions.output,
                 errorText: anyOptions.errorText,
-                providerExecuted: anyOptions.providerExecuted
+                providerExecuted: anyOptions.providerExecuted,
+                ...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
               });
             }
           }
@@ -3183,27 +3195,31 @@ function processUIMessageStream({
               state.message.metadata = mergedMetadata;
             }
           }
-          switch (part.type) {
+          switch (chunk.type) {
             case "text-start": {
               const textPart = {
                 type: "text",
                 text: "",
+                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": {
-              state.activeTextParts[part.id].text += part.delta;
+              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";
-              delete state.activeTextParts[part.id];
+              textPart.providerMetadata = (_b = chunk.providerMetadata) != null ? _b : textPart.providerMetadata;
+              delete state.activeTextParts[chunk.id];
               write();
               break;
             }
@@ -3211,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 = (_a16 = part.providerMetadata) != null ? _a16 : 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 = (_b = part.providerMetadata) != null ? _b : 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;
@@ -3246,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;
@@ -3257,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
@@ -3300,23 +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
+                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();
@@ -3330,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(
@@ -3341,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;
@@ -3357,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(
@@ -3368,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;
@@ -3388,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({
@@ -4107,7 +4127,7 @@ var DefaultStreamTextResult = class {
     let activeReasoningContent = {};
     const eventProcessor = new TransformStream({
       async transform(chunk, controller) {
-        var _a16, _b;
+        var _a16, _b, _c;
         controller.enqueue(chunk);
         const { part } = chunk;
         if (part.type === "text" || part.type === "reasoning" || part.type === "source" || part.type === "tool-call" || part.type === "tool-result" || part.type === "tool-input-start" || part.type === "tool-input-delta" || part.type === "raw") {
@@ -4137,7 +4157,7 @@ var DefaultStreamTextResult = class {
             return;
           }
           activeText.text += part.text;
-          activeText.providerMetadata = part.providerMetadata;
+          activeText.providerMetadata = (_a16 = part.providerMetadata) != null ? _a16 : activeText.providerMetadata;
         }
         if (part.type === "text-end") {
           delete activeTextContent[part.id];
@@ -4163,7 +4183,7 @@ var DefaultStreamTextResult = class {
             return;
           }
           activeReasoning.text += part.text;
-          activeReasoning.providerMetadata = (_a16 = part.providerMetadata) != null ? _a16 : activeReasoning.providerMetadata;
+          activeReasoning.providerMetadata = (_b = part.providerMetadata) != null ? _b : activeReasoning.providerMetadata;
         }
         if (part.type === "reasoning-end") {
           const activeReasoning = activeReasoningContent[part.id];
@@ -4177,7 +4197,7 @@ var DefaultStreamTextResult = class {
             });
             return;
           }
-          activeReasoning.providerMetadata = (_b = part.providerMetadata) != null ? _b : activeReasoning.providerMetadata;
+          activeReasoning.providerMetadata = (_c = part.providerMetadata) != null ? _c : activeReasoning.providerMetadata;
           delete activeReasoningContent[part.id];
         }
         if (part.type === "file") {
@@ -4866,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) => {
@@ -4877,26 +4897,35 @@ var DefaultStreamTextResult = class {
           const partType = part.type;
           switch (partType) {
             case "text-start": {
-              controller.enqueue({ type: "text-start", id: part.id });
+              controller.enqueue({
+                type: "text-start",
+                id: part.id,
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+              });
               break;
             }
             case "text": {
               controller.enqueue({
                 type: "text-delta",
                 id: part.id,
-                delta: part.text
+                delta: part.text,
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
               });
               break;
             }
             case "text-end": {
-              controller.enqueue({ type: "text-end", id: part.id });
+              controller.enqueue({
+                type: "text-end",
+                id: part.id,
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+              });
               break;
             }
             case "reasoning-start": {
               controller.enqueue({
                 type: "reasoning-start",
                 id: part.id,
-                providerMetadata: part.providerMetadata
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
               });
               break;
             }
@@ -4906,7 +4935,7 @@ var DefaultStreamTextResult = class {
                   type: "reasoning-delta",
                   id: part.id,
                   delta: part.text,
-                  providerMetadata: part.providerMetadata
+                  ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
                 });
               }
               break;
@@ -4915,7 +4944,7 @@ var DefaultStreamTextResult = class {
               controller.enqueue({
                 type: "reasoning-end",
                 id: part.id,
-                providerMetadata: part.providerMetadata
+                ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
               });
               break;
             }
@@ -4934,7 +4963,7 @@ var DefaultStreamTextResult = class {
                   sourceId: part.id,
                   url: part.url,
                   title: part.title,
-                  providerMetadata: part.providerMetadata
+                  ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
                 });
               }
               if (sendSources && part.sourceType === "document") {
@@ -4944,7 +4973,7 @@ var DefaultStreamTextResult = class {
                   mediaType: part.mediaType,
                   title: part.title,
                   filename: part.filename,
-                  providerMetadata: part.providerMetadata
+                  ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
                 });
               }
               break;
@@ -4972,7 +5001,8 @@ var DefaultStreamTextResult = class {
                 toolCallId: part.toolCallId,
                 toolName: part.toolName,
                 input: part.input,
-                providerExecuted: part.providerExecuted
+                providerExecuted: part.providerExecuted,
+                providerMetadata: part.providerMetadata
               });
               break;
             }
@@ -5013,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;
@@ -5023,7 +5053,7 @@ var DefaultStreamTextResult = class {
               if (sendFinish) {
                 controller.enqueue({
                   type: "finish",
-                  messageMetadata: messageMetadataValue
+                  ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {}
                 });
               }
               break;
@@ -7430,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({
@@ -7518,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 });
@@ -7530,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,
@@ -7566,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) {
@@ -8951,7 +9015,8 @@ function convertToModelMessages(messages, options) {
               if (part.type === "text") {
                 content.push({
                   type: "text",
-                  text: part.text
+                  text: part.text,
+                  ...part.providerMetadata != null ? { providerOptions: part.providerMetadata } : {}
                 });
               } else if (part.type === "file") {
                 content.push({
@@ -8978,7 +9043,8 @@ function convertToModelMessages(messages, options) {
                     toolCallId: part.toolCallId,
                     toolName,
                     input: part.input,
-                    providerExecuted: part.providerExecuted
+                    providerExecuted: part.providerExecuted,
+                    ...part.callProviderMetadata != null ? { providerOptions: part.callProviderMetadata } : {}
                   });
                   if (part.providerExecuted === true && (part.state === "output-available" || part.state === "output-error")) {
                     content.push({
@@ -9309,6 +9375,7 @@ function readUIMessageStream({
   tool,
   toolModelMessageSchema,
   userModelMessageSchema,
-  wrapLanguageModel
+  wrapLanguageModel,
+  wrapProvider
 });
 //# sourceMappingURL=index.js.map