ai 5.0.0-alpha.3 → 5.0.0-alpha.5

package/dist/index.js

@@ -47,6 +47,7 @@ __export(src_exports, {
   NoSuchToolError: () => NoSuchToolError,
   Output: () => output_exports,
   RetryError: () => RetryError,
+  SerialJobExecutor: () => SerialJobExecutor,
   TextStreamChatTransport: () => TextStreamChatTransport,
   ToolCallRepairError: () => ToolCallRepairError,
   ToolExecutionError: () => ToolExecutionError,
@@ -72,7 +73,7 @@ __export(src_exports, {
   createUIMessageStream: () => createUIMessageStream,
   createUIMessageStreamResponse: () => createUIMessageStreamResponse,
   customProvider: () => customProvider,
-  defaultChatStore: () => defaultChatStore,
+  defaultChatStoreOptions: () => defaultChatStoreOptions,
   defaultSettingsMiddleware: () => defaultSettingsMiddleware,
   embed: () => embed,
   embedMany: () => embedMany,
@@ -93,7 +94,6 @@ __export(src_exports, {
   isAssistantMessageWithCompletedToolCalls: () => isAssistantMessageWithCompletedToolCalls,
   isDeepEqualData: () => isDeepEqualData,
   jsonSchema: () => import_provider_utils26.jsonSchema,
-  maxSteps: () => maxSteps,
   modelMessageSchema: () => modelMessageSchema,
   parsePartialJson: () => parsePartialJson,
   pipeTextStreamToResponse: () => pipeTextStreamToResponse,
@@ -102,6 +102,7 @@ __export(src_exports, {
   simulateReadableStream: () => simulateReadableStream,
   simulateStreamingMiddleware: () => simulateStreamingMiddleware,
   smoothStream: () => smoothStream,
+  stepCountIs: () => stepCountIs,
   streamObject: () => streamObject,
   streamText: () => streamText,
   systemModelMessageSchema: () => systemModelMessageSchema,
@@ -1048,12 +1049,12 @@ function getToolInvocations(message) {
 // src/ui/process-ui-message-stream.ts
 function createStreamingUIMessageState({
   lastMessage,
-  newMessageId = "no-id"
+  newMessageId = ""
 } = {}) {
   var _a17;
   const isContinuation = (lastMessage == null ? void 0 : lastMessage.role) === "assistant";
   const step = isContinuation ? 1 + ((_a17 = extractMaxToolInvocationStep(getToolInvocations(lastMessage))) != null ? _a17 : 0) : 0;
-  const message = isContinuation ? structuredClone(lastMessage) : {
+  const message = isContinuation ? lastMessage : {
     id: newMessageId,
     metadata: {},
     role: "assistant",
@@ -1433,7 +1434,7 @@ async function consumeUIMessageStream({
   messageMetadataSchema
 }) {
   const state = createStreamingUIMessageState({
-    lastMessage,
+    lastMessage: lastMessage ? structuredClone(lastMessage) : void 0,
     newMessageId: generateId3()
   });
   const runUpdateMessageJob = async (job) => {
@@ -1634,54 +1635,22 @@ async function callCompletionApi({
 // 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,
+  maxSteps,
   messages
 }) {
   var _a17;
   const lastMessage = messages[messages.length - 1];
   return (
     // check if the feature is enabled:
-    maxSteps2 > 1 && // ensure there is a last message:
+    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) < maxSteps2
+    ((_a17 = extractMaxToolInvocationStep(getToolInvocations(lastMessage))) != null ? _a17 : 0) < maxSteps
   );
 }
 function isAssistantMessageWithCompletedToolCalls(message) {
@@ -1721,23 +1690,19 @@ var ChatStore = class {
     chats = {},
     generateId: generateId3,
     transport,
-    maxSteps: maxSteps2 = 1,
+    maxSteps = 1,
     messageMetadataSchema,
-    dataPartSchemas
+    dataPartSchemas,
+    createChat
   }) {
+    this.createChat = createChat;
     this.chats = new Map(
-      Object.entries(chats).map(([id, state]) => [
+      Object.entries(chats).map(([id, chat]) => [
         id,
-        {
-          messages: [...state.messages],
-          status: "ready",
-          activeResponse: void 0,
-          error: void 0,
-          jobExecutor: new SerialJobExecutor()
-        }
+        this.createChat({ messages: chat.messages })
       ])
     );
-    this.maxSteps = maxSteps2;
+    this.maxSteps = maxSteps;
     this.transport = transport;
     this.subscribers = /* @__PURE__ */ new Set();
     this.generateId = generateId3 != null ? generateId3 : import_provider_utils5.generateId;
@@ -1748,11 +1713,7 @@ var ChatStore = class {
     return this.chats.has(id);
   }
   addChat(id, messages) {
-    this.chats.set(id, {
-      messages,
-      status: "ready",
-      jobExecutor: new SerialJobExecutor()
-    });
+    this.chats.set(id, this.createChat({ messages }));
   }
   getChats() {
     return Array.from(this.chats.entries());
@@ -1761,28 +1722,28 @@ var ChatStore = class {
     return this.chats.size;
   }
   getStatus(id) {
-    return this.getChat(id).status;
+    return this.getChatState(id).status;
   }
   setStatus({
     id,
     status,
     error
   }) {
-    const chat = this.getChat(id);
-    if (chat.status === status)
+    const state = this.getChatState(id);
+    if (state.status === status)
       return;
-    chat.status = status;
-    chat.error = error;
+    state.setStatus(status);
+    state.setError(error);
     this.emit({ type: "chat-status-changed", chatId: id, error });
   }
   getError(id) {
-    return this.getChat(id).error;
+    return this.getChatState(id).error;
   }
   getMessages(id) {
-    return this.getChat(id).messages;
+    return this.getChatState(id).messages;
   }
   getLastMessage(id) {
-    const chat = this.getChat(id);
+    const chat = this.getChatState(id);
     return chat.messages[chat.messages.length - 1];
   }
   subscribe(subscriber) {
@@ -1793,11 +1754,11 @@ var ChatStore = class {
     id,
     messages
   }) {
-    this.getChat(id).messages = [...messages];
+    this.getChatState(id).setMessages(messages);
     this.emit({ type: "chat-messages-changed", chatId: id });
   }
   removeAssistantResponse(id) {
-    const chat = this.getChat(id);
+    const chat = this.getChatState(id);
     const lastMessage = chat.messages[chat.messages.length - 1];
     if (lastMessage == null) {
       throw new Error("Cannot remove assistant response from empty chat");
@@ -1805,7 +1766,8 @@ var ChatStore = class {
     if (lastMessage.role !== "assistant") {
       throw new Error("Last message is not an assistant message");
     }
-    this.setMessages({ id, messages: chat.messages.slice(0, -1) });
+    chat.popMessage();
+    this.emit({ type: "chat-messages-changed", chatId: id });
   }
   async submitMessage({
     chatId,
@@ -1817,14 +1779,14 @@ var ChatStore = class {
     onFinish
   }) {
     var _a17;
-    const chat = this.getChat(chatId);
-    const currentMessages = chat.messages;
+    const state = this.getChatState(chatId);
+    state.pushMessage({ ...message, id: (_a17 = message.id) != null ? _a17 : this.generateId() });
+    this.emit({
+      type: "chat-messages-changed",
+      chatId
+    });
     await this.triggerRequest({
       chatId,
-      messages: currentMessages.concat({
-        ...message,
-        id: (_a17 = message.id) != null ? _a17 : this.generateId()
-      }),
       headers,
       body,
       requestType: "generate",
@@ -1841,15 +1803,20 @@ var ChatStore = class {
     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) {
+    const chat = this.getChatState(chatId);
+    if (chat.messages[chat.messages.length - 1].role === "assistant") {
+      chat.popMessage();
+      this.emit({
+        type: "chat-messages-changed",
+        chatId
+      });
+    }
+    if (chat.messages.length === 0) {
       return;
     }
     return this.triggerRequest({
       chatId,
       requestType: "generate",
-      messages: messagesToSubmit,
       headers,
       body,
       onError,
@@ -1865,11 +1832,8 @@ var ChatStore = class {
     onToolCall,
     onFinish
   }) {
-    const chat = this.getChat(chatId);
-    const currentMessages = chat.messages;
     return this.triggerRequest({
       chatId,
-      messages: currentMessages,
       requestType: "resume",
       headers,
       body,
@@ -1883,22 +1847,23 @@ var ChatStore = class {
     toolCallId,
     result
   }) {
-    const chat = this.getChat(chatId);
+    const chat = this.getChatState(chatId);
     chat.jobExecutor.run(async () => {
-      const currentMessages = chat.messages;
       updateToolCallResult({
-        messages: currentMessages,
+        messages: chat.messages,
         toolCallId,
         toolResult: result
       });
-      this.setMessages({ id: chatId, messages: currentMessages });
+      this.setMessages({
+        id: chatId,
+        messages: chat.messages
+      });
       if (chat.status === "submitted" || chat.status === "streaming") {
         return;
       }
-      const lastMessage = currentMessages[currentMessages.length - 1];
+      const lastMessage = chat.messages[chat.messages.length - 1];
       if (isAssistantMessageWithCompletedToolCalls(lastMessage)) {
-        await this.triggerRequest({
-          messages: currentMessages,
+        this.triggerRequest({
           requestType: "generate",
           chatId
         });
@@ -1907,7 +1872,7 @@ var ChatStore = class {
   }
   async stopStream({ chatId }) {
     var _a17;
-    const chat = this.getChat(chatId);
+    const chat = this.getChatState(chatId);
     if (chat.status !== "streaming" && chat.status !== "submitted")
       return;
     if ((_a17 = chat.activeResponse) == null ? void 0 : _a17.abortController) {
@@ -1920,15 +1885,14 @@ var ChatStore = class {
       subscriber.onChatChanged(event);
     }
   }
-  getChat(id) {
+  getChatState(id) {
     if (!this.hasChat(id)) {
-      throw new Error(`chat '${id}' not found`);
+      this.addChat(id, []);
     }
     return this.chats.get(id);
   }
   async triggerRequest({
     chatId,
-    messages: chatMessages,
     requestType,
     headers,
     body,
@@ -1936,26 +1900,25 @@ var ChatStore = class {
     onToolCall,
     onFinish
   }) {
-    const self = this;
-    const chat = this.getChat(chatId);
-    this.setMessages({ id: chatId, messages: chatMessages });
+    const chat = this.getChatState(chatId);
     this.setStatus({ id: chatId, status: "submitted", error: void 0 });
-    const messageCount = chatMessages.length;
+    const messageCount = chat.messages.length;
     const maxStep = extractMaxToolInvocationStep(
-      getToolInvocations(chatMessages[chatMessages.length - 1])
+      getToolInvocations(chat.messages[chat.messages.length - 1])
     );
     try {
+      const lastMessage = chat.messages[chat.messages.length - 1];
       const activeResponse = {
         state: createStreamingUIMessageState({
-          lastMessage: chatMessages[chatMessages.length - 1],
-          newMessageId: self.generateId()
+          lastMessage: chat.snapshot ? chat.snapshot(lastMessage) : lastMessage,
+          newMessageId: this.generateId()
         }),
         abortController: new AbortController()
       };
-      chat.activeResponse = activeResponse;
-      const stream = await self.transport.submitMessages({
+      chat.setActiveResponse(activeResponse);
+      const stream = await this.transport.submitMessages({
         chatId,
-        messages: chatMessages,
+        messages: chat.messages,
         body,
         headers,
         abortController: activeResponse.abortController,
@@ -1967,15 +1930,19 @@ var ChatStore = class {
           () => 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
+              this.setStatus({ id: chatId, status: "streaming" });
+              const replaceLastMessage = activeResponse.state.message.id === chat.messages[chat.messages.length - 1].id;
+              if (replaceLastMessage) {
+                chat.replaceMessage(
+                  chat.messages.length - 1,
+                  activeResponse.state.message
+                );
+              } else {
+                chat.pushMessage(activeResponse.state.message);
+              }
+              this.emit({
+                type: "chat-messages-changed",
+                chatId
               });
             }
           })
@@ -1985,8 +1952,8 @@ var ChatStore = class {
         stream: processUIMessageStream({
           stream,
           onToolCall,
-          messageMetadataSchema: self.messageMetadataSchema,
-          dataPartSchemas: self.dataPartSchemas,
+          messageMetadataSchema: this.messageMetadataSchema,
+          dataPartSchemas: this.dataPartSchemas,
           runUpdateMessageJob
         }),
         onError: (error) => {
@@ -2005,24 +1972,22 @@ var ChatStore = class {
       }
       this.setStatus({ id: chatId, status: "error", error: err });
     } finally {
-      chat.activeResponse = void 0;
+      chat.setActiveResponse(void 0);
     }
-    const currentMessages = self.getMessages(chatId);
     if (shouldResubmitMessages({
       originalMaxToolInvocationStep: maxStep,
       originalMessageCount: messageCount,
-      maxSteps: self.maxSteps,
-      messages: currentMessages
+      maxSteps: this.maxSteps,
+      messages: chat.messages
     })) {
-      await self.triggerRequest({
+      await this.triggerRequest({
         chatId,
         requestType,
         onError,
         onToolCall,
         onFinish,
         headers,
-        body,
-        messages: currentMessages
+        body
       });
     }
   }
@@ -2314,22 +2279,22 @@ function convertToModelMessages(messages, options) {
 }
 var convertToCoreMessages = convertToModelMessages;
 
-// src/ui/default-chat-store.ts
+// src/ui/default-chat-store-options.ts
 var import_provider_utils6 = require("@ai-sdk/provider-utils");
-function defaultChatStore({
-  api,
+function defaultChatStoreOptions({
+  api = "/api/chat",
   fetch: fetch2,
   credentials,
   headers,
   body,
   prepareRequestBody,
   generateId: generateId3 = import_provider_utils6.generateId,
-  dataPartSchemas,
   messageMetadataSchema,
-  maxSteps: maxSteps2 = 1,
+  maxSteps = 1,
+  dataPartSchemas,
   chats
 }) {
-  return new ChatStore({
+  return () => ({
     transport: new DefaultChatTransport({
       api,
       fetch: fetch2,
@@ -2341,16 +2306,62 @@ function defaultChatStore({
     generateId: generateId3,
     messageMetadataSchema,
     dataPartSchemas,
-    maxSteps: maxSteps2,
+    maxSteps,
     chats
   });
 }
 
+// src/ui-message-stream/handle-ui-message-stream-finish.ts
+function handleUIMessageStreamFinish({
+  newMessageId,
+  originalMessages = [],
+  onFinish,
+  stream
+}) {
+  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 state = createStreamingUIMessageState({
+    lastMessage: structuredClone(lastMessage),
+    newMessageId: messageId
+  });
+  const runUpdateMessageJob = async (job) => {
+    await job({ state, write: () => {
+    } });
+  };
+  return processUIMessageStream({
+    stream,
+    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
+          ]
+        });
+      }
+    })
+  );
+}
+
 // src/ui-message-stream/create-ui-message-stream.ts
 function createUIMessageStream({
   execute,
-  onError = () => "An error occurred."
+  onError = () => "An error occurred.",
   // mask error messages for safety by default
+  originalMessages,
+  onFinish
 }) {
   let controller;
   const ongoingStreamPromises = [];
@@ -2367,25 +2378,27 @@ function createUIMessageStream({
   }
   try {
     const result = execute({
-      write(part) {
-        safeEnqueue(part);
-      },
-      merge(streamArg) {
-        ongoingStreamPromises.push(
-          (async () => {
-            const reader = streamArg.getReader();
-            while (true) {
-              const { done, value } = await reader.read();
-              if (done)
-                break;
-              safeEnqueue(value);
-            }
-          })().catch((error) => {
-            safeEnqueue({ type: "error", errorText: onError(error) });
-          })
-        );
-      },
-      onError
+      writer: {
+        write(part) {
+          safeEnqueue(part);
+        },
+        merge(streamArg) {
+          ongoingStreamPromises.push(
+            (async () => {
+              const reader = streamArg.getReader();
+              while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                  break;
+                safeEnqueue(value);
+              }
+            })().catch((error) => {
+              safeEnqueue({ type: "error", errorText: onError(error) });
+            })
+          );
+        },
+        onError
+      }
     });
     if (result) {
       ongoingStreamPromises.push(
@@ -2409,7 +2422,12 @@ function createUIMessageStream({
     } catch (error) {
     }
   });
-  return stream;
+  return handleUIMessageStreamFinish({
+    stream,
+    newMessageId: "",
+    originalMessages,
+    onFinish
+  });
 }
 
 // src/ui-message-stream/ui-message-stream-headers.ts
@@ -2474,6 +2492,32 @@ function pipeUIMessageStreamToResponse({
   });
 }
 
+// src/util/cosine-similarity.ts
+function cosineSimilarity(vector1, vector2) {
+  if (vector1.length !== vector2.length) {
+    throw new InvalidArgumentError({
+      parameter: "vector1,vector2",
+      value: { vector1Length: vector1.length, vector2Length: vector2.length },
+      message: `Vectors must have the same length`
+    });
+  }
+  const n = vector1.length;
+  if (n === 0) {
+    return 0;
+  }
+  let magnitudeSquared1 = 0;
+  let magnitudeSquared2 = 0;
+  let dotProduct = 0;
+  for (let i = 0; i < n; i++) {
+    const value1 = vector1[i];
+    const value2 = vector2[i];
+    magnitudeSquared1 += value1 * value1;
+    magnitudeSquared2 += value2 * value2;
+    dotProduct += value1 * value2;
+  }
+  return magnitudeSquared1 === 0 || magnitudeSquared2 === 0 ? 0 : dotProduct / (Math.sqrt(magnitudeSquared1) * Math.sqrt(magnitudeSquared2));
+}
+
 // src/util/data-url.ts
 function getTextFromDataUrl(dataUrl) {
   const [header, base64Content] = dataUrl.split(",");
@@ -2523,31 +2567,37 @@ function isDeepEqualData(obj1, obj2) {
   return true;
 }
 
-// src/util/cosine-similarity.ts
-function cosineSimilarity(vector1, vector2) {
-  if (vector1.length !== vector2.length) {
-    throw new InvalidArgumentError({
-      parameter: "vector1,vector2",
-      value: { vector1Length: vector1.length, vector2Length: vector2.length },
-      message: `Vectors must have the same length`
-    });
+// src/util/serial-job-executor.ts
+var SerialJobExecutor = class {
+  constructor() {
+    this.queue = [];
+    this.isProcessing = false;
   }
-  const n = vector1.length;
-  if (n === 0) {
-    return 0;
+  async processQueue() {
+    if (this.isProcessing) {
+      return;
+    }
+    this.isProcessing = true;
+    while (this.queue.length > 0) {
+      await this.queue[0]();
+      this.queue.shift();
+    }
+    this.isProcessing = false;
   }
-  let magnitudeSquared1 = 0;
-  let magnitudeSquared2 = 0;
-  let dotProduct = 0;
-  for (let i = 0; i < n; i++) {
-    const value1 = vector1[i];
-    const value2 = vector2[i];
-    magnitudeSquared1 += value1 * value1;
-    magnitudeSquared2 += value2 * value2;
-    dotProduct += value1 * value2;
+  async run(job) {
+    return new Promise((resolve, reject) => {
+      this.queue.push(async () => {
+        try {
+          await job();
+          resolve();
+        } catch (error) {
+          reject(error);
+        }
+      });
+      void this.processQueue();
+    });
   }
-  return magnitudeSquared1 === 0 || magnitudeSquared2 === 0 ? 0 : dotProduct / (Math.sqrt(magnitudeSquared1) * Math.sqrt(magnitudeSquared2));
-}
+};
 
 // src/util/simulate-readable-stream.ts
 var import_provider_utils7 = require("@ai-sdk/provider-utils");
@@ -3541,6 +3591,15 @@ function convertToLanguageModelV2DataContent(content) {
   }
   return { data: content, mediaType: void 0 };
 }
+function convertDataContentToBase64String(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (content instanceof ArrayBuffer) {
+    return (0, import_provider_utils11.convertUint8ArrayToBase64)(new Uint8Array(content));
+  }
+  return (0, import_provider_utils11.convertUint8ArrayToBase64)(content);
+}
 function convertDataContentToUint8Array(content) {
   if (content instanceof Uint8Array) {
     return content;
@@ -4058,6 +4117,21 @@ async function standardizePrompt(prompt) {
   };
 }
 
+// core/telemetry/stringify-for-telemetry.ts
+function stringifyForTelemetry(prompt) {
+  return JSON.stringify(
+    prompt.map((message) => ({
+      ...message,
+      content: typeof message.content === "string" ? message.content : message.content.map(
+        (part) => part.type === "file" ? {
+          ...part,
+          data: part.data instanceof Uint8Array ? convertDataContentToBase64String(part.data) : part.data
+        } : part
+      )
+    }))
+  );
+}
+
 // core/generate-object/output-strategy.ts
 var import_provider20 = require("@ai-sdk/provider");
 var import_provider_utils14 = require("@ai-sdk/provider-utils");
@@ -4550,7 +4624,7 @@ async function generateObject(options) {
               }),
               ...baseTelemetryAttributes,
               "ai.prompt.messages": {
-                input: () => JSON.stringify(promptMessages)
+                input: () => stringifyForTelemetry(promptMessages)
               },
               // standardized gen-ai llm span attributes:
               "gen_ai.system": model.provider,
@@ -5052,7 +5126,7 @@ var DefaultStreamObjectResult = class {
                 }),
                 ...baseTelemetryAttributes,
                 "ai.prompt.messages": {
-                  input: () => JSON.stringify(callOptions.prompt)
+                  input: () => stringifyForTelemetry(callOptions.prompt)
                 },
                 // standardized gen-ai llm span attributes:
                 "gen_ai.system": model.provider,
@@ -5465,6 +5539,11 @@ var DefaultSpeechResult = class {
 // core/generate-text/generate-text.ts
 var import_provider_utils19 = require("@ai-sdk/provider-utils");
 
+// src/util/as-array.ts
+function asArray(value) {
+  return value === void 0 ? [] : Array.isArray(value) ? value : [value];
+}
+
 // core/prompt/prepare-tools-and-tool-choice.ts
 var import_provider_utils17 = require("@ai-sdk/provider-utils");
 
@@ -5682,8 +5761,8 @@ var DefaultStepResult = class {
 };
 
 // core/generate-text/stop-condition.ts
-function maxSteps(maxSteps2) {
-  return ({ steps }) => steps.length >= maxSteps2;
+function stepCountIs(stepCount) {
+  return ({ steps }) => steps.length === stepCount;
 }
 function hasToolCall(toolName) {
   return ({ steps }) => {
@@ -5693,6 +5772,12 @@ function hasToolCall(toolName) {
     )) != null ? _c : false;
   };
 }
+async function isStopConditionMet({
+  stopConditions,
+  steps
+}) {
+  return (await Promise.all(stopConditions.map((condition) => condition({ steps })))).some((result) => result);
+}
 
 // core/generate-text/to-response-messages.ts
 function toResponseMessages({
@@ -5767,12 +5852,14 @@ async function generateText({
   maxRetries: maxRetriesArg,
   abortSignal,
   headers,
-  continueUntil = maxSteps(1),
+  stopWhen = stepCountIs(1),
   experimental_output: output,
   experimental_telemetry: telemetry,
   providerOptions,
-  experimental_activeTools: activeTools,
-  experimental_prepareStep: prepareStep,
+  experimental_activeTools,
+  activeTools = experimental_activeTools,
+  experimental_prepareStep,
+  prepareStep = experimental_prepareStep,
   experimental_repairToolCall: repairToolCall,
   _internal: {
     generateId: generateId3 = originalGenerateId3,
@@ -5781,6 +5868,7 @@ async function generateText({
   onStepFinish,
   ...settings
 }) {
+  const stopConditions = asArray(stopWhen);
   const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
   const callSettings = prepareCallSettings(settings);
   const baseTelemetryAttributes = getBaseTelemetryAttributes({
@@ -5844,7 +5932,7 @@ async function generateText({
         const { toolChoice: stepToolChoice, tools: stepTools } = prepareToolsAndToolChoice({
           tools,
           toolChoice: (_b = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _b : toolChoice,
-          activeTools: (_c = prepareStepResult == null ? void 0 : prepareStepResult.experimental_activeTools) != null ? _c : activeTools
+          activeTools: (_c = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _c : activeTools
         });
         currentModelResponse = await retry(
           () => {
@@ -5864,7 +5952,7 @@ async function generateText({
                   "ai.model.id": stepModel.modelId,
                   // prompt:
                   "ai.prompt.messages": {
-                    input: () => JSON.stringify(promptMessages)
+                    input: () => stringifyForTelemetry(promptMessages)
                   },
                   "ai.prompt.tools": {
                     // convert the language model level tools:
@@ -5989,8 +6077,8 @@ async function generateText({
       } 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 })
+        currentToolResults.length === currentToolCalls.length && // continue until a stop condition is met:
+        !await isStopConditionMet({ stopConditions, steps })
       );
       span.setAttributes(
         selectTelemetryAttributes({
@@ -6336,11 +6424,6 @@ function smoothStream({
 // core/generate-text/stream-text.ts
 var import_provider_utils23 = require("@ai-sdk/provider-utils");
 
-// src/util/as-array.ts
-function asArray(value) {
-  return value === void 0 ? [] : Array.isArray(value) ? value : [value];
-}
-
 // core/generate-text/run-tools-transformation.ts
 var import_provider_utils22 = require("@ai-sdk/provider-utils");
 function runToolsTransformation({
@@ -6551,13 +6634,15 @@ function streamText({
   maxRetries,
   abortSignal,
   headers,
-  continueUntil = maxSteps(1),
+  stopWhen = stepCountIs(1),
   experimental_output: output,
   experimental_telemetry: telemetry,
+  prepareStep,
   providerOptions,
   experimental_toolCallStreaming = false,
   toolCallStreaming = experimental_toolCallStreaming,
-  experimental_activeTools: activeTools,
+  experimental_activeTools,
+  activeTools = experimental_activeTools,
   experimental_repairToolCall: repairToolCall,
   experimental_transform: transform,
   onChunk,
@@ -6587,9 +6672,10 @@ function streamText({
     transforms: asArray(transform),
     activeTools,
     repairToolCall,
-    continueUntil,
+    stopConditions: asArray(stopWhen),
     output,
     providerOptions,
+    prepareStep,
     onChunk,
     onError,
     onFinish,
@@ -6664,9 +6750,10 @@ var DefaultStreamTextResult = class {
     transforms,
     activeTools,
     repairToolCall,
-    continueUntil,
+    stopConditions,
     output,
     providerOptions,
+    prepareStep,
     now: now2,
     currentDate,
     generateId: generateId3,
@@ -6885,6 +6972,7 @@ var DefaultStreamTextResult = class {
           responseMessages,
           usage
         }) {
+          var _a17, _b, _c;
           stepFinish = new DelayedPromise();
           const initialPrompt = await standardizePrompt({
             system,
@@ -6895,6 +6983,11 @@ var DefaultStreamTextResult = class {
             ...initialPrompt.messages,
             ...responseMessages
           ];
+          const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
+            model,
+            steps: recordedSteps,
+            stepNumber: recordedSteps.length
+          }));
           const promptMessages = await convertToLanguageModelPrompt({
             prompt: {
               system: initialPrompt.system,
@@ -6902,9 +6995,12 @@ var DefaultStreamTextResult = class {
             },
             supportedUrls: await model.supportedUrls
           });
-          const toolsAndToolChoice = {
-            ...prepareToolsAndToolChoice({ tools, toolChoice, activeTools })
-          };
+          const stepModel = (_a17 = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _a17 : model;
+          const { toolChoice: stepToolChoice, tools: stepTools } = prepareToolsAndToolChoice({
+            tools,
+            toolChoice: (_b = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _b : toolChoice,
+            activeTools: (_c = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _c : activeTools
+          });
           const {
             result: { stream: stream2, response, request },
             doStreamSpan,
@@ -6920,24 +7016,23 @@ var DefaultStreamTextResult = class {
                     telemetry
                   }),
                   ...baseTelemetryAttributes,
+                  // model:
+                  "ai.model.provider": stepModel.provider,
+                  "ai.model.id": stepModel.modelId,
+                  // prompt:
                   "ai.prompt.messages": {
-                    input: () => JSON.stringify(promptMessages)
+                    input: () => stringifyForTelemetry(promptMessages)
                   },
                   "ai.prompt.tools": {
                     // convert the language model level tools:
-                    input: () => {
-                      var _a17;
-                      return (_a17 = toolsAndToolChoice.tools) == null ? void 0 : _a17.map(
-                        (tool2) => JSON.stringify(tool2)
-                      );
-                    }
+                    input: () => stepTools == null ? void 0 : stepTools.map((tool2) => JSON.stringify(tool2))
                   },
                   "ai.prompt.toolChoice": {
-                    input: () => toolsAndToolChoice.toolChoice != null ? JSON.stringify(toolsAndToolChoice.toolChoice) : void 0
+                    input: () => stepToolChoice != null ? JSON.stringify(stepToolChoice) : void 0
                   },
                   // standardized gen-ai llm span attributes:
-                  "gen_ai.system": model.provider,
-                  "gen_ai.request.model": model.modelId,
+                  "gen_ai.system": stepModel.provider,
+                  "gen_ai.request.model": stepModel.modelId,
                   "gen_ai.request.frequency_penalty": callSettings.frequencyPenalty,
                   "gen_ai.request.max_tokens": callSettings.maxOutputTokens,
                   "gen_ai.request.presence_penalty": callSettings.presencePenalty,
@@ -6954,9 +7049,10 @@ var DefaultStreamTextResult = class {
                   startTimestampMs: now2(),
                   // get before the call
                   doStreamSpan: doStreamSpan2,
-                  result: await model.doStream({
+                  result: await stepModel.doStream({
                     ...callSettings,
-                    ...toolsAndToolChoice,
+                    tools: stepTools,
+                    toolChoice: stepToolChoice,
                     responseFormat: output == null ? void 0 : output.responseFormat,
                     prompt: promptMessages,
                     providerOptions,
@@ -7009,7 +7105,7 @@ var DefaultStreamTextResult = class {
             streamWithToolResults.pipeThrough(
               new TransformStream({
                 async transform(chunk, controller) {
-                  var _a17, _b, _c, _d;
+                  var _a18, _b2, _c2, _d;
                   if (chunk.type === "stream-start") {
                     warnings = chunk.warnings;
                     return;
@@ -7072,9 +7168,9 @@ var DefaultStreamTextResult = class {
                     }
                     case "response-metadata": {
                       stepResponse = {
-                        id: (_a17 = chunk.id) != null ? _a17 : stepResponse.id,
-                        timestamp: (_b = chunk.timestamp) != null ? _b : stepResponse.timestamp,
-                        modelId: (_c = chunk.modelId) != null ? _c : stepResponse.modelId
+                        id: (_a18 = chunk.id) != null ? _a18 : stepResponse.id,
+                        timestamp: (_b2 = chunk.timestamp) != null ? _b2 : stepResponse.timestamp,
+                        modelId: (_c2 = chunk.modelId) != null ? _c2 : stepResponse.modelId
                       };
                       break;
                     }
@@ -7163,7 +7259,11 @@ var DefaultStreamTextResult = class {
                   const combinedUsage = addLanguageModelUsage(usage, stepUsage);
                   await stepFinish.promise;
                   if (stepToolCalls.length > 0 && // all current tool calls have results:
-                  stepToolResults.length === stepToolCalls.length && !await continueUntil({ steps: recordedSteps })) {
+                  stepToolResults.length === stepToolCalls.length && // continue until a stop condition is met:
+                  !await isStopConditionMet({
+                    stopConditions,
+                    steps: recordedSteps
+                  })) {
                     responseMessages.push(
                       ...toResponseMessages({
                         content: stepContent,
@@ -7332,14 +7432,14 @@ var DefaultStreamTextResult = class {
     messageMetadata,
     sendReasoning = false,
     sendSources = false,
-    experimental_sendStart = true,
-    experimental_sendFinish = true,
+    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 messageId = isContinuation ? lastMessage.id : newMessageId != null ? newMessageId : this.generateId();
     const baseStream = this.fullStream.pipeThrough(
       new TransformStream({
         transform: async (part, controller) => {
@@ -7445,7 +7545,7 @@ var DefaultStreamTextResult = class {
               break;
             }
             case "start": {
-              if (experimental_sendStart) {
+              if (sendStart) {
                 const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "start",
@@ -7456,7 +7556,7 @@ var DefaultStreamTextResult = class {
               break;
             }
             case "finish": {
-              if (experimental_sendFinish) {
+              if (sendFinish) {
                 const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "finish",
@@ -7473,38 +7573,12 @@ var DefaultStreamTextResult = class {
         }
       })
     );
-    if (onFinish == null) {
-      return baseStream;
-    }
-    const state = createStreamingUIMessageState({
-      lastMessage,
-      newMessageId: messageId != null ? messageId : this.generateId()
-    });
-    const runUpdateMessageJob = async (job) => {
-      await job({ state, write: () => {
-      } });
-    };
-    return processUIMessageStream({
+    return handleUIMessageStreamFinish({
       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
-            ]
-          });
-        }
-      })
-    );
+      newMessageId: messageId,
+      originalMessages,
+      onFinish
+    });
   }
   pipeUIMessageStreamToResponse(response, {
     newMessageId,
@@ -7513,8 +7587,8 @@ var DefaultStreamTextResult = class {
     messageMetadata,
     sendReasoning,
     sendSources,
-    experimental_sendFinish,
-    experimental_sendStart,
+    sendFinish,
+    sendStart,
     onError,
     ...init
   } = {}) {
@@ -7527,8 +7601,8 @@ var DefaultStreamTextResult = class {
         messageMetadata,
         sendReasoning,
         sendSources,
-        experimental_sendFinish,
-        experimental_sendStart,
+        sendFinish,
+        sendStart,
         onError
       }),
       ...init
@@ -7548,8 +7622,8 @@ var DefaultStreamTextResult = class {
     messageMetadata,
     sendReasoning,
     sendSources,
-    experimental_sendFinish,
-    experimental_sendStart,
+    sendFinish,
+    sendStart,
     onError,
     ...init
   } = {}) {
@@ -7561,8 +7635,8 @@ var DefaultStreamTextResult = class {
         messageMetadata,
         sendReasoning,
         sendSources,
-        experimental_sendFinish,
-        experimental_sendStart,
+        sendFinish,
+        sendStart,
         onError
       }),
       ...init
@@ -8610,6 +8684,7 @@ var DefaultTranscriptionResult = class {
   NoSuchToolError,
   Output,
   RetryError,
+  SerialJobExecutor,
   TextStreamChatTransport,
   ToolCallRepairError,
   ToolExecutionError,
@@ -8635,7 +8710,7 @@ var DefaultTranscriptionResult = class {
   createUIMessageStream,
   createUIMessageStreamResponse,
   customProvider,
-  defaultChatStore,
+  defaultChatStoreOptions,
   defaultSettingsMiddleware,
   embed,
   embedMany,
@@ -8656,7 +8731,6 @@ var DefaultTranscriptionResult = class {
   isAssistantMessageWithCompletedToolCalls,
   isDeepEqualData,
   jsonSchema,
-  maxSteps,
   modelMessageSchema,
   parsePartialJson,
   pipeTextStreamToResponse,
@@ -8665,6 +8739,7 @@ var DefaultTranscriptionResult = class {
   simulateReadableStream,
   simulateStreamingMiddleware,
   smoothStream,
+  stepCountIs,
   streamObject,
   streamText,
   systemModelMessageSchema,