ai 5.0.0-canary.22 → 5.0.0-canary.24

package/dist/index.js

@@ -22,6 +22,8 @@ var src_exports = {};
 __export(src_exports, {
   AISDKError: () => import_provider16.AISDKError,
   APICallError: () => import_provider16.APICallError,
+  ChatStore: () => ChatStore,
+  DefaultChatTransport: () => DefaultChatTransport,
   DownloadError: () => DownloadError,
   EmptyResponseBodyError: () => import_provider16.EmptyResponseBodyError,
   InvalidArgumentError: () => InvalidArgumentError,
@@ -50,7 +52,7 @@ __export(src_exports, {
   TypeValidationError: () => import_provider16.TypeValidationError,
   UnsupportedFunctionalityError: () => import_provider16.UnsupportedFunctionalityError,
   appendClientMessage: () => appendClientMessage,
-  asSchema: () => import_provider_utils25.asSchema,
+  asSchema: () => import_provider_utils26.asSchema,
   assistantModelMessageSchema: () => assistantModelMessageSchema,
   callChatApi: () => callChatApi,
   callCompletionApi: () => callCompletionApi,
@@ -63,12 +65,13 @@ __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,
   createUIMessageStreamResponse: () => createUIMessageStreamResponse,
   customProvider: () => customProvider,
+  defaultChatStore: () => defaultChatStore,
   defaultSettingsMiddleware: () => defaultSettingsMiddleware,
   embed: () => embed,
   embedMany: () => embedMany,
@@ -80,19 +83,20 @@ __export(src_exports, {
   experimental_transcribe: () => transcribe,
   extractMaxToolInvocationStep: () => extractMaxToolInvocationStep,
   extractReasoningMiddleware: () => extractReasoningMiddleware,
-  generateId: () => import_provider_utils25.generateId,
+  generateId: () => import_provider_utils26.generateId,
   generateObject: () => generateObject,
   generateText: () => generateText,
   getTextFromDataUrl: () => getTextFromDataUrl,
   getToolInvocations: () => getToolInvocations,
+  hasToolCall: () => hasToolCall,
   isAssistantMessageWithCompletedToolCalls: () => isAssistantMessageWithCompletedToolCalls,
   isDeepEqualData: () => isDeepEqualData,
-  jsonSchema: () => import_provider_utils25.jsonSchema,
+  jsonSchema: () => import_provider_utils26.jsonSchema,
+  maxSteps: () => maxSteps,
   modelMessageSchema: () => modelMessageSchema,
   parsePartialJson: () => parsePartialJson,
   pipeTextStreamToResponse: () => pipeTextStreamToResponse,
   pipeUIMessageStreamToResponse: () => pipeUIMessageStreamToResponse,
-  processTextStream: () => processTextStream,
   shouldResubmitMessages: () => shouldResubmitMessages,
   simulateReadableStream: () => simulateReadableStream,
   simulateStreamingMiddleware: () => simulateStreamingMiddleware,
@@ -107,7 +111,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");
@@ -546,106 +550,88 @@ function appendClientMessage({
 }
 
 // src/ui/call-chat-api.ts
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
 
 // src/ui-message-stream/ui-message-stream-parts.ts
 var import_zod = require("zod");
-var toolCallSchema = import_zod.z.object({
-  toolCallId: import_zod.z.string(),
-  toolName: import_zod.z.string(),
-  args: import_zod.z.unknown()
-});
-var toolResultValueSchema = import_zod.z.object({
-  toolCallId: import_zod.z.string(),
-  result: import_zod.z.unknown(),
-  providerMetadata: import_zod.z.any().optional()
-});
-var sourceSchema = import_zod.z.object({
-  type: import_zod.z.literal("source"),
-  sourceType: import_zod.z.literal("url"),
-  id: import_zod.z.string(),
-  url: import_zod.z.string(),
-  title: import_zod.z.string().optional(),
-  providerMetadata: import_zod.z.any().optional()
-  // Use z.any() for generic metadata
-});
-var uiMessageStreamPartSchema = import_zod.z.discriminatedUnion("type", [
+var uiMessageStreamPartSchema = import_zod.z.union([
   import_zod.z.object({
     type: import_zod.z.literal("text"),
-    value: import_zod.z.string()
+    text: import_zod.z.string()
   }),
   import_zod.z.object({
     type: import_zod.z.literal("error"),
-    value: import_zod.z.string()
+    errorText: import_zod.z.string()
   }),
   import_zod.z.object({
-    type: import_zod.z.literal("tool-call"),
-    value: toolCallSchema
+    type: import_zod.z.literal("tool-call-streaming-start"),
+    toolCallId: import_zod.z.string(),
+    toolName: import_zod.z.string()
   }),
   import_zod.z.object({
-    type: import_zod.z.literal("tool-result"),
-    value: toolResultValueSchema
+    type: import_zod.z.literal("tool-call-delta"),
+    toolCallId: import_zod.z.string(),
+    argsTextDelta: import_zod.z.string()
   }),
   import_zod.z.object({
-    type: import_zod.z.literal("tool-call-streaming-start"),
-    value: import_zod.z.object({ toolCallId: import_zod.z.string(), toolName: import_zod.z.string() })
+    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({
-    type: import_zod.z.literal("tool-call-delta"),
-    value: import_zod.z.object({ toolCallId: import_zod.z.string(), argsTextDelta: import_zod.z.string() })
+    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({
     type: import_zod.z.literal("reasoning"),
-    value: import_zod.z.object({
-      text: import_zod.z.string(),
-      providerMetadata: import_zod.z.record(import_zod.z.any()).optional()
-    })
+    text: import_zod.z.string(),
+    providerMetadata: import_zod.z.record(import_zod.z.any()).optional()
   }),
   import_zod.z.object({
     type: import_zod.z.literal("source"),
-    value: sourceSchema
+    sourceType: import_zod.z.literal("url"),
+    id: import_zod.z.string(),
+    url: import_zod.z.string(),
+    title: import_zod.z.string().optional(),
+    providerMetadata: import_zod.z.any().optional()
+    // Use z.any() for generic metadata
   }),
   import_zod.z.object({
     type: import_zod.z.literal("file"),
-    value: import_zod.z.object({
-      url: import_zod.z.string(),
-      mediaType: import_zod.z.string()
-    })
+    url: import_zod.z.string(),
+    mediaType: import_zod.z.string()
+  }),
+  import_zod.z.object({
+    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()
-    })
+    value: import_zod.z.object({ metadata: import_zod.z.unknown() })
   }),
   import_zod.z.object({
     type: import_zod.z.literal("start-step"),
-    value: import_zod.z.object({
-      metadata: import_zod.z.unknown()
-    })
+    metadata: import_zod.z.unknown().optional()
   }),
   import_zod.z.object({
     type: import_zod.z.literal("finish-step"),
-    value: import_zod.z.object({
-      metadata: import_zod.z.unknown()
-    })
+    metadata: import_zod.z.unknown().optional()
   }),
   import_zod.z.object({
     type: import_zod.z.literal("start"),
-    value: import_zod.z.object({
-      messageId: import_zod.z.string().optional(),
-      metadata: import_zod.z.unknown()
-    })
+    messageId: import_zod.z.string().optional(),
+    metadata: import_zod.z.unknown().optional()
   }),
   import_zod.z.object({
     type: import_zod.z.literal("finish"),
-    value: import_zod.z.object({
-      metadata: import_zod.z.unknown()
-    })
+    metadata: import_zod.z.unknown().optional()
   }),
   import_zod.z.object({
-    type: import_zod.z.literal("reasoning-part-finish"),
-    value: import_zod.z.null()
+    type: import_zod.z.literal("reasoning-part-finish")
   })
 ]);
 
@@ -1057,303 +1043,305 @@ function getToolInvocations(message) {
 }
 
 // src/ui/process-ui-message-stream.ts
-function processUIMessageStream({
-  stream,
-  onUpdate,
-  onToolCall,
-  onFinish,
+function createStreamingUIMessageState({
   lastMessage,
-  newMessageId,
-  messageMetadataSchema
-}) {
+  newMessageId = "no-id"
+} = {}) {
   var _a17;
   const isContinuation = (lastMessage == null ? void 0 : lastMessage.role) === "assistant";
-  let step = isContinuation ? 1 + ((_a17 = extractMaxToolInvocationStep(getToolInvocations(lastMessage))) != null ? _a17 : 0) : 0;
+  const step = isContinuation ? 1 + ((_a17 = extractMaxToolInvocationStep(getToolInvocations(lastMessage))) != null ? _a17 : 0) : 0;
   const message = isContinuation ? structuredClone(lastMessage) : {
     id: newMessageId,
     metadata: {},
     role: "assistant",
     parts: []
   };
-  let currentTextPart = void 0;
-  let currentReasoningPart = void 0;
-  function updateToolInvocationPart(toolCallId, invocation) {
-    const part = message.parts.find(
-      (part2) => part2.type === "tool-invocation" && part2.toolInvocation.toolCallId === toolCallId
-    );
-    if (part != null) {
-      part.toolInvocation = invocation;
-    } else {
-      message.parts.push({
-        type: "tool-invocation",
-        toolInvocation: invocation
-      });
-    }
-  }
-  const partialToolCalls = {};
-  async function updateMessageMetadata(metadata) {
-    if (metadata != null) {
-      const mergedMetadata = message.metadata != null ? mergeObjects(message.metadata, metadata) : metadata;
-      if (messageMetadataSchema != null) {
-        await (0, import_provider_utils2.validateTypes)({
-          value: mergedMetadata,
-          schema: messageMetadataSchema
-        });
-      }
-      message.metadata = mergedMetadata;
-    }
-  }
+  return {
+    message,
+    activeTextPart: void 0,
+    activeReasoningPart: void 0,
+    partialToolCalls: {},
+    step
+  };
+}
+function processUIMessageStream({
+  stream,
+  onToolCall,
+  messageMetadataSchema,
+  runUpdateMessageJob
+}) {
   return stream.pipeThrough(
     new TransformStream({
-      async transform(chunk, controller) {
-        const { type, value } = chunk;
-        switch (type) {
-          case "text": {
-            if (currentTextPart == null) {
-              currentTextPart = {
-                type: "text",
-                text: value
-              };
-              message.parts.push(currentTextPart);
+      async transform(part, controller) {
+        await runUpdateMessageJob(async ({ state, write }) => {
+          function updateToolInvocationPart(toolCallId, invocation) {
+            const part2 = state.message.parts.find(
+              (part3) => part3.type === "tool-invocation" && part3.toolInvocation.toolCallId === toolCallId
+            );
+            if (part2 != null) {
+              part2.toolInvocation = invocation;
             } else {
-              currentTextPart.text += value;
+              state.message.parts.push({
+                type: "tool-invocation",
+                toolInvocation: invocation
+              });
             }
-            onUpdate == null ? void 0 : onUpdate({ message });
-            break;
           }
-          case "reasoning": {
-            if (currentReasoningPart == null) {
-              currentReasoningPart = {
-                type: "reasoning",
-                text: value.text,
-                providerMetadata: value.providerMetadata
-              };
-              message.parts.push(currentReasoningPart);
-            } else {
-              currentReasoningPart.text += value.text;
-              currentReasoningPart.providerMetadata = value.providerMetadata;
+          async function updateMessageMetadata(metadata) {
+            if (metadata != null) {
+              const mergedMetadata = state.message.metadata != null ? mergeObjects(state.message.metadata, metadata) : metadata;
+              if (messageMetadataSchema != null) {
+                await (0, import_provider_utils2.validateTypes)({
+                  value: mergedMetadata,
+                  schema: messageMetadataSchema
+                });
+              }
+              state.message.metadata = mergedMetadata;
             }
-            onUpdate == null ? void 0 : onUpdate({ message });
-            break;
           }
-          case "reasoning-part-finish": {
-            if (currentReasoningPart != null) {
-              currentReasoningPart = void 0;
+          switch (part.type) {
+            case "text": {
+              if (state.activeTextPart == null) {
+                state.activeTextPart = {
+                  type: "text",
+                  text: part.text
+                };
+                state.message.parts.push(state.activeTextPart);
+              } else {
+                state.activeTextPart.text += part.text;
+              }
+              write();
+              break;
             }
-            break;
-          }
-          case "file": {
-            message.parts.push({
-              type: "file",
-              mediaType: value.mediaType,
-              url: value.url
-            });
-            onUpdate == null ? void 0 : onUpdate({ message });
-            break;
-          }
-          case "source": {
-            message.parts.push({
-              type: "source",
-              source: value
-            });
-            onUpdate == null ? void 0 : onUpdate({ message });
-            break;
-          }
-          case "tool-call-streaming-start": {
-            const toolInvocations = getToolInvocations(message);
-            partialToolCalls[value.toolCallId] = {
-              text: "",
-              step,
-              toolName: value.toolName,
-              index: toolInvocations.length
-            };
-            updateToolInvocationPart(value.toolCallId, {
-              state: "partial-call",
-              step,
-              toolCallId: value.toolCallId,
-              toolName: value.toolName,
-              args: void 0
-            });
-            onUpdate == null ? void 0 : onUpdate({ message });
-            break;
-          }
-          case "tool-call-delta": {
-            const partialToolCall = partialToolCalls[value.toolCallId];
-            partialToolCall.text += value.argsTextDelta;
-            const { value: partialArgs } = await parsePartialJson(
-              partialToolCall.text
-            );
-            updateToolInvocationPart(value.toolCallId, {
-              state: "partial-call",
-              step: partialToolCall.step,
-              toolCallId: value.toolCallId,
-              toolName: partialToolCall.toolName,
-              args: partialArgs
-            });
-            onUpdate == null ? void 0 : onUpdate({ message });
-            break;
-          }
-          case "tool-call": {
-            const call = { args: value.args, ...value };
-            updateToolInvocationPart(value.toolCallId, {
-              state: "call",
-              step,
-              ...call
-            });
-            onUpdate == null ? void 0 : onUpdate({ message });
-            if (onToolCall) {
-              const result = await onToolCall({
-                toolCall: call
+            case "reasoning": {
+              if (state.activeReasoningPart == null) {
+                state.activeReasoningPart = {
+                  type: "reasoning",
+                  text: part.text,
+                  providerMetadata: part.providerMetadata
+                };
+                state.message.parts.push(state.activeReasoningPart);
+              } else {
+                state.activeReasoningPart.text += part.text;
+                state.activeReasoningPart.providerMetadata = part.providerMetadata;
+              }
+              write();
+              break;
+            }
+            case "reasoning-part-finish": {
+              if (state.activeReasoningPart != null) {
+                state.activeReasoningPart = void 0;
+              }
+              break;
+            }
+            case "file": {
+              state.message.parts.push({
+                type: "file",
+                mediaType: part.mediaType,
+                url: part.url
+              });
+              write();
+              break;
+            }
+            case "source": {
+              state.message.parts.push({
+                type: "source",
+                source: {
+                  sourceType: "url",
+                  id: part.id,
+                  url: part.url,
+                  title: part.title,
+                  providerMetadata: part.providerMetadata
+                }
+              });
+              write();
+              break;
+            }
+            case "tool-call-streaming-start": {
+              const toolInvocations = getToolInvocations(state.message);
+              state.partialToolCalls[part.toolCallId] = {
+                text: "",
+                step: state.step,
+                toolName: part.toolName,
+                index: toolInvocations.length
+              };
+              updateToolInvocationPart(part.toolCallId, {
+                state: "partial-call",
+                step: state.step,
+                toolCallId: part.toolCallId,
+                toolName: part.toolName,
+                args: void 0
+              });
+              write();
+              break;
+            }
+            case "tool-call-delta": {
+              const partialToolCall = state.partialToolCalls[part.toolCallId];
+              partialToolCall.text += part.argsTextDelta;
+              const { value: partialArgs } = await parsePartialJson(
+                partialToolCall.text
+              );
+              updateToolInvocationPart(part.toolCallId, {
+                state: "partial-call",
+                step: partialToolCall.step,
+                toolCallId: part.toolCallId,
+                toolName: partialToolCall.toolName,
+                args: partialArgs
+              });
+              write();
+              break;
+            }
+            case "tool-call": {
+              updateToolInvocationPart(part.toolCallId, {
+                state: "call",
+                step: state.step,
+                toolCallId: part.toolCallId,
+                toolName: part.toolName,
+                args: part.args
               });
-              if (result != null) {
-                updateToolInvocationPart(value.toolCallId, {
-                  state: "result",
-                  step,
-                  ...call,
-                  result
+              write();
+              if (onToolCall) {
+                const result = await onToolCall({
+                  toolCall: part
                 });
-                onUpdate == null ? void 0 : onUpdate({ message });
+                if (result != null) {
+                  updateToolInvocationPart(part.toolCallId, {
+                    state: "result",
+                    step: state.step,
+                    toolCallId: part.toolCallId,
+                    toolName: part.toolName,
+                    args: part.args,
+                    result
+                  });
+                  write();
+                }
               }
+              break;
             }
-            break;
-          }
-          case "tool-result": {
-            const toolInvocations = getToolInvocations(message);
-            if (toolInvocations == null) {
-              throw new Error("tool_result must be preceded by a tool_call");
-            }
-            const toolInvocationIndex = toolInvocations.findIndex(
-              (invocation) => invocation.toolCallId === value.toolCallId
-            );
-            if (toolInvocationIndex === -1) {
-              throw new Error(
-                "tool_result must be preceded by a tool_call with the same toolCallId"
+            case "tool-result": {
+              const toolInvocations = getToolInvocations(state.message);
+              if (toolInvocations == null) {
+                throw new Error("tool_result must be preceded by a tool_call");
+              }
+              const toolInvocationIndex = toolInvocations.findIndex(
+                (invocation) => invocation.toolCallId === part.toolCallId
               );
+              if (toolInvocationIndex === -1) {
+                throw new Error(
+                  "tool_result must be preceded by a tool_call with the same toolCallId"
+                );
+              }
+              updateToolInvocationPart(part.toolCallId, {
+                ...toolInvocations[toolInvocationIndex],
+                state: "result",
+                result: part.result
+              });
+              write();
+              break;
             }
-            const result = { result: value.result, ...value };
-            updateToolInvocationPart(value.toolCallId, {
-              ...toolInvocations[toolInvocationIndex],
-              state: "result",
-              ...result
-            });
-            onUpdate == null ? void 0 : onUpdate({ message });
-            break;
-          }
-          case "start-step": {
-            message.parts.push({ type: "step-start" });
-            await updateMessageMetadata(value.metadata);
-            onUpdate == null ? void 0 : onUpdate({ message });
-            break;
-          }
-          case "finish-step": {
-            step += 1;
-            currentTextPart = void 0;
-            currentReasoningPart = void 0;
-            await updateMessageMetadata(value.metadata);
-            if (value.metadata != null) {
-              onUpdate == null ? void 0 : onUpdate({ message });
+            case "start-step": {
+              state.message.parts.push({ type: "step-start" });
+              await updateMessageMetadata(part.metadata);
+              write();
+              break;
             }
-            break;
-          }
-          case "start": {
-            if (value.messageId != null) {
-              message.id = value.messageId;
+            case "finish-step": {
+              state.step += 1;
+              state.activeTextPart = void 0;
+              state.activeReasoningPart = void 0;
+              await updateMessageMetadata(part.metadata);
+              if (part.metadata != null) {
+                write();
+              }
+              break;
             }
-            await updateMessageMetadata(value.metadata);
-            if (value.messageId != null || value.metadata != null) {
-              onUpdate == null ? void 0 : onUpdate({ message });
+            case "start": {
+              if (part.messageId != null) {
+                state.message.id = part.messageId;
+              }
+              await updateMessageMetadata(part.metadata);
+              if (part.messageId != null || part.metadata != null) {
+                write();
+              }
+              break;
             }
-            break;
-          }
-          case "finish": {
-            await updateMessageMetadata(value.metadata);
-            if (value.metadata != null) {
-              onUpdate == null ? void 0 : onUpdate({ message });
+            case "finish": {
+              await updateMessageMetadata(part.metadata);
+              if (part.metadata != null) {
+                write();
+              }
+              break;
             }
-            break;
-          }
-          case "metadata": {
-            await updateMessageMetadata(value.metadata);
-            if (value.metadata != null) {
-              onUpdate == null ? void 0 : onUpdate({ message });
+            case "metadata": {
+              await updateMessageMetadata(part.metadata);
+              if (part.metadata != null) {
+                write();
+              }
+              break;
+            }
+            case "error": {
+              throw new Error(part.errorText);
+            }
+            default: {
+              if (part.type.startsWith("data-")) {
+                const existingPart = part.id != null ? state.message.parts.find(
+                  (partArg) => part.type === partArg.type && part.id === partArg.id
+                ) : void 0;
+                if (existingPart != null) {
+                  existingPart.value = mergeObjects(
+                    existingPart.data,
+                    part.data
+                  );
+                } else {
+                  state.message.parts.push({
+                    type: part.type,
+                    id: part.id,
+                    value: part.data
+                  });
+                }
+                write();
+              }
             }
-            break;
-          }
-          case "error": {
-            throw new Error(value);
-          }
-          default: {
-            const _exhaustiveCheck = type;
-            throw new Error(`Unhandled stream part: ${_exhaustiveCheck}`);
           }
-        }
-        controller.enqueue(chunk);
-      },
-      flush() {
-        onFinish == null ? void 0 : onFinish({ message });
+          controller.enqueue(part);
+        });
       }
     })
   );
 }
 
-// src/ui/process-chat-text-response.ts
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
-
-// src/ui/process-text-stream.ts
-async function processTextStream({
-  stream,
-  onTextPart
-}) {
-  const reader = stream.pipeThrough(new TextDecoderStream()).getReader();
-  while (true) {
-    const { done, value } = await reader.read();
-    if (done) {
-      break;
-    }
-    await onTextPart(value);
-  }
-}
-
-// src/ui/process-chat-text-response.ts
-async function processChatTextResponse({
-  stream,
-  update,
-  onFinish,
-  generateId: generateId3 = import_provider_utils3.generateId
+// src/ui/transform-text-to-ui-message-stream.ts
+function transformTextToUiMessageStream({
+  stream
 }) {
-  const textPart = { type: "text", text: "" };
-  const resultMessage = {
-    id: generateId3(),
-    role: "assistant",
-    parts: [textPart]
-  };
-  await processTextStream({
-    stream,
-    onTextPart: (chunk) => {
-      textPart.text += chunk;
-      update({ message: { ...resultMessage } });
-    }
-  });
-  onFinish == null ? void 0 : onFinish({ message: resultMessage });
+  return stream.pipeThrough(
+    new TransformStream({
+      start(controller) {
+        controller.enqueue({ type: "start" });
+        controller.enqueue({ type: "start-step" });
+      },
+      async transform(part, controller) {
+        controller.enqueue({ type: "text", text: part });
+      },
+      async flush(controller) {
+        controller.enqueue({ type: "finish-step" });
+        controller.enqueue({ type: "finish" });
+      }
+    })
+  );
 }
 
 // src/ui/call-chat-api.ts
 var getOriginalFetch = () => fetch;
-async function callChatApi({
+async function fetchUIMessageStream({
   api,
   body,
   streamProtocol = "ui-message",
   credentials,
   headers,
   abortController,
-  onUpdate,
-  onFinish,
-  onToolCall,
-  generateId: generateId3,
   fetch: fetch2 = getOriginalFetch(),
-  lastMessage,
-  requestType = "generate",
-  messageMetadataSchema
+  requestType = "generate"
 }) {
   var _a17, _b, _c;
   const response = requestType === "resume" ? await fetch2(`${api}?chatId=${body.id}`, {
@@ -1382,67 +1370,112 @@ async function callChatApi({
   if (!response.body) {
     throw new Error("The response body is empty.");
   }
-  switch (streamProtocol) {
-    case "text": {
-      await processChatTextResponse({
-        stream: response.body,
-        update: onUpdate,
-        onFinish,
-        generateId: generateId3
-      });
-      return;
-    }
-    case "ui-message": {
-      await consumeStream({
-        stream: processUIMessageStream({
-          stream: (0, import_provider_utils4.parseJsonEventStream)({
-            stream: response.body,
-            schema: uiMessageStreamPartSchema
-          }).pipeThrough(
-            new TransformStream({
-              async transform(part, controller) {
-                if (!part.success) {
-                  throw part.error;
-                }
-                controller.enqueue(part.value);
-              }
-            })
-          ),
-          onUpdate({ message }) {
-            const copiedMessage = {
-              // deep copy the message to ensure that deep changes (msg attachments) are updated
-              // with SolidJS. SolidJS uses referential integration of sub-objects to detect changes.
-              ...structuredClone(message),
-              // add a revision id to ensure that the message is updated with SWR. SWR uses a
-              // hashing approach by default to detect changes, but it only works for shallow
-              // changes. This is why we need to add a revision id to ensure that the message
-              // is updated with SWR (without it, the changes get stuck in SWR and are not
-              // forwarded to rendering):
-              revisionId: generateId3()
-            };
-            onUpdate({ message: copiedMessage });
-          },
-          lastMessage,
-          onToolCall,
-          onFinish,
-          newMessageId: generateId3(),
-          messageMetadataSchema
-        }),
-        onError: (error) => {
-          throw error;
+  return streamProtocol === "text" ? transformTextToUiMessageStream({
+    stream: response.body.pipeThrough(new TextDecoderStream())
+  }) : (0, import_provider_utils3.parseJsonEventStream)({
+    stream: response.body,
+    schema: uiMessageStreamPartSchema
+  }).pipeThrough(
+    new TransformStream({
+      async transform(part, controller) {
+        if (!part.success) {
+          throw part.error;
         }
-      });
-      return;
+        controller.enqueue(part.value);
+      }
+    })
+  );
+}
+async function consumeUIMessageStream({
+  stream,
+  onUpdate,
+  onFinish,
+  onToolCall,
+  generateId: generateId3,
+  lastMessage,
+  messageMetadataSchema
+}) {
+  const state = createStreamingUIMessageState({
+    lastMessage,
+    newMessageId: generateId3()
+  });
+  const runUpdateMessageJob = async (job) => {
+    await job({
+      state,
+      write: () => {
+        onUpdate({ message: state.message });
+      }
+    });
+  };
+  await consumeStream({
+    stream: processUIMessageStream({
+      stream,
+      onToolCall,
+      messageMetadataSchema,
+      runUpdateMessageJob
+    }),
+    onError: (error) => {
+      throw error;
     }
-    default: {
-      const exhaustiveCheck = streamProtocol;
-      throw new Error(`Unknown stream protocol: ${exhaustiveCheck}`);
+  });
+  onFinish == null ? void 0 : onFinish({ message: state.message });
+}
+async function callChatApi({
+  api,
+  body,
+  streamProtocol = "ui-message",
+  credentials,
+  headers,
+  abortController,
+  onUpdate,
+  onFinish,
+  onToolCall,
+  generateId: generateId3,
+  fetch: fetch2 = getOriginalFetch(),
+  lastMessage,
+  requestType = "generate",
+  messageMetadataSchema
+}) {
+  const stream = await fetchUIMessageStream({
+    api,
+    body,
+    streamProtocol,
+    credentials,
+    headers,
+    abortController,
+    fetch: fetch2,
+    requestType
+  });
+  await consumeUIMessageStream({
+    stream,
+    onUpdate,
+    onFinish,
+    onToolCall,
+    generateId: generateId3,
+    lastMessage,
+    messageMetadataSchema
+  });
+}
+
+// src/ui/call-completion-api.ts
+var import_provider_utils4 = require("@ai-sdk/provider-utils");
+
+// src/ui/process-text-stream.ts
+async function processTextStream({
+  stream,
+  onTextPart
+}) {
+  const reader = stream.pipeThrough(new TextDecoderStream()).getReader();
+  while (true) {
+    const { done, value } = await reader.read();
+    if (done) {
+      break;
     }
+    await onTextPart(value);
   }
 }
 
 // src/ui/call-completion-api.ts
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
 var getOriginalFetch2 = () => fetch;
 async function callCompletionApi({
   api,
@@ -1501,58 +1534,508 @@ async function callCompletionApi({
         });
         break;
       }
-      case "data": {
-        await consumeStream({
-          stream: (0, import_provider_utils5.parseJsonEventStream)({
-            stream: response.body,
-            schema: uiMessageStreamPartSchema
-          }).pipeThrough(
-            new TransformStream({
-              async transform(part) {
-                if (!part.success) {
-                  throw part.error;
-                }
-                const { type, value } = part.value;
-                if (type === "text") {
-                  result += value;
-                  setCompletion(result);
-                } else if (type === "error") {
-                  throw new Error(value);
-                }
-              }
-            })
-          ),
-          onError: (error) => {
-            throw error;
-          }
+      case "data": {
+        await consumeStream({
+          stream: (0, import_provider_utils4.parseJsonEventStream)({
+            stream: response.body,
+            schema: uiMessageStreamPartSchema
+          }).pipeThrough(
+            new TransformStream({
+              async transform(part) {
+                if (!part.success) {
+                  throw part.error;
+                }
+                const streamPart = part.value;
+                if (streamPart.type === "text") {
+                  result += streamPart.text;
+                  setCompletion(result);
+                } else if (streamPart.type === "error") {
+                  throw new Error(streamPart.errorText);
+                }
+              }
+            })
+          ),
+          onError: (error) => {
+            throw error;
+          }
+        });
+        break;
+      }
+      default: {
+        const exhaustiveCheck = streamProtocol;
+        throw new Error(`Unknown stream protocol: ${exhaustiveCheck}`);
+      }
+    }
+    if (onFinish) {
+      onFinish(prompt, result);
+    }
+    setAbortController(null);
+    return result;
+  } catch (err) {
+    if (err.name === "AbortError") {
+      setAbortController(null);
+      return null;
+    }
+    if (err instanceof Error) {
+      if (onError) {
+        onError(err);
+      }
+    }
+    setError(err);
+  } finally {
+    setLoading(false);
+  }
+}
+
+// src/ui/chat-store.ts
+var import_provider_utils5 = require("@ai-sdk/provider-utils");
+
+// src/util/serial-job-executor.ts
+var SerialJobExecutor = class {
+  constructor() {
+    this.queue = [];
+    this.isProcessing = false;
+  }
+  async processQueue() {
+    if (this.isProcessing) {
+      return;
+    }
+    this.isProcessing = true;
+    while (this.queue.length > 0) {
+      await this.queue[0]();
+      this.queue.shift();
+    }
+    this.isProcessing = false;
+  }
+  async run(job) {
+    return new Promise((resolve, reject) => {
+      this.queue.push(async () => {
+        try {
+          await job();
+          resolve();
+        } catch (error) {
+          reject(error);
+        }
+      });
+      void this.processQueue();
+    });
+  }
+};
+
+// src/ui/should-resubmit-messages.ts
+function shouldResubmitMessages({
+  originalMaxToolInvocationStep,
+  originalMessageCount,
+  maxSteps: maxSteps2,
+  messages
+}) {
+  var _a17;
+  const lastMessage = messages[messages.length - 1];
+  return (
+    // check if the feature is enabled:
+    maxSteps2 > 1 && // ensure there is a last message:
+    lastMessage != null && // ensure we actually have new steps (to prevent infinite loops in case of errors):
+    (messages.length > originalMessageCount || extractMaxToolInvocationStep(getToolInvocations(lastMessage)) !== originalMaxToolInvocationStep) && // check that next step is possible:
+    isAssistantMessageWithCompletedToolCalls(lastMessage) && // limit the number of automatic steps:
+    ((_a17 = extractMaxToolInvocationStep(getToolInvocations(lastMessage))) != null ? _a17 : 0) < maxSteps2
+  );
+}
+function isAssistantMessageWithCompletedToolCalls(message) {
+  if (message.role !== "assistant") {
+    return false;
+  }
+  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);
+}
+
+// src/ui/update-tool-call-result.ts
+function updateToolCallResult({
+  messages,
+  toolCallId,
+  toolResult: result
+}) {
+  const lastMessage = messages[messages.length - 1];
+  const invocationPart = lastMessage.parts.find(
+    (part) => part.type === "tool-invocation" && part.toolInvocation.toolCallId === toolCallId
+  );
+  if (invocationPart == null) {
+    return;
+  }
+  invocationPart.toolInvocation = {
+    ...invocationPart.toolInvocation,
+    state: "result",
+    result
+  };
+}
+
+// src/ui/chat-store.ts
+var ChatStore = class {
+  constructor({
+    chats = {},
+    generateId: generateId3,
+    messageMetadataSchema,
+    transport,
+    maxSteps: maxSteps2 = 1
+  }) {
+    this.chats = new Map(
+      Object.entries(chats).map(([id, state]) => [
+        id,
+        {
+          messages: [...state.messages],
+          status: "ready",
+          activeResponse: void 0,
+          error: void 0,
+          jobExecutor: new SerialJobExecutor()
+        }
+      ])
+    );
+    this.maxSteps = maxSteps2;
+    this.transport = transport;
+    this.subscribers = /* @__PURE__ */ new Set();
+    this.generateId = generateId3 != null ? generateId3 : import_provider_utils5.generateId;
+    this.messageMetadataSchema = messageMetadataSchema;
+  }
+  hasChat(id) {
+    return this.chats.has(id);
+  }
+  addChat(id, messages) {
+    this.chats.set(id, {
+      messages,
+      status: "ready",
+      jobExecutor: new SerialJobExecutor()
+    });
+  }
+  getChats() {
+    return Array.from(this.chats.entries());
+  }
+  get chatCount() {
+    return this.chats.size;
+  }
+  getStatus(id) {
+    return this.getChat(id).status;
+  }
+  setStatus({
+    id,
+    status,
+    error
+  }) {
+    const chat = this.getChat(id);
+    if (chat.status === status)
+      return;
+    chat.status = status;
+    chat.error = error;
+    this.emit({ type: "chat-status-changed", chatId: id, error });
+  }
+  getError(id) {
+    return this.getChat(id).error;
+  }
+  getMessages(id) {
+    return this.getChat(id).messages;
+  }
+  getLastMessage(id) {
+    const chat = this.getChat(id);
+    return chat.messages[chat.messages.length - 1];
+  }
+  subscribe(subscriber) {
+    this.subscribers.add(subscriber);
+    return () => this.subscribers.delete(subscriber);
+  }
+  setMessages({
+    id,
+    messages
+  }) {
+    this.getChat(id).messages = [...messages];
+    this.emit({ type: "chat-messages-changed", chatId: id });
+  }
+  removeAssistantResponse(id) {
+    const chat = this.getChat(id);
+    const lastMessage = chat.messages[chat.messages.length - 1];
+    if (lastMessage == null) {
+      throw new Error("Cannot remove assistant response from empty chat");
+    }
+    if (lastMessage.role !== "assistant") {
+      throw new Error("Last message is not an assistant message");
+    }
+    this.setMessages({ id, messages: chat.messages.slice(0, -1) });
+  }
+  async submitMessage({
+    chatId,
+    message,
+    headers,
+    body,
+    onError,
+    onToolCall,
+    onFinish
+  }) {
+    var _a17;
+    const chat = this.getChat(chatId);
+    const currentMessages = chat.messages;
+    await this.triggerRequest({
+      chatId,
+      messages: currentMessages.concat({
+        ...message,
+        id: (_a17 = message.id) != null ? _a17 : this.generateId()
+      }),
+      headers,
+      body,
+      requestType: "generate",
+      onError,
+      onToolCall,
+      onFinish
+    });
+  }
+  async resubmitLastUserMessage({
+    chatId,
+    headers,
+    body,
+    onError,
+    onToolCall,
+    onFinish
+  }) {
+    const messages = this.getChat(chatId).messages;
+    const messagesToSubmit = messages[messages.length - 1].role === "assistant" ? messages.slice(0, -1) : messages;
+    if (messagesToSubmit.length === 0) {
+      return;
+    }
+    return this.triggerRequest({
+      chatId,
+      requestType: "generate",
+      messages: messagesToSubmit,
+      headers,
+      body,
+      onError,
+      onToolCall,
+      onFinish
+    });
+  }
+  async resumeStream({
+    chatId,
+    headers,
+    body,
+    onError,
+    onToolCall,
+    onFinish
+  }) {
+    const chat = this.getChat(chatId);
+    const currentMessages = chat.messages;
+    return this.triggerRequest({
+      chatId,
+      messages: currentMessages,
+      requestType: "resume",
+      headers,
+      body,
+      onError,
+      onToolCall,
+      onFinish
+    });
+  }
+  async addToolResult({
+    chatId,
+    toolCallId,
+    result
+  }) {
+    const chat = this.getChat(chatId);
+    chat.jobExecutor.run(async () => {
+      const currentMessages = chat.messages;
+      updateToolCallResult({
+        messages: currentMessages,
+        toolCallId,
+        toolResult: result
+      });
+      this.setMessages({ id: chatId, messages: currentMessages });
+      if (chat.status === "submitted" || chat.status === "streaming") {
+        return;
+      }
+      const lastMessage = currentMessages[currentMessages.length - 1];
+      if (isAssistantMessageWithCompletedToolCalls(lastMessage)) {
+        await this.triggerRequest({
+          messages: currentMessages,
+          requestType: "generate",
+          chatId
         });
-        break;
-      }
-      default: {
-        const exhaustiveCheck = streamProtocol;
-        throw new Error(`Unknown stream protocol: ${exhaustiveCheck}`);
       }
+    });
+  }
+  async stopStream({ chatId }) {
+    var _a17;
+    const chat = this.getChat(chatId);
+    if (chat.status !== "streaming" && chat.status !== "submitted")
+      return;
+    if ((_a17 = chat.activeResponse) == null ? void 0 : _a17.abortController) {
+      chat.activeResponse.abortController.abort();
+      chat.activeResponse.abortController = void 0;
     }
-    if (onFinish) {
-      onFinish(prompt, result);
+  }
+  emit(event) {
+    for (const subscriber of this.subscribers) {
+      subscriber.onChatChanged(event);
     }
-    setAbortController(null);
-    return result;
-  } catch (err) {
-    if (err.name === "AbortError") {
-      setAbortController(null);
-      return null;
+  }
+  getChat(id) {
+    if (!this.hasChat(id)) {
+      throw new Error(`chat '${id}' not found`);
     }
-    if (err instanceof Error) {
-      if (onError) {
+    return this.chats.get(id);
+  }
+  async triggerRequest({
+    chatId,
+    messages: chatMessages,
+    requestType,
+    headers,
+    body,
+    onError,
+    onToolCall,
+    onFinish
+  }) {
+    const self = this;
+    const chat = this.getChat(chatId);
+    this.setMessages({ id: chatId, messages: chatMessages });
+    this.setStatus({ id: chatId, status: "submitted", error: void 0 });
+    const messageCount = chatMessages.length;
+    const maxStep = extractMaxToolInvocationStep(
+      getToolInvocations(chatMessages[chatMessages.length - 1])
+    );
+    try {
+      const activeResponse = {
+        state: createStreamingUIMessageState({
+          lastMessage: chatMessages[chatMessages.length - 1],
+          newMessageId: self.generateId()
+        }),
+        abortController: new AbortController()
+      };
+      chat.activeResponse = activeResponse;
+      const stream = await self.transport.submitMessages({
+        chatId,
+        messages: chatMessages,
+        body,
+        headers,
+        abortController: activeResponse.abortController,
+        requestType
+      });
+      const runUpdateMessageJob = (job) => (
+        // serialize the job execution to avoid race conditions:
+        chat.jobExecutor.run(
+          () => job({
+            state: activeResponse.state,
+            write: () => {
+              self.setStatus({ id: chatId, status: "streaming" });
+              const replaceLastMessage = activeResponse.state.message.id === chatMessages[chatMessages.length - 1].id;
+              const newMessages = [
+                ...replaceLastMessage ? chatMessages.slice(0, chatMessages.length - 1) : chatMessages,
+                activeResponse.state.message
+              ];
+              self.setMessages({
+                id: chatId,
+                messages: newMessages
+              });
+            }
+          })
+        )
+      );
+      await consumeStream({
+        stream: processUIMessageStream({
+          stream,
+          onToolCall,
+          messageMetadataSchema: self.messageMetadataSchema,
+          runUpdateMessageJob
+        }),
+        onError: (error) => {
+          throw error;
+        }
+      });
+      onFinish == null ? void 0 : onFinish({ message: activeResponse.state.message });
+      this.setStatus({ id: chatId, status: "ready" });
+    } catch (err) {
+      if (err.name === "AbortError") {
+        this.setStatus({ id: chatId, status: "ready" });
+        return null;
+      }
+      if (onError && err instanceof Error) {
         onError(err);
       }
+      this.setStatus({ id: chatId, status: "error", error: err });
+    } finally {
+      chat.activeResponse = void 0;
+    }
+    const currentMessages = self.getMessages(chatId);
+    if (shouldResubmitMessages({
+      originalMaxToolInvocationStep: maxStep,
+      originalMessageCount: messageCount,
+      maxSteps: self.maxSteps,
+      messages: currentMessages
+    })) {
+      await self.triggerRequest({
+        chatId,
+        requestType,
+        onError,
+        onToolCall,
+        onFinish,
+        headers,
+        body,
+        messages: currentMessages
+      });
     }
-    setError(err);
-  } finally {
-    setLoading(false);
   }
-}
+};
+
+// src/ui/chat-transport.ts
+var DefaultChatTransport = class {
+  constructor({
+    api,
+    credentials,
+    headers,
+    body,
+    streamProtocol,
+    fetch: fetch2,
+    prepareRequestBody
+  }) {
+    this.api = api;
+    this.credentials = credentials;
+    this.headers = headers;
+    this.body = body;
+    this.streamProtocol = streamProtocol;
+    this.fetch = fetch2;
+    this.prepareRequestBody = prepareRequestBody;
+  }
+  submitMessages({
+    chatId,
+    messages,
+    abortController,
+    body,
+    headers,
+    requestType
+  }) {
+    var _a17, _b;
+    return fetchUIMessageStream({
+      api: this.api,
+      headers: {
+        ...this.headers,
+        ...headers
+      },
+      body: (_b = (_a17 = this.prepareRequestBody) == null ? void 0 : _a17.call(this, {
+        id: chatId,
+        // TODO change to chatId
+        messages,
+        ...this.body,
+        ...body
+      })) != null ? _b : {
+        id: chatId,
+        // TODO change to chatId
+        messages,
+        ...this.body,
+        ...body
+      },
+      streamProtocol: this.streamProtocol,
+      credentials: this.credentials,
+      abortController: () => abortController,
+      fetch: this.fetch,
+      requestType
+    });
+  }
+};
 
 // src/ui/convert-file-list-to-file-ui-parts.ts
 async function convertFileListToFileUIParts(files) {
@@ -1740,53 +2223,36 @@ function convertToModelMessages(messages, options) {
 }
 var convertToCoreMessages = convertToModelMessages;
 
-// src/ui/should-resubmit-messages.ts
-function shouldResubmitMessages({
-  originalMaxToolInvocationStep,
-  originalMessageCount,
-  maxSteps,
-  messages
-}) {
-  var _a17;
-  const lastMessage = messages[messages.length - 1];
-  return (
-    // check if the feature is enabled:
-    maxSteps > 1 && // ensure there is a last message:
-    lastMessage != null && // ensure we actually have new steps (to prevent infinite loops in case of errors):
-    (messages.length > originalMessageCount || extractMaxToolInvocationStep(getToolInvocations(lastMessage)) !== originalMaxToolInvocationStep) && // check that next step is possible:
-    isAssistantMessageWithCompletedToolCalls(lastMessage) && // limit the number of automatic steps:
-    ((_a17 = extractMaxToolInvocationStep(getToolInvocations(lastMessage))) != null ? _a17 : 0) < maxSteps
-  );
-}
-function isAssistantMessageWithCompletedToolCalls(message) {
-  if (message.role !== "assistant") {
-    return false;
-  }
-  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);
-}
-
-// src/ui/update-tool-call-result.ts
-function updateToolCallResult({
-  messages,
-  toolCallId,
-  toolResult: result
+// src/ui/default-chat-store.ts
+var import_provider_utils6 = require("@ai-sdk/provider-utils");
+function defaultChatStore({
+  api,
+  fetch: fetch2,
+  streamProtocol = "ui-message",
+  credentials,
+  headers,
+  body,
+  prepareRequestBody,
+  generateId: generateId3 = import_provider_utils6.generateId,
+  messageMetadataSchema,
+  maxSteps: maxSteps2 = 1,
+  chats
 }) {
-  const lastMessage = messages[messages.length - 1];
-  const invocationPart = lastMessage.parts.find(
-    (part) => part.type === "tool-invocation" && part.toolInvocation.toolCallId === toolCallId
-  );
-  if (invocationPart == null) {
-    return;
-  }
-  invocationPart.toolInvocation = {
-    ...invocationPart.toolInvocation,
-    state: "result",
-    result
-  };
+  return new ChatStore({
+    transport: new DefaultChatTransport({
+      api,
+      fetch: fetch2,
+      streamProtocol,
+      credentials,
+      headers,
+      body,
+      prepareRequestBody
+    }),
+    generateId: generateId3,
+    messageMetadataSchema,
+    maxSteps: maxSteps2,
+    chats
+  });
 }
 
 // src/ui-message-stream/create-ui-message-stream.ts
@@ -1824,7 +2290,7 @@ function createUIMessageStream({
               safeEnqueue(value);
             }
           })().catch((error) => {
-            safeEnqueue({ type: "error", value: onError(error) });
+            safeEnqueue({ type: "error", errorText: onError(error) });
           })
         );
       },
@@ -1833,12 +2299,12 @@ function createUIMessageStream({
     if (result) {
       ongoingStreamPromises.push(
         result.catch((error) => {
-          safeEnqueue({ type: "error", value: onError(error) });
+          safeEnqueue({ type: "error", errorText: onError(error) });
         })
       );
     }
   } catch (error) {
-    safeEnqueue({ type: "error", value: onError(error) });
+    safeEnqueue({ type: "error", errorText: onError(error) });
   }
   const waitForStreams = new Promise(async (resolve) => {
     while (ongoingStreamPromises.length > 0) {
@@ -1993,7 +2459,7 @@ function cosineSimilarity(vector1, vector2) {
 }
 
 // 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,
@@ -2001,7 +2467,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) {
@@ -2017,7 +2483,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,
@@ -2035,13 +2501,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) {
@@ -2052,7 +2518,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 },
@@ -2611,7 +3077,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",
@@ -2718,7 +3184,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);
 };
@@ -2744,7 +3210,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,
@@ -2758,14 +3224,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;
   }
@@ -2872,7 +3338,7 @@ var DefaultGenerateImageResult = class {
 
 // core/generate-object/generate-object.ts
 var import_provider21 = 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) {
@@ -2886,7 +3352,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 }) {
@@ -2915,7 +3381,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
@@ -2981,7 +3447,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.",
@@ -3132,7 +3598,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
@@ -3303,7 +3769,7 @@ function prepareCallSettings({
 
 // 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
@@ -3475,14 +3941,14 @@ 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)
   });
   if (!validationResult.success) {
     throw new import_provider19.InvalidPromptError({
       prompt,
-      message: "messages must be an array of ModelMessage",
+      message: "The messages must be a ModelMessage[]. If you have passed a UIMessage[], you can use convertToModelMessages to convert them.",
       cause: validationResult.error
     });
   }
@@ -3494,7 +3960,7 @@ async function standardizePrompt(prompt) {
 
 // core/generate-object/output-strategy.ts
 var import_provider20 = 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) {
@@ -3550,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_provider20.UnsupportedFunctionalityError({
@@ -3594,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;
         }
@@ -3635,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;
         }
@@ -3753,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":
@@ -3886,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,
@@ -4062,7 +4528,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.",
@@ -4158,7 +4624,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() {
@@ -4302,7 +4768,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,
@@ -4897,10 +5363,10 @@ 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");
 
 // 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) {
@@ -4932,7 +5398,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 {
@@ -5002,7 +5468,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,
@@ -5026,7 +5492,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,
@@ -5056,8 +5522,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,
@@ -5115,6 +5581,19 @@ var DefaultStepResult = class {
   }
 };
 
+// core/generate-text/stop-condition.ts
+function maxSteps(maxSteps2) {
+  return ({ steps }) => steps.length >= maxSteps2;
+}
+function hasToolCall(toolName) {
+  return ({ steps }) => {
+    var _a17, _b, _c;
+    return (_c = (_b = (_a17 = steps[steps.length - 1]) == null ? void 0 : _a17.toolCalls) == null ? void 0 : _b.some(
+      (toolCall) => toolCall.toolName === toolName
+    )) != null ? _c : false;
+  };
+}
+
 // core/generate-text/to-response-messages.ts
 function toResponseMessages({
   content: inputContent,
@@ -5174,7 +5653,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
 });
@@ -5188,7 +5667,7 @@ async function generateText({
   maxRetries: maxRetriesArg,
   abortSignal,
   headers,
-  maxSteps = 1,
+  continueUntil = maxSteps(1),
   experimental_output: output,
   experimental_telemetry: telemetry,
   providerOptions,
@@ -5202,13 +5681,6 @@ async function generateText({
   onStepFinish,
   ...settings
 }) {
-  if (maxSteps < 1) {
-    throw new InvalidArgumentError({
-      parameter: "maxSteps",
-      value: maxSteps,
-      message: "maxSteps must be at least 1"
-    });
-  }
   const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
   const callSettings = prepareCallSettings(settings);
   const baseTelemetryAttributes = getBaseTelemetryAttributes({
@@ -5239,8 +5711,7 @@ async function generateText({
         // specific settings that only make sense on the outer level:
         "ai.prompt": {
           input: () => JSON.stringify({ system, prompt, messages })
-        },
-        "ai.settings.maxSteps": maxSteps
+        }
       }
     }),
     tracer,
@@ -5250,7 +5721,6 @@ async function generateText({
       let currentModelResponse;
       let currentToolCalls = [];
       let currentToolResults = [];
-      let stepCount = 0;
       const responseMessages = [];
       const steps = [];
       do {
@@ -5261,8 +5731,7 @@ async function generateText({
         const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
           model,
           steps,
-          maxSteps,
-          stepNumber: stepCount
+          stepNumber: steps.length
         }));
         const promptMessages = await convertToLanguageModelPrompt({
           prompt: {
@@ -5417,9 +5886,12 @@ async function generateText({
         });
         steps.push(currentStepResult);
         await (onStepFinish == null ? void 0 : onStepFinish(currentStepResult));
-      } while (++stepCount < maxSteps && // there are tool calls:
-      currentToolCalls.length > 0 && // all current tool calls have results:
-      currentToolResults.length === currentToolCalls.length);
+      } while (
+        // there are tool calls:
+        currentToolCalls.length > 0 && // all current tool calls have results:
+        currentToolResults.length === currentToolCalls.length && // continue until the stop condition is met:
+        !await continueUntil({ steps })
+      );
       span.setAttributes(
         selectTelemetryAttributes({
           telemetry,
@@ -5622,7 +6094,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" },
@@ -5636,7 +6108,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: {
@@ -5662,7 +6134,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.",
@@ -5673,7 +6145,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
       });
@@ -5693,7 +6165,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_provider23 = require("@ai-sdk/provider");
 var CHUNKING_REGEXPS = {
   word: /\S+\s+/m,
@@ -5702,7 +6174,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") {
@@ -5762,7 +6234,7 @@ 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");
 
 // src/util/as-array.ts
 function asArray(value) {
@@ -5770,7 +6242,7 @@ function asArray(value) {
 }
 
 // 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,
@@ -5856,7 +6328,7 @@ function runToolsTransformation({
             controller.enqueue(toolCall);
             const tool2 = tools[toolCall.toolName];
             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",
@@ -5965,7 +6437,7 @@ function runToolsTransformation({
 }
 
 // core/generate-text/stream-text.ts
-var originalGenerateId4 = (0, import_provider_utils22.createIdGenerator)({
+var originalGenerateId4 = (0, import_provider_utils23.createIdGenerator)({
   prefix: "aitxt",
   size: 24
 });
@@ -5979,7 +6451,7 @@ function streamText({
   maxRetries,
   abortSignal,
   headers,
-  maxSteps = 1,
+  maxSteps: maxSteps2 = 1,
   experimental_output: output,
   experimental_telemetry: telemetry,
   providerOptions,
@@ -6015,7 +6487,7 @@ function streamText({
     transforms: asArray(transform),
     activeTools,
     repairToolCall,
-    maxSteps,
+    maxSteps: maxSteps2,
     output,
     providerOptions,
     onChunk,
@@ -6092,7 +6564,7 @@ var DefaultStreamTextResult = class {
     transforms,
     activeTools,
     repairToolCall,
-    maxSteps,
+    maxSteps: maxSteps2,
     output,
     providerOptions,
     now: now2,
@@ -6106,10 +6578,10 @@ var DefaultStreamTextResult = class {
     this.totalUsagePromise = new DelayedPromise();
     this.finishReasonPromise = new DelayedPromise();
     this.stepsPromise = new DelayedPromise();
-    if (maxSteps < 1) {
+    if (maxSteps2 < 1) {
       throw new InvalidArgumentError({
         parameter: "maxSteps",
-        value: maxSteps,
+        value: maxSteps2,
         message: "maxSteps must be at least 1"
       });
     }
@@ -6266,6 +6738,13 @@ var DefaultStreamTextResult = class {
     this.addStream = stitchableStream.addStream;
     this.closeStream = stitchableStream.close;
     let stream = stitchableStream.stream;
+    stream = stream.pipeThrough(
+      new TransformStream({
+        start(controller) {
+          controller.enqueue({ type: "start" });
+        }
+      })
+    );
     for (const transform of transforms) {
       stream = stream.pipeThrough(
         transform({
@@ -6300,7 +6779,7 @@ var DefaultStreamTextResult = class {
           "ai.prompt": {
             input: () => JSON.stringify({ system, prompt, messages })
           },
-          "ai.settings.maxSteps": maxSteps
+          "ai.settings.maxSteps": maxSteps2
         }
       }),
       tracer,
@@ -6438,7 +6917,6 @@ var DefaultStreamTextResult = class {
                   var _a17, _b, _c, _d;
                   if (chunk.type === "stream-start") {
                     warnings = chunk.warnings;
-                    controller.enqueue({ type: "start" });
                     return;
                   }
                   if (stepFirstChunk) {
@@ -6588,7 +7066,7 @@ var DefaultStreamTextResult = class {
                     }
                   });
                   const combinedUsage = addLanguageModelUsage(usage, stepUsage);
-                  if (currentStep + 1 < maxSteps && // there are tool calls:
+                  if (currentStep + 1 < maxSteps2 && // there are tool calls:
                   stepToolCalls.length > 0 && // all current tool calls have results:
                   stepToolResults.length === stepToolCalls.length) {
                     responseMessages.push(
@@ -6773,125 +7251,122 @@ var DefaultStreamTextResult = class {
           const partType = part.type;
           switch (partType) {
             case "text": {
-              controller.enqueue({ type: "text", value: part.text });
+              controller.enqueue({
+                type: "text",
+                text: part.text
+              });
               break;
             }
             case "reasoning": {
               if (sendReasoning) {
-                controller.enqueue({ type: "reasoning", value: part });
+                controller.enqueue({
+                  type: "reasoning",
+                  text: part.text,
+                  providerMetadata: part.providerMetadata
+                });
               }
               break;
             }
             case "reasoning-part-finish": {
               if (sendReasoning) {
-                controller.enqueue({
-                  type: "reasoning-part-finish",
-                  value: null
-                });
+                controller.enqueue({ type: "reasoning-part-finish" });
               }
               break;
             }
             case "file": {
               controller.enqueue({
                 type: "file",
-                value: {
-                  mediaType: part.file.mediaType,
-                  url: `data:${part.file.mediaType};base64,${part.file.base64}`
-                }
+                mediaType: part.file.mediaType,
+                url: `data:${part.file.mediaType};base64,${part.file.base64}`
               });
               break;
             }
             case "source": {
               if (sendSources) {
-                controller.enqueue({ type: "source", value: part });
+                controller.enqueue({
+                  type: "source",
+                  sourceType: part.sourceType,
+                  id: part.id,
+                  url: part.url,
+                  title: part.title,
+                  providerMetadata: part.providerMetadata
+                });
               }
               break;
             }
             case "tool-call-streaming-start": {
               controller.enqueue({
                 type: "tool-call-streaming-start",
-                value: {
-                  toolCallId: part.toolCallId,
-                  toolName: part.toolName
-                }
+                toolCallId: part.toolCallId,
+                toolName: part.toolName
               });
               break;
             }
             case "tool-call-delta": {
               controller.enqueue({
                 type: "tool-call-delta",
-                value: {
-                  toolCallId: part.toolCallId,
-                  argsTextDelta: part.argsTextDelta
-                }
+                toolCallId: part.toolCallId,
+                argsTextDelta: part.argsTextDelta
               });
               break;
             }
             case "tool-call": {
               controller.enqueue({
                 type: "tool-call",
-                value: {
-                  toolCallId: part.toolCallId,
-                  toolName: part.toolName,
-                  args: part.args
-                }
+                toolCallId: part.toolCallId,
+                toolName: part.toolName,
+                args: part.args
               });
               break;
             }
             case "tool-result": {
               controller.enqueue({
                 type: "tool-result",
-                value: {
-                  toolCallId: part.toolCallId,
-                  result: part.result
-                }
+                toolCallId: part.toolCallId,
+                result: part.result
               });
               break;
             }
             case "error": {
               controller.enqueue({
                 type: "error",
-                value: onError(part.error)
+                errorText: onError(part.error)
               });
               break;
             }
             case "start-step": {
+              const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
               controller.enqueue({
                 type: "start-step",
-                value: {
-                  metadata: messageMetadata == null ? void 0 : messageMetadata({ part })
-                }
+                metadata
               });
               break;
             }
             case "finish-step": {
+              const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
               controller.enqueue({
                 type: "finish-step",
-                value: {
-                  metadata: messageMetadata == null ? void 0 : messageMetadata({ part })
-                }
+                metadata
               });
               break;
             }
             case "start": {
               if (experimental_sendStart) {
+                const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "start",
-                  value: {
-                    messageId,
-                    metadata: messageMetadata == null ? void 0 : messageMetadata({ part })
-                  }
+                  messageId,
+                  metadata
                 });
               }
               break;
             }
             case "finish": {
               if (experimental_sendFinish) {
+                const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "finish",
-                  value: {
-                    metadata: messageMetadata == null ? void 0 : messageMetadata({ part })
-                  }
+                  metadata
                 });
               }
               break;
@@ -6904,22 +7379,38 @@ var DefaultStreamTextResult = class {
         }
       })
     );
-    return onFinish == null ? baseStream : processUIMessageStream({
-      stream: baseStream,
+    if (onFinish == null) {
+      return baseStream;
+    }
+    const state = createStreamingUIMessageState({
       lastMessage,
-      newMessageId: messageId != null ? messageId : this.generateId(),
-      onFinish: ({ message }) => {
-        const isContinuation2 = message.id === (lastMessage == null ? void 0 : lastMessage.id);
-        onFinish({
-          isContinuation: isContinuation2,
-          responseMessage: message,
-          messages: [
-            ...isContinuation2 ? originalMessages.slice(0, -1) : originalMessages,
-            message
-          ]
-        });
-      }
+      newMessageId: messageId != null ? messageId : this.generateId()
     });
+    const runUpdateMessageJob = async (job) => {
+      await job({ state, write: () => {
+      } });
+    };
+    return processUIMessageStream({
+      stream: baseStream,
+      runUpdateMessageJob
+    }).pipeThrough(
+      new TransformStream({
+        transform(chunk, controller) {
+          controller.enqueue(chunk);
+        },
+        flush() {
+          const isContinuation2 = state.message.id === (lastMessage == null ? void 0 : lastMessage.id);
+          onFinish({
+            isContinuation: isContinuation2,
+            responseMessage: state.message,
+            messages: [
+              ...isContinuation2 ? originalMessages.slice(0, -1) : originalMessages,
+              state.message
+            ]
+          });
+        }
+      })
+    );
   }
   pipeUIMessageStreamToResponse(response, {
     newMessageId,
@@ -7367,7 +7858,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) {
@@ -7375,7 +7866,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");
@@ -7546,7 +8037,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;
@@ -7870,7 +8361,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
@@ -8000,6 +8491,8 @@ var DefaultTranscriptionResult = class {
 0 && (module.exports = {
   AISDKError,
   APICallError,
+  ChatStore,
+  DefaultChatTransport,
   DownloadError,
   EmptyResponseBodyError,
   InvalidArgumentError,
@@ -8047,6 +8540,7 @@ var DefaultTranscriptionResult = class {
   createUIMessageStream,
   createUIMessageStreamResponse,
   customProvider,
+  defaultChatStore,
   defaultSettingsMiddleware,
   embed,
   embedMany,
@@ -8063,14 +8557,15 @@ var DefaultTranscriptionResult = class {
   generateText,
   getTextFromDataUrl,
   getToolInvocations,
+  hasToolCall,
   isAssistantMessageWithCompletedToolCalls,
   isDeepEqualData,
   jsonSchema,
+  maxSteps,
   modelMessageSchema,
   parsePartialJson,
   pipeTextStreamToResponse,
   pipeUIMessageStreamToResponse,
-  processTextStream,
   shouldResubmitMessages,
   simulateReadableStream,
   simulateStreamingMiddleware,