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

package/dist/index.mjs

@@ -3086,24 +3086,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({
@@ -3130,31 +3130,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;
             }
@@ -3162,34 +3162,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;
@@ -3197,10 +3197,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;
@@ -3208,40 +3208,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
@@ -3251,24 +3251,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();
@@ -3282,7 +3282,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(
@@ -3293,12 +3293,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;
@@ -3309,7 +3309,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(
@@ -3320,12 +3320,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;
@@ -3340,62 +3340,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 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({
@@ -4818,10 +4821,10 @@ var DefaultStreamTextResult = class {
     sendFinish = true,
     onError = getErrorMessage5
   } = {}) {
-    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) => {
@@ -4975,8 +4978,8 @@ var DefaultStreamTextResult = class {
               if (sendStart) {
                 controller.enqueue({
                   type: "start",
-                  messageId: responseMessageId,
-                  messageMetadata: messageMetadataValue
+                  ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {},
+                  ...responseMessageId != null ? { messageId: responseMessageId } : {}
                 });
               }
               break;
@@ -4985,7 +4988,7 @@ var DefaultStreamTextResult = class {
               if (sendFinish) {
                 controller.enqueue({
                   type: "finish",
-                  messageMetadata: messageMetadataValue
+                  ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {}
                 });
               }
               break;
@@ -7412,6 +7415,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
 import {
   NoSuchModelError as NoSuchModelError2
@@ -7504,10 +7529,12 @@ import {
   NoSuchModelError as NoSuchModelError4
 } from "@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 });
@@ -7516,9 +7543,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,
@@ -7552,13 +7583,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 NoSuchModelError4({ modelId: id, modelType: "languageModel" });
     }
+    if (this.languageModelMiddleware != null) {
+      model = wrapLanguageModel({
+        model,
+        middleware: this.languageModelMiddleware
+      });
+    }
     return model;
   }
   textEmbeddingModel(id) {
@@ -9303,6 +9340,7 @@ export {
   tool2 as tool,
   toolModelMessageSchema,
   userModelMessageSchema,
-  wrapLanguageModel
+  wrapLanguageModel,
+  wrapProvider
 };
 //# sourceMappingURL=index.mjs.map