ai 5.0.0-alpha.12 → 5.0.0-alpha.13

package/dist/index.js

@@ -54,7 +54,7 @@ __export(src_exports, {
   ToolExecutionError: () => ToolExecutionError,
   TypeValidationError: () => import_provider16.TypeValidationError,
   UnsupportedFunctionalityError: () => import_provider16.UnsupportedFunctionalityError,
-  asSchema: () => import_provider_utils25.asSchema,
+  asSchema: () => import_provider_utils26.asSchema,
   assistantModelMessageSchema: () => assistantModelMessageSchema,
   callCompletionApi: () => callCompletionApi,
   convertFileListToFileUIParts: () => convertFileListToFileUIParts,
@@ -66,7 +66,7 @@ __export(src_exports, {
   coreToolMessageSchema: () => coreToolMessageSchema,
   coreUserMessageSchema: () => coreUserMessageSchema,
   cosineSimilarity: () => cosineSimilarity,
-  createIdGenerator: () => import_provider_utils25.createIdGenerator,
+  createIdGenerator: () => import_provider_utils26.createIdGenerator,
   createProviderRegistry: () => createProviderRegistry,
   createTextStreamResponse: () => createTextStreamResponse,
   createUIMessageStream: () => createUIMessageStream,
@@ -82,14 +82,15 @@ __export(src_exports, {
   experimental_generateSpeech: () => generateSpeech,
   experimental_transcribe: () => transcribe,
   extractReasoningMiddleware: () => extractReasoningMiddleware,
-  generateId: () => import_provider_utils25.generateId,
+  generateId: () => import_provider_utils26.generateId,
   generateObject: () => generateObject,
   generateText: () => generateText,
   getTextFromDataUrl: () => getTextFromDataUrl,
-  getToolInvocations: () => getToolInvocations,
+  getToolName: () => getToolName,
   hasToolCall: () => hasToolCall,
   isDeepEqualData: () => isDeepEqualData,
-  jsonSchema: () => import_provider_utils25.jsonSchema,
+  isToolUIPart: () => isToolUIPart,
+  jsonSchema: () => import_provider_utils26.jsonSchema,
   modelMessageSchema: () => modelMessageSchema,
   parsePartialJson: () => parsePartialJson,
   pipeTextStreamToResponse: () => pipeTextStreamToResponse,
@@ -107,7 +108,7 @@ __export(src_exports, {
   wrapLanguageModel: () => wrapLanguageModel
 });
 module.exports = __toCommonJS(src_exports);
-var import_provider_utils25 = require("@ai-sdk/provider-utils");
+var import_provider_utils26 = require("@ai-sdk/provider-utils");
 
 // src/error/index.ts
 var import_provider16 = require("@ai-sdk/provider");
@@ -540,42 +541,45 @@ var import_provider_utils = require("@ai-sdk/provider-utils");
 // src/ui-message-stream/ui-message-stream-parts.ts
 var import_zod = require("zod");
 var uiMessageStreamPartSchema = import_zod.z.union([
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("text"),
     text: import_zod.z.string()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("error"),
     errorText: import_zod.z.string()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("tool-call-streaming-start"),
     toolCallId: import_zod.z.string(),
     toolName: import_zod.z.string()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("tool-call-delta"),
     toolCallId: import_zod.z.string(),
     argsTextDelta: import_zod.z.string()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("tool-call"),
     toolCallId: import_zod.z.string(),
     toolName: import_zod.z.string(),
     args: import_zod.z.unknown()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("tool-result"),
     toolCallId: import_zod.z.string(),
     result: import_zod.z.unknown(),
     providerMetadata: import_zod.z.any().optional()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("reasoning"),
     text: import_zod.z.string(),
     providerMetadata: import_zod.z.record(import_zod.z.any()).optional()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
+    type: import_zod.z.literal("reasoning-part-finish")
+  }),
+  import_zod.z.strictObject({
     type: import_zod.z.literal("source-url"),
     sourceId: import_zod.z.string(),
     url: import_zod.z.string(),
@@ -583,7 +587,7 @@ var uiMessageStreamPartSchema = import_zod.z.union([
     providerMetadata: import_zod.z.any().optional()
     // Use z.any() for generic metadata
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("source-document"),
     sourceId: import_zod.z.string(),
     mediaType: import_zod.z.string(),
@@ -592,39 +596,34 @@ var uiMessageStreamPartSchema = import_zod.z.union([
     providerMetadata: import_zod.z.any().optional()
     // Use z.any() for generic metadata
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("file"),
     url: import_zod.z.string(),
     mediaType: import_zod.z.string()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.string().startsWith("data-"),
     id: import_zod.z.string().optional(),
     data: import_zod.z.unknown()
   }),
-  import_zod.z.object({
-    type: import_zod.z.literal("metadata"),
-    value: import_zod.z.object({ metadata: import_zod.z.unknown() })
+  import_zod.z.strictObject({
+    type: import_zod.z.literal("start-step")
   }),
-  import_zod.z.object({
-    type: import_zod.z.literal("start-step"),
-    metadata: import_zod.z.unknown().optional()
+  import_zod.z.strictObject({
+    type: import_zod.z.literal("finish-step")
   }),
-  import_zod.z.object({
-    type: import_zod.z.literal("finish-step"),
-    metadata: import_zod.z.unknown().optional()
-  }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("start"),
     messageId: import_zod.z.string().optional(),
-    metadata: import_zod.z.unknown().optional()
+    messageMetadata: import_zod.z.unknown().optional()
   }),
-  import_zod.z.object({
+  import_zod.z.strictObject({
     type: import_zod.z.literal("finish"),
-    metadata: import_zod.z.unknown().optional()
+    messageMetadata: import_zod.z.unknown().optional()
   }),
-  import_zod.z.object({
-    type: import_zod.z.literal("reasoning-part-finish")
+  import_zod.z.strictObject({
+    type: import_zod.z.literal("message-metadata"),
+    messageMetadata: import_zod.z.unknown()
   })
 ]);
 function isDataUIMessageStreamPart(part) {
@@ -1314,27 +1313,26 @@ async function parsePartialJson(jsonText) {
   return { value: void 0, state: "failed-parse" };
 }
 
-// src/ui/get-tool-invocations.ts
-function getToolInvocations(message) {
-  return message.parts.filter(
-    (part) => part.type === "tool-invocation"
-  ).map((part) => part.toolInvocation);
+// src/ui/ui-messages.ts
+function isToolUIPart(part) {
+  return part.type.startsWith("tool-");
+}
+function getToolName(part) {
+  return part.type.split("-")[1];
 }
 
 // src/ui/process-ui-message-stream.ts
 function createStreamingUIMessageState({
   lastMessage,
-  newMessageId = ""
-} = {}) {
-  const isContinuation = (lastMessage == null ? void 0 : lastMessage.role) === "assistant";
-  const message = isContinuation ? lastMessage : {
-    id: newMessageId,
-    metadata: {},
-    role: "assistant",
-    parts: []
-  };
+  messageId
+}) {
   return {
-    message,
+    message: (lastMessage == null ? void 0 : lastMessage.role) === "assistant" ? lastMessage : {
+      id: messageId,
+      metadata: void 0,
+      role: "assistant",
+      parts: []
+    },
     activeTextPart: void 0,
     activeReasoningPart: void 0,
     partialToolCalls: {}
@@ -1351,16 +1349,21 @@ function processUIMessageStream({
     new TransformStream({
       async transform(part, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
-          function updateToolInvocationPart(toolCallId, invocation) {
+          function updateToolInvocationPart(options) {
             const part2 = state.message.parts.find(
-              (part3) => isToolInvocationUIPart(part3) && part3.toolInvocation.toolCallId === toolCallId
+              (part3) => isToolUIPart(part3) && part3.toolCallId === options.toolCallId
             );
             if (part2 != null) {
-              part2.toolInvocation = invocation;
+              part2.state = options.state;
+              part2.args = options.args;
+              part2.result = options.result;
             } else {
               state.message.parts.push({
-                type: "tool-invocation",
-                toolInvocation: invocation
+                type: `tool-${options.toolName}`,
+                toolCallId: options.toolCallId,
+                state: options.state,
+                args: options.args,
+                result: options.result
               });
             }
           }
@@ -1444,16 +1447,16 @@ function processUIMessageStream({
               break;
             }
             case "tool-call-streaming-start": {
-              const toolInvocations = getToolInvocations(state.message);
+              const toolInvocations = state.message.parts.filter(isToolUIPart);
               state.partialToolCalls[part.toolCallId] = {
                 text: "",
                 toolName: part.toolName,
                 index: toolInvocations.length
               };
-              updateToolInvocationPart(part.toolCallId, {
-                state: "partial-call",
+              updateToolInvocationPart({
                 toolCallId: part.toolCallId,
                 toolName: part.toolName,
+                state: "partial-call",
                 args: void 0
               });
               write();
@@ -1465,20 +1468,20 @@ function processUIMessageStream({
               const { value: partialArgs } = await parsePartialJson(
                 partialToolCall.text
               );
-              updateToolInvocationPart(part.toolCallId, {
-                state: "partial-call",
+              updateToolInvocationPart({
                 toolCallId: part.toolCallId,
                 toolName: partialToolCall.toolName,
+                state: "partial-call",
                 args: partialArgs
               });
               write();
               break;
             }
             case "tool-call": {
-              updateToolInvocationPart(part.toolCallId, {
-                state: "call",
+              updateToolInvocationPart({
                 toolCallId: part.toolCallId,
                 toolName: part.toolName,
+                state: "call",
                 args: part.args
               });
               write();
@@ -1487,10 +1490,10 @@ function processUIMessageStream({
                   toolCall: part
                 });
                 if (result != null) {
-                  updateToolInvocationPart(part.toolCallId, {
-                    state: "result",
+                  updateToolInvocationPart({
                     toolCallId: part.toolCallId,
                     toolName: part.toolName,
+                    state: "result",
                     args: part.args,
                     result
                   });
@@ -1500,7 +1503,7 @@ function processUIMessageStream({
               break;
             }
             case "tool-result": {
-              const toolInvocations = getToolInvocations(state.message);
+              const toolInvocations = state.message.parts.filter(isToolUIPart);
               if (toolInvocations == null) {
                 throw new Error("tool_result must be preceded by a tool_call");
               }
@@ -1512,9 +1515,14 @@ function processUIMessageStream({
                   "tool_result must be preceded by a tool_call with the same toolCallId"
                 );
               }
-              updateToolInvocationPart(part.toolCallId, {
-                ...toolInvocations[toolInvocationIndex],
+              const toolName = getToolName(
+                toolInvocations[toolInvocationIndex]
+              );
+              updateToolInvocationPart({
+                toolCallId: part.toolCallId,
+                toolName,
                 state: "result",
+                args: toolInvocations[toolInvocationIndex].args,
                 result: part.result
               });
               write();
@@ -1522,39 +1530,33 @@ function processUIMessageStream({
             }
             case "start-step": {
               state.message.parts.push({ type: "step-start" });
-              await updateMessageMetadata(part.metadata);
-              write();
               break;
             }
             case "finish-step": {
               state.activeTextPart = void 0;
               state.activeReasoningPart = void 0;
-              await updateMessageMetadata(part.metadata);
-              if (part.metadata != null) {
-                write();
-              }
               break;
             }
             case "start": {
               if (part.messageId != null) {
                 state.message.id = part.messageId;
               }
-              await updateMessageMetadata(part.metadata);
-              if (part.messageId != null || part.metadata != null) {
+              await updateMessageMetadata(part.messageMetadata);
+              if (part.messageId != null || part.messageMetadata != null) {
                 write();
               }
               break;
             }
             case "finish": {
-              await updateMessageMetadata(part.metadata);
-              if (part.metadata != null) {
+              await updateMessageMetadata(part.messageMetadata);
+              if (part.messageMetadata != null) {
                 write();
               }
               break;
             }
-            case "metadata": {
-              await updateMessageMetadata(part.metadata);
-              if (part.metadata != null) {
+            case "message-metadata": {
+              await updateMessageMetadata(part.messageMetadata);
+              if (part.messageMetadata != null) {
                 write();
               }
               break;
@@ -1582,9 +1584,6 @@ function processUIMessageStream({
     })
   );
 }
-function isToolInvocationUIPart(part) {
-  return part.type === "tool-invocation";
-}
 function isObject(value) {
   return typeof value === "object" && value !== null;
 }
@@ -1619,8 +1618,8 @@ function isAssistantMessageWithCompletedToolCalls(message) {
   const lastStepStartIndex = message.parts.reduce((lastIndex, part, index) => {
     return part.type === "step-start" ? index : lastIndex;
   }, -1);
-  const lastStepToolInvocations = message.parts.slice(lastStepStartIndex + 1).filter((part) => part.type === "tool-invocation");
-  return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every((part) => "result" in part.toolInvocation);
+  const lastStepToolInvocations = message.parts.slice(lastStepStartIndex + 1).filter(isToolUIPart);
+  return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every((part) => part.state === "result");
 }
 
 // src/ui/chat.ts
@@ -1697,7 +1696,7 @@ var AbstractChat = class {
       result
     }) => {
       this.jobExecutor.run(async () => {
-        updateToolCallResult({
+        updateToolResult({
           messages: this.state.messages,
           toolCallId,
           toolResult: result
@@ -1723,7 +1722,6 @@ var AbstractChat = class {
         return;
       if ((_a17 = this.activeResponse) == null ? void 0 : _a17.abortController) {
         this.activeResponse.abortController.abort();
-        this.activeResponse.abortController = void 0;
       }
     };
     this.id = id;
@@ -1784,7 +1782,7 @@ var AbstractChat = class {
       const activeResponse = {
         state: createStreamingUIMessageState({
           lastMessage: this.state.snapshot(lastMessage),
-          newMessageId: this.generateId()
+          messageId: this.generateId()
         }),
         abortController: new AbortController()
       };
@@ -1861,23 +1859,20 @@ var AbstractChat = class {
     }
   }
 };
-function updateToolCallResult({
+function updateToolResult({
   messages,
   toolCallId,
   toolResult: result
 }) {
   const lastMessage = messages[messages.length - 1];
-  const invocationPart = lastMessage.parts.find(
-    (part) => part.type === "tool-invocation" && part.toolInvocation.toolCallId === toolCallId
+  const toolPart = lastMessage.parts.find(
+    (part) => isToolUIPart(part) && part.toolCallId === toolCallId
   );
-  if (invocationPart == null) {
+  if (toolPart == null) {
     return;
   }
-  invocationPart.toolInvocation = {
-    ...invocationPart.toolInvocation,
-    state: "result",
-    result
-  };
+  toolPart.state = "result";
+  toolPart.result = result;
 }
 
 // src/ui/convert-to-model-messages.ts
@@ -1918,75 +1913,71 @@ function convertToModelMessages(messages, options) {
             }
             const content = [];
             for (const part of block) {
-              switch (part.type) {
-                case "text": {
-                  content.push(part);
-                  break;
-                }
-                case "file": {
-                  content.push({
-                    type: "file",
-                    mediaType: part.mediaType,
-                    data: part.url
-                  });
-                  break;
-                }
-                case "reasoning": {
-                  content.push({
-                    type: "reasoning",
-                    text: part.text,
-                    providerOptions: part.providerMetadata
+              if (part.type === "text") {
+                content.push(part);
+              } else if (part.type === "file") {
+                content.push({
+                  type: "file",
+                  mediaType: part.mediaType,
+                  data: part.url
+                });
+              } else if (part.type === "reasoning") {
+                content.push({
+                  type: "reasoning",
+                  text: part.text,
+                  providerOptions: part.providerMetadata
+                });
+              } else if (isToolUIPart(part)) {
+                const toolName = getToolName(part);
+                if (part.state === "partial-call") {
+                  throw new MessageConversionError({
+                    originalMessage: message,
+                    message: `Partial tool call is not supported: ${part.toolCallId}`
                   });
-                  break;
-                }
-                case "tool-invocation":
+                } else {
                   content.push({
                     type: "tool-call",
-                    toolCallId: part.toolInvocation.toolCallId,
-                    toolName: part.toolInvocation.toolName,
-                    args: part.toolInvocation.args
+                    toolCallId: part.toolCallId,
+                    toolName,
+                    args: part.args
                   });
-                  break;
-                default: {
-                  const _exhaustiveCheck = part;
-                  throw new Error(`Unsupported part: ${_exhaustiveCheck}`);
                 }
+              } else {
+                const _exhaustiveCheck = part;
+                throw new Error(`Unsupported part: ${_exhaustiveCheck}`);
               }
             }
             modelMessages.push({
               role: "assistant",
               content
             });
-            const stepInvocations = block.filter(
-              (part) => part.type === "tool-invocation"
-            ).map((part) => part.toolInvocation);
-            if (stepInvocations.length > 0) {
+            const toolParts = block.filter(isToolUIPart);
+            if (toolParts.length > 0) {
               modelMessages.push({
                 role: "tool",
-                content: stepInvocations.map(
-                  (toolInvocation) => {
-                    if (!("result" in toolInvocation)) {
-                      throw new MessageConversionError({
-                        originalMessage: message,
-                        message: "ToolInvocation must have a result: " + JSON.stringify(toolInvocation)
-                      });
-                    }
-                    const { toolCallId, toolName, result } = toolInvocation;
-                    const tool2 = tools[toolName];
-                    return (tool2 == null ? void 0 : tool2.experimental_toToolResultContent) != null ? {
-                      type: "tool-result",
-                      toolCallId,
-                      toolName,
-                      result: tool2.experimental_toToolResultContent(result),
-                      experimental_content: tool2.experimental_toToolResultContent(result)
-                    } : {
-                      type: "tool-result",
-                      toolCallId,
-                      toolName,
-                      result
-                    };
+                content: toolParts.map((toolPart) => {
+                  if (toolPart.state !== "result") {
+                    throw new MessageConversionError({
+                      originalMessage: message,
+                      message: "ToolInvocation must have a result: " + JSON.stringify(toolPart)
+                    });
                   }
-                )
+                  const toolName = getToolName(toolPart);
+                  const { toolCallId, result } = toolPart;
+                  const tool2 = tools[toolName];
+                  return (tool2 == null ? void 0 : tool2.experimental_toToolResultContent) != null ? {
+                    type: "tool-result",
+                    toolCallId,
+                    toolName,
+                    result: tool2.experimental_toToolResultContent(result),
+                    experimental_content: tool2.experimental_toToolResultContent(result)
+                  } : {
+                    type: "tool-result",
+                    toolCallId,
+                    toolName,
+                    result
+                  };
+                })
               });
             }
             block = [];
@@ -1994,18 +1985,10 @@ function convertToModelMessages(messages, options) {
           var processBlock = processBlock2;
           let block = [];
           for (const part of message.parts) {
-            switch (part.type) {
-              case "text":
-              case "reasoning":
-              case "file":
-              case "tool-invocation": {
-                block.push(part);
-                break;
-              }
-              case "step-start": {
-                processBlock2();
-                break;
-              }
+            if (part.type === "text" || part.type === "reasoning" || part.type === "file" || isToolUIPart(part)) {
+              block.push(part);
+            } else if (part.type === "step-start") {
+              processBlock2();
             }
           }
           processBlock2();
@@ -2135,9 +2118,12 @@ var TextStreamChatTransport = class {
   }
 };
 
+// src/ui-message-stream/create-ui-message-stream.ts
+var import_provider_utils6 = require("@ai-sdk/provider-utils");
+
 // src/ui-message-stream/handle-ui-message-stream-finish.ts
 function handleUIMessageStreamFinish({
-  newMessageId,
+  messageId,
   originalMessages = [],
   onFinish,
   stream
@@ -2145,19 +2131,30 @@ function handleUIMessageStreamFinish({
   if (onFinish == null) {
     return stream;
   }
-  const lastMessage = originalMessages[originalMessages.length - 1];
-  const isContinuation = (lastMessage == null ? void 0 : lastMessage.role) === "assistant";
-  const messageId = isContinuation ? lastMessage.id : newMessageId;
+  const lastMessage = originalMessages == null ? void 0 : originalMessages[originalMessages.length - 1];
   const state = createStreamingUIMessageState({
-    lastMessage: structuredClone(lastMessage),
-    newMessageId: messageId
+    lastMessage: lastMessage ? structuredClone(lastMessage) : void 0,
+    messageId
+    // will be overridden by the stream
   });
   const runUpdateMessageJob = async (job) => {
     await job({ state, write: () => {
     } });
   };
   return processUIMessageStream({
-    stream,
+    stream: stream.pipeThrough(
+      new TransformStream({
+        transform(chunk, controller) {
+          if (chunk.type === "start") {
+            const startChunk = chunk;
+            if (startChunk.messageId == null) {
+              startChunk.messageId = messageId;
+            }
+          }
+          controller.enqueue(chunk);
+        }
+      })
+    ),
     runUpdateMessageJob
   }).pipeThrough(
     new TransformStream({
@@ -2165,12 +2162,12 @@ function handleUIMessageStreamFinish({
         controller.enqueue(chunk);
       },
       flush() {
-        const isContinuation2 = state.message.id === (lastMessage == null ? void 0 : lastMessage.id);
+        const isContinuation = state.message.id === (lastMessage == null ? void 0 : lastMessage.id);
         onFinish({
-          isContinuation: isContinuation2,
+          isContinuation,
           responseMessage: state.message,
           messages: [
-            ...isContinuation2 ? originalMessages.slice(0, -1) : originalMessages,
+            ...isContinuation ? originalMessages.slice(0, -1) : originalMessages,
             state.message
           ]
         });
@@ -2185,7 +2182,8 @@ function createUIMessageStream({
   onError = () => "An error occurred.",
   // mask error messages for safety by default
   originalMessages,
-  onFinish
+  onFinish,
+  generateId: generateId3 = import_provider_utils6.generateId
 }) {
   let controller;
   const ongoingStreamPromises = [];
@@ -2257,7 +2255,7 @@ function createUIMessageStream({
   });
   return handleUIMessageStreamFinish({
     stream,
-    newMessageId: "",
+    messageId: generateId3(),
     originalMessages,
     onFinish
   });
@@ -2401,7 +2399,7 @@ function isDeepEqualData(obj1, obj2) {
 }
 
 // src/util/simulate-readable-stream.ts
-var import_provider_utils6 = require("@ai-sdk/provider-utils");
+var import_provider_utils7 = require("@ai-sdk/provider-utils");
 function simulateReadableStream({
   chunks,
   initialDelayInMs = 0,
@@ -2409,7 +2407,7 @@ function simulateReadableStream({
   _internal
 }) {
   var _a17;
-  const delay2 = (_a17 = _internal == null ? void 0 : _internal.delay) != null ? _a17 : import_provider_utils6.delay;
+  const delay2 = (_a17 = _internal == null ? void 0 : _internal.delay) != null ? _a17 : import_provider_utils7.delay;
   let index = 0;
   return new ReadableStream({
     async pull(controller) {
@@ -2425,7 +2423,7 @@ function simulateReadableStream({
 
 // src/util/retry-with-exponential-backoff.ts
 var import_provider17 = require("@ai-sdk/provider");
-var import_provider_utils7 = require("@ai-sdk/provider-utils");
+var import_provider_utils8 = require("@ai-sdk/provider-utils");
 var retryWithExponentialBackoff = ({
   maxRetries = 2,
   initialDelayInMs = 2e3,
@@ -2443,13 +2441,13 @@ async function _retryWithExponentialBackoff(f, {
   try {
     return await f();
   } catch (error) {
-    if ((0, import_provider_utils7.isAbortError)(error)) {
+    if ((0, import_provider_utils8.isAbortError)(error)) {
       throw error;
     }
     if (maxRetries === 0) {
       throw error;
     }
-    const errorMessage = (0, import_provider_utils7.getErrorMessage)(error);
+    const errorMessage = (0, import_provider_utils8.getErrorMessage)(error);
     const newErrors = [...errors, error];
     const tryNumber = newErrors.length;
     if (tryNumber > maxRetries) {
@@ -2460,7 +2458,7 @@ async function _retryWithExponentialBackoff(f, {
       });
     }
     if (error instanceof Error && import_provider17.APICallError.isInstance(error) && error.isRetryable === true && tryNumber <= maxRetries) {
-      await (0, import_provider_utils7.delay)(delayInMs);
+      await (0, import_provider_utils8.delay)(delayInMs);
       return _retryWithExponentialBackoff(
         f,
         { maxRetries, delayInMs: backoffFactor * delayInMs, backoffFactor },
@@ -3019,7 +3017,7 @@ var DefaultEmbedManyResult = class {
 };
 
 // src/util/detect-media-type.ts
-var import_provider_utils8 = require("@ai-sdk/provider-utils");
+var import_provider_utils9 = require("@ai-sdk/provider-utils");
 var imageMediaTypeSignatures = [
   {
     mediaType: "image/gif",
@@ -3126,7 +3124,7 @@ var audioMediaTypeSignatures = [
   }
 ];
 var stripID3 = (data) => {
-  const bytes = typeof data === "string" ? (0, import_provider_utils8.convertBase64ToUint8Array)(data) : data;
+  const bytes = typeof data === "string" ? (0, import_provider_utils9.convertBase64ToUint8Array)(data) : data;
   const id3Size = (bytes[6] & 127) << 21 | (bytes[7] & 127) << 14 | (bytes[8] & 127) << 7 | bytes[9] & 127;
   return bytes.slice(id3Size + 10);
 };
@@ -3152,7 +3150,7 @@ function detectMediaType({
 }
 
 // core/generate-text/generated-file.ts
-var import_provider_utils9 = require("@ai-sdk/provider-utils");
+var import_provider_utils10 = require("@ai-sdk/provider-utils");
 var DefaultGeneratedFile = class {
   constructor({
     data,
@@ -3166,14 +3164,14 @@ var DefaultGeneratedFile = class {
   // lazy conversion with caching to avoid unnecessary conversion overhead:
   get base64() {
     if (this.base64Data == null) {
-      this.base64Data = (0, import_provider_utils9.convertUint8ArrayToBase64)(this.uint8ArrayData);
+      this.base64Data = (0, import_provider_utils10.convertUint8ArrayToBase64)(this.uint8ArrayData);
     }
     return this.base64Data;
   }
   // lazy conversion with caching to avoid unnecessary conversion overhead:
   get uint8Array() {
     if (this.uint8ArrayData == null) {
-      this.uint8ArrayData = (0, import_provider_utils9.convertBase64ToUint8Array)(this.base64Data);
+      this.uint8ArrayData = (0, import_provider_utils10.convertBase64ToUint8Array)(this.base64Data);
     }
     return this.uint8ArrayData;
   }
@@ -3289,7 +3287,7 @@ async function invokeModelMaxImagesPerCall(model) {
 
 // core/generate-object/generate-object.ts
 var import_provider22 = require("@ai-sdk/provider");
-var import_provider_utils14 = require("@ai-sdk/provider-utils");
+var import_provider_utils15 = require("@ai-sdk/provider-utils");
 
 // core/generate-text/extract-content-text.ts
 function extractContentText(content) {
@@ -3303,7 +3301,7 @@ function extractContentText(content) {
 }
 
 // core/prompt/convert-to-language-model-prompt.ts
-var import_provider_utils11 = require("@ai-sdk/provider-utils");
+var import_provider_utils12 = require("@ai-sdk/provider-utils");
 
 // src/util/download.ts
 async function download({ url }) {
@@ -3332,7 +3330,7 @@ async function download({ url }) {
 
 // core/prompt/data-content.ts
 var import_provider18 = require("@ai-sdk/provider");
-var import_provider_utils10 = require("@ai-sdk/provider-utils");
+var import_provider_utils11 = require("@ai-sdk/provider-utils");
 var import_zod2 = require("zod");
 
 // core/prompt/split-data-url.ts
@@ -3397,9 +3395,9 @@ function convertDataContentToBase64String(content) {
     return content;
   }
   if (content instanceof ArrayBuffer) {
-    return (0, import_provider_utils10.convertUint8ArrayToBase64)(new Uint8Array(content));
+    return (0, import_provider_utils11.convertUint8ArrayToBase64)(new Uint8Array(content));
   }
-  return (0, import_provider_utils10.convertUint8ArrayToBase64)(content);
+  return (0, import_provider_utils11.convertUint8ArrayToBase64)(content);
 }
 function convertDataContentToUint8Array(content) {
   if (content instanceof Uint8Array) {
@@ -3407,7 +3405,7 @@ function convertDataContentToUint8Array(content) {
   }
   if (typeof content === "string") {
     try {
-      return (0, import_provider_utils10.convertBase64ToUint8Array)(content);
+      return (0, import_provider_utils11.convertBase64ToUint8Array)(content);
     } catch (error) {
       throw new InvalidDataContentError({
         message: "Invalid data content. Content string is not a base64-encoded media.",
@@ -3558,7 +3556,7 @@ async function downloadAssets(messages, downloadImplementation, supportedUrls) {
     }
     return { mediaType, data };
   }).filter(
-    (part) => part.data instanceof URL && part.mediaType != null && !(0, import_provider_utils11.isUrlSupported)({
+    (part) => part.data instanceof URL && part.mediaType != null && !(0, import_provider_utils12.isUrlSupported)({
       url: part.data.toString(),
       mediaType: part.mediaType,
       supportedUrls
@@ -3742,7 +3740,7 @@ function resolveLanguageModel(model) {
 
 // core/prompt/standardize-prompt.ts
 var import_provider19 = require("@ai-sdk/provider");
-var import_provider_utils12 = require("@ai-sdk/provider-utils");
+var import_provider_utils13 = require("@ai-sdk/provider-utils");
 var import_zod8 = require("zod");
 
 // core/prompt/message.ts
@@ -3914,7 +3912,7 @@ async function standardizePrompt(prompt) {
       message: "messages must not be empty"
     });
   }
-  const validationResult = await (0, import_provider_utils12.safeValidateTypes)({
+  const validationResult = await (0, import_provider_utils13.safeValidateTypes)({
     value: messages,
     schema: import_zod8.z.array(modelMessageSchema)
   });
@@ -3962,7 +3960,7 @@ function stringifyForTelemetry(prompt) {
 
 // core/generate-object/output-strategy.ts
 var import_provider21 = require("@ai-sdk/provider");
-var import_provider_utils13 = require("@ai-sdk/provider-utils");
+var import_provider_utils14 = require("@ai-sdk/provider-utils");
 
 // src/util/async-iterable-stream.ts
 function createAsyncIterableStream(source) {
@@ -4018,7 +4016,7 @@ var objectOutputStrategy = (schema) => ({
     };
   },
   async validateFinalResult(value) {
-    return (0, import_provider_utils13.safeValidateTypes)({ value, schema });
+    return (0, import_provider_utils14.safeValidateTypes)({ value, schema });
   },
   createElementStream() {
     throw new import_provider21.UnsupportedFunctionalityError({
@@ -4062,7 +4060,7 @@ var arrayOutputStrategy = (schema) => {
       const resultArray = [];
       for (let i = 0; i < inputArray.length; i++) {
         const element = inputArray[i];
-        const result = await (0, import_provider_utils13.safeValidateTypes)({ value: element, schema });
+        const result = await (0, import_provider_utils14.safeValidateTypes)({ value: element, schema });
         if (i === inputArray.length - 1 && !isFinalDelta) {
           continue;
         }
@@ -4103,7 +4101,7 @@ var arrayOutputStrategy = (schema) => {
       }
       const inputArray = value.elements;
       for (const element of inputArray) {
-        const result = await (0, import_provider_utils13.safeValidateTypes)({ value: element, schema });
+        const result = await (0, import_provider_utils14.safeValidateTypes)({ value: element, schema });
         if (!result.success) {
           return result;
         }
@@ -4221,9 +4219,9 @@ function getOutputStrategy({
 }) {
   switch (output) {
     case "object":
-      return objectOutputStrategy((0, import_provider_utils13.asSchema)(schema));
+      return objectOutputStrategy((0, import_provider_utils14.asSchema)(schema));
     case "array":
-      return arrayOutputStrategy((0, import_provider_utils13.asSchema)(schema));
+      return arrayOutputStrategy((0, import_provider_utils14.asSchema)(schema));
     case "enum":
       return enumOutputStrategy(enumValues);
     case "no-schema":
@@ -4354,7 +4352,7 @@ function validateObjectGenerationInput({
 }
 
 // core/generate-object/generate-object.ts
-var originalGenerateId = (0, import_provider_utils14.createIdGenerator)({ prefix: "aiobj", size: 24 });
+var originalGenerateId = (0, import_provider_utils15.createIdGenerator)({ prefix: "aiobj", size: 24 });
 async function generateObject(options) {
   const {
     model: modelArg,
@@ -4532,7 +4530,7 @@ async function generateObject(options) {
         request = (_a17 = generateResult.request) != null ? _a17 : {};
         response = generateResult.responseData;
         async function processResult(result2) {
-          const parseResult = await (0, import_provider_utils14.safeParseJSON)({ text: result2 });
+          const parseResult = await (0, import_provider_utils15.safeParseJSON)({ text: result2 });
           if (!parseResult.success) {
             throw new NoObjectGeneratedError({
               message: "No object generated: could not parse the response.",
@@ -4631,7 +4629,7 @@ var DefaultGenerateObjectResult = class {
 };
 
 // core/generate-object/stream-object.ts
-var import_provider_utils15 = require("@ai-sdk/provider-utils");
+var import_provider_utils16 = require("@ai-sdk/provider-utils");
 
 // src/util/create-resolvable-promise.ts
 function createResolvablePromise() {
@@ -4775,7 +4773,7 @@ function now() {
 }
 
 // core/generate-object/stream-object.ts
-var originalGenerateId2 = (0, import_provider_utils15.createIdGenerator)({ prefix: "aiobj", size: 24 });
+var originalGenerateId2 = (0, import_provider_utils16.createIdGenerator)({ prefix: "aiobj", size: 24 });
 function streamObject(options) {
   const {
     model,
@@ -5374,7 +5372,7 @@ var DefaultSpeechResult = class {
 };
 
 // core/generate-text/generate-text.ts
-var import_provider_utils18 = require("@ai-sdk/provider-utils");
+var import_provider_utils19 = require("@ai-sdk/provider-utils");
 
 // src/util/as-array.ts
 function asArray(value) {
@@ -5382,7 +5380,7 @@ function asArray(value) {
 }
 
 // core/prompt/prepare-tools-and-tool-choice.ts
-var import_provider_utils16 = require("@ai-sdk/provider-utils");
+var import_provider_utils17 = require("@ai-sdk/provider-utils");
 
 // src/util/is-non-empty-object.ts
 function isNonEmptyObject(object2) {
@@ -5414,7 +5412,7 @@ function prepareToolsAndToolChoice({
             type: "function",
             name: name17,
             description: tool2.description,
-            parameters: (0, import_provider_utils16.asSchema)(tool2.parameters).jsonSchema
+            parameters: (0, import_provider_utils17.asSchema)(tool2.parameters).jsonSchema
           };
         case "provider-defined":
           return {
@@ -5484,7 +5482,7 @@ function asContent({
 }
 
 // core/generate-text/parse-tool-call.ts
-var import_provider_utils17 = require("@ai-sdk/provider-utils");
+var import_provider_utils18 = require("@ai-sdk/provider-utils");
 async function parseToolCall({
   toolCall,
   tools,
@@ -5508,7 +5506,7 @@ async function parseToolCall({
         tools,
         parameterSchema: ({ toolName }) => {
           const { parameters } = tools[toolName];
-          return (0, import_provider_utils17.asSchema)(parameters).jsonSchema;
+          return (0, import_provider_utils18.asSchema)(parameters).jsonSchema;
         },
         system,
         messages,
@@ -5538,8 +5536,8 @@ async function doParseToolCall({
       availableTools: Object.keys(tools)
     });
   }
-  const schema = (0, import_provider_utils17.asSchema)(tool2.parameters);
-  const parseResult = toolCall.args.trim() === "" ? await (0, import_provider_utils17.safeValidateTypes)({ value: {}, schema }) : await (0, import_provider_utils17.safeParseJSON)({ text: toolCall.args, schema });
+  const schema = (0, import_provider_utils18.asSchema)(tool2.parameters);
+  const parseResult = toolCall.args.trim() === "" ? await (0, import_provider_utils18.safeValidateTypes)({ value: {}, schema }) : await (0, import_provider_utils18.safeParseJSON)({ text: toolCall.args, schema });
   if (parseResult.success === false) {
     throw new InvalidToolArgumentsError({
       toolName,
@@ -5675,7 +5673,7 @@ function toResponseMessages({
 }
 
 // core/generate-text/generate-text.ts
-var originalGenerateId3 = (0, import_provider_utils18.createIdGenerator)({
+var originalGenerateId3 = (0, import_provider_utils19.createIdGenerator)({
   prefix: "aitxt",
   size: 24
 });
@@ -6134,7 +6132,7 @@ __export(output_exports, {
   object: () => object,
   text: () => text
 });
-var import_provider_utils19 = require("@ai-sdk/provider-utils");
+var import_provider_utils20 = require("@ai-sdk/provider-utils");
 var text = () => ({
   type: "text",
   responseFormat: { type: "text" },
@@ -6148,7 +6146,7 @@ var text = () => ({
 var object = ({
   schema: inputSchema
 }) => {
-  const schema = (0, import_provider_utils19.asSchema)(inputSchema);
+  const schema = (0, import_provider_utils20.asSchema)(inputSchema);
   return {
     type: "object",
     responseFormat: {
@@ -6174,7 +6172,7 @@ var object = ({
       }
     },
     async parseOutput({ text: text2 }, context) {
-      const parseResult = await (0, import_provider_utils19.safeParseJSON)({ text: text2 });
+      const parseResult = await (0, import_provider_utils20.safeParseJSON)({ text: text2 });
       if (!parseResult.success) {
         throw new NoObjectGeneratedError({
           message: "No object generated: could not parse the response.",
@@ -6185,7 +6183,7 @@ var object = ({
           finishReason: context.finishReason
         });
       }
-      const validationResult = await (0, import_provider_utils19.safeValidateTypes)({
+      const validationResult = await (0, import_provider_utils20.safeValidateTypes)({
         value: parseResult.value,
         schema
       });
@@ -6205,7 +6203,7 @@ var object = ({
 };
 
 // core/generate-text/smooth-stream.ts
-var import_provider_utils20 = require("@ai-sdk/provider-utils");
+var import_provider_utils21 = require("@ai-sdk/provider-utils");
 var import_provider24 = require("@ai-sdk/provider");
 var CHUNKING_REGEXPS = {
   word: /\S+\s+/m,
@@ -6214,7 +6212,7 @@ var CHUNKING_REGEXPS = {
 function smoothStream({
   delayInMs = 10,
   chunking = "word",
-  _internal: { delay: delay2 = import_provider_utils20.delay } = {}
+  _internal: { delay: delay2 = import_provider_utils21.delay } = {}
 } = {}) {
   let detectChunk;
   if (typeof chunking === "function") {
@@ -6274,10 +6272,10 @@ function smoothStream({
 }
 
 // core/generate-text/stream-text.ts
-var import_provider_utils22 = require("@ai-sdk/provider-utils");
+var import_provider_utils23 = require("@ai-sdk/provider-utils");
 
 // core/generate-text/run-tools-transformation.ts
-var import_provider_utils21 = require("@ai-sdk/provider-utils");
+var import_provider_utils22 = require("@ai-sdk/provider-utils");
 function runToolsTransformation({
   tools,
   generatorStream,
@@ -6372,7 +6370,7 @@ function runToolsTransformation({
               });
             }
             if (tool2.execute != null) {
-              const toolExecutionId = (0, import_provider_utils21.generateId)();
+              const toolExecutionId = (0, import_provider_utils22.generateId)();
               outstandingToolResults.add(toolExecutionId);
               recordSpan({
                 name: "ai.toolCall",
@@ -6480,8 +6478,20 @@ function runToolsTransformation({
   });
 }
 
+// src/ui-message-stream/get-response-ui-message-id.ts
+function getResponseUIMessageId({
+  originalMessages,
+  responseMessageId
+}) {
+  if (originalMessages == null) {
+    return void 0;
+  }
+  const lastMessage = originalMessages[originalMessages.length - 1];
+  return (lastMessage == null ? void 0 : lastMessage.role) === "assistant" ? lastMessage.id : typeof responseMessageId === "function" ? responseMessageId() : responseMessageId;
+}
+
 // core/generate-text/stream-text.ts
-var originalGenerateId4 = (0, import_provider_utils22.createIdGenerator)({
+var originalGenerateId4 = (0, import_provider_utils23.createIdGenerator)({
   prefix: "aitxt",
   size: 24
 });
@@ -6644,7 +6654,7 @@ var DefaultStreamTextResult = class {
       async transform(chunk, controller) {
         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-call-streaming-start" || part.type === "tool-call-delta") {
+        if (part.type === "text" || part.type === "reasoning" || part.type === "source" || part.type === "tool-call" || part.type === "tool-result" || part.type === "tool-call-streaming-start" || part.type === "tool-call-delta" || part.type === "raw") {
           await (onChunk == null ? void 0 : onChunk({ chunk: part }));
         }
         if (part.type === "error") {
@@ -7093,7 +7103,9 @@ var DefaultStreamTextResult = class {
                       break;
                     }
                     case "raw": {
-                      controller.enqueue(chunk);
+                      if (includeRawChunks2) {
+                        controller.enqueue(chunk);
+                      }
                       break;
                     }
                     default: {
@@ -7316,23 +7328,24 @@ var DefaultStreamTextResult = class {
     );
   }
   toUIMessageStream({
-    newMessageId,
-    originalMessages = [],
+    originalMessages,
     onFinish,
     messageMetadata,
-    sendReasoning = false,
+    sendReasoning = true,
     sendSources = false,
     sendStart = true,
     sendFinish = true,
     onError = () => "An error occurred."
     // mask error messages for safety by default
   } = {}) {
-    const lastMessage = originalMessages[originalMessages.length - 1];
-    const isContinuation = (lastMessage == null ? void 0 : lastMessage.role) === "assistant";
-    const messageId = isContinuation ? lastMessage.id : newMessageId;
+    const responseMessageId = getResponseUIMessageId({
+      originalMessages,
+      responseMessageId: this.generateId
+    });
     const baseStream = this.fullStream.pipeThrough(
       new TransformStream({
         transform: async (part, controller) => {
+          const messageMetadataValue = messageMetadata == null ? void 0 : messageMetadata({ part });
           const partType = part.type;
           switch (partType) {
             case "text": {
@@ -7429,38 +7442,28 @@ var DefaultStreamTextResult = class {
               break;
             }
             case "start-step": {
-              const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
-              controller.enqueue({
-                type: "start-step",
-                metadata
-              });
+              controller.enqueue({ type: "start-step" });
               break;
             }
             case "finish-step": {
-              const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
-              controller.enqueue({
-                type: "finish-step",
-                metadata
-              });
+              controller.enqueue({ type: "finish-step" });
               break;
             }
             case "start": {
               if (sendStart) {
-                const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "start",
-                  messageId,
-                  metadata
+                  messageId: responseMessageId,
+                  messageMetadata: messageMetadataValue
                 });
               }
               break;
             }
             case "finish": {
               if (sendFinish) {
-                const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "finish",
-                  metadata
+                  messageMetadata: messageMetadataValue
                 });
               }
               break;
@@ -7473,18 +7476,23 @@ var DefaultStreamTextResult = class {
               throw new Error(`Unknown chunk type: ${exhaustiveCheck}`);
             }
           }
+          if (messageMetadataValue != null && partType !== "start" && partType !== "finish") {
+            controller.enqueue({
+              type: "message-metadata",
+              messageMetadata: messageMetadataValue
+            });
+          }
         }
       })
     );
     return handleUIMessageStreamFinish({
       stream: baseStream,
-      newMessageId: messageId != null ? messageId : this.generateId(),
+      messageId: responseMessageId != null ? responseMessageId : this.generateId(),
       originalMessages,
       onFinish
     });
   }
   pipeUIMessageStreamToResponse(response, {
-    newMessageId,
     originalMessages,
     onFinish,
     messageMetadata,
@@ -7498,7 +7506,6 @@ var DefaultStreamTextResult = class {
     pipeUIMessageStreamToResponse({
       response,
       stream: this.toUIMessageStream({
-        newMessageId,
         originalMessages,
         onFinish,
         messageMetadata,
@@ -7519,7 +7526,6 @@ var DefaultStreamTextResult = class {
     });
   }
   toUIMessageStreamResponse({
-    newMessageId,
     originalMessages,
     onFinish,
     messageMetadata,
@@ -7532,7 +7538,6 @@ var DefaultStreamTextResult = class {
   } = {}) {
     return createUIMessageStreamResponse({
       stream: this.toUIMessageStream({
-        newMessageId,
         originalMessages,
         onFinish,
         messageMetadata,
@@ -7929,7 +7934,7 @@ var DefaultProviderRegistry = class {
 };
 
 // core/tool/mcp/mcp-client.ts
-var import_provider_utils24 = require("@ai-sdk/provider-utils");
+var import_provider_utils25 = require("@ai-sdk/provider-utils");
 
 // core/tool/tool.ts
 function tool(tool2) {
@@ -7937,7 +7942,7 @@ function tool(tool2) {
 }
 
 // core/tool/mcp/mcp-sse-transport.ts
-var import_provider_utils23 = require("@ai-sdk/provider-utils");
+var import_provider_utils24 = require("@ai-sdk/provider-utils");
 
 // core/tool/mcp/json-rpc-message.ts
 var import_zod10 = require("zod");
@@ -8108,7 +8113,7 @@ var SseMCPTransport = class {
             (_b = this.onerror) == null ? void 0 : _b.call(this, error);
             return reject(error);
           }
-          const stream = response.body.pipeThrough(new TextDecoderStream()).pipeThrough((0, import_provider_utils23.createEventSourceParserStream)());
+          const stream = response.body.pipeThrough(new TextDecoderStream()).pipeThrough((0, import_provider_utils24.createEventSourceParserStream)());
           const reader = stream.getReader();
           const processEvents = async () => {
             var _a18, _b2, _c2;
@@ -8432,7 +8437,7 @@ var MCPClient = class {
         if (schemas !== "automatic" && !(name17 in schemas)) {
           continue;
         }
-        const parameters = schemas === "automatic" ? (0, import_provider_utils24.jsonSchema)({
+        const parameters = schemas === "automatic" ? (0, import_provider_utils25.jsonSchema)({
           ...inputSchema,
           properties: (_a17 = inputSchema.properties) != null ? _a17 : {},
           additionalProperties: false
@@ -8626,9 +8631,10 @@ var DefaultTranscriptionResult = class {
   generateObject,
   generateText,
   getTextFromDataUrl,
-  getToolInvocations,
+  getToolName,
   hasToolCall,
   isDeepEqualData,
+  isToolUIPart,
   jsonSchema,
   modelMessageSchema,
   parsePartialJson,