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

package/dist/index.mjs

@@ -962,12 +962,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",
@@ -1347,7 +1347,7 @@ async function consumeUIMessageStream({
   messageMetadataSchema
 }) {
   const state = createStreamingUIMessageState({
-    lastMessage,
+    lastMessage: lastMessage ? structuredClone(lastMessage) : void 0,
     newMessageId: generateId3()
   });
   const runUpdateMessageJob = async (job) => {
@@ -1550,54 +1550,22 @@ import {
   generateId as generateIdFunc
 } from "@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) {
@@ -1637,23 +1605,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 : generateIdFunc;
@@ -1664,11 +1628,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());
@@ -1677,28 +1637,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) {
@@ -1709,11 +1669,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");
@@ -1721,7 +1681,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,
@@ -1733,14 +1694,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",
@@ -1757,15 +1718,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,
@@ -1781,11 +1747,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,
@@ -1799,22 +1762,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
         });
@@ -1823,7 +1787,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) {
@@ -1836,15 +1800,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,
@@ -1852,26 +1815,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,
@@ -1883,15 +1845,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
               });
             }
           })
@@ -1901,8 +1867,8 @@ var ChatStore = class {
         stream: processUIMessageStream({
           stream,
           onToolCall,
-          messageMetadataSchema: self.messageMetadataSchema,
-          dataPartSchemas: self.dataPartSchemas,
+          messageMetadataSchema: this.messageMetadataSchema,
+          dataPartSchemas: this.dataPartSchemas,
           runUpdateMessageJob
         }),
         onError: (error) => {
@@ -1921,24 +1887,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
       });
     }
   }
@@ -2230,24 +2194,24 @@ function convertToModelMessages(messages, options) {
 }
 var convertToCoreMessages = convertToModelMessages;
 
-// src/ui/default-chat-store.ts
+// src/ui/default-chat-store-options.ts
 import {
   generateId as generateIdFunc2
 } from "@ai-sdk/provider-utils";
-function defaultChatStore({
-  api,
+function defaultChatStoreOptions({
+  api = "/api/chat",
   fetch: fetch2,
   credentials,
   headers,
   body,
   prepareRequestBody,
   generateId: generateId3 = generateIdFunc2,
-  dataPartSchemas,
   messageMetadataSchema,
-  maxSteps: maxSteps2 = 1,
+  maxSteps = 1,
+  dataPartSchemas,
   chats
 }) {
-  return new ChatStore({
+  return () => ({
     transport: new DefaultChatTransport({
       api,
       fetch: fetch2,
@@ -2259,16 +2223,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 = [];
@@ -2285,25 +2295,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(
@@ -2327,7 +2339,12 @@ function createUIMessageStream({
     } catch (error) {
     }
   });
-  return stream;
+  return handleUIMessageStreamFinish({
+    stream,
+    newMessageId: "",
+    originalMessages,
+    onFinish
+  });
 }
 
 // src/ui-message-stream/ui-message-stream-headers.ts
@@ -2392,6 +2409,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(",");
@@ -2441,31 +2484,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
 import { delay as delayFunction } from "@ai-sdk/provider-utils";
@@ -3471,6 +3520,15 @@ function convertToLanguageModelV2DataContent(content) {
   }
   return { data: content, mediaType: void 0 };
 }
+function convertDataContentToBase64String(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (content instanceof ArrayBuffer) {
+    return convertUint8ArrayToBase642(new Uint8Array(content));
+  }
+  return convertUint8ArrayToBase642(content);
+}
 function convertDataContentToUint8Array(content) {
   if (content instanceof Uint8Array) {
     return content;
@@ -3988,6 +4046,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
 import {
   isJSONArray,
@@ -4488,7 +4561,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,
@@ -4990,7 +5063,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,
@@ -5403,6 +5476,11 @@ var DefaultSpeechResult = class {
 // core/generate-text/generate-text.ts
 import { createIdGenerator as createIdGenerator3 } from "@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
 import { asSchema as asSchema2 } from "@ai-sdk/provider-utils";
 
@@ -5624,8 +5702,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 }) => {
@@ -5635,6 +5713,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({
@@ -5709,12 +5793,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,
@@ -5723,6 +5809,7 @@ async function generateText({
   onStepFinish,
   ...settings
 }) {
+  const stopConditions = asArray(stopWhen);
   const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
   const callSettings = prepareCallSettings(settings);
   const baseTelemetryAttributes = getBaseTelemetryAttributes({
@@ -5786,7 +5873,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(
           () => {
@@ -5806,7 +5893,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:
@@ -5931,8 +6018,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({
@@ -6282,11 +6369,6 @@ function smoothStream({
 // core/generate-text/stream-text.ts
 import { createIdGenerator as createIdGenerator4 } from "@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
 import { generateId } from "@ai-sdk/provider-utils";
 function runToolsTransformation({
@@ -6497,13 +6579,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,
@@ -6533,9 +6617,10 @@ function streamText({
     transforms: asArray(transform),
     activeTools,
     repairToolCall,
-    continueUntil,
+    stopConditions: asArray(stopWhen),
     output,
     providerOptions,
+    prepareStep,
     onChunk,
     onError,
     onFinish,
@@ -6610,9 +6695,10 @@ var DefaultStreamTextResult = class {
     transforms,
     activeTools,
     repairToolCall,
-    continueUntil,
+    stopConditions,
     output,
     providerOptions,
+    prepareStep,
     now: now2,
     currentDate,
     generateId: generateId3,
@@ -6831,6 +6917,7 @@ var DefaultStreamTextResult = class {
           responseMessages,
           usage
         }) {
+          var _a17, _b, _c;
           stepFinish = new DelayedPromise();
           const initialPrompt = await standardizePrompt({
             system,
@@ -6841,6 +6928,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,
@@ -6848,9 +6940,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,
@@ -6866,24 +6961,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,
@@ -6900,9 +6994,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,
@@ -6955,7 +7050,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;
@@ -7018,9 +7113,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;
                     }
@@ -7109,7 +7204,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,
@@ -7278,14 +7377,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) => {
@@ -7391,7 +7490,7 @@ var DefaultStreamTextResult = class {
               break;
             }
             case "start": {
-              if (experimental_sendStart) {
+              if (sendStart) {
                 const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "start",
@@ -7402,7 +7501,7 @@ var DefaultStreamTextResult = class {
               break;
             }
             case "finish": {
-              if (experimental_sendFinish) {
+              if (sendFinish) {
                 const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "finish",
@@ -7419,38 +7518,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,
@@ -7459,8 +7532,8 @@ var DefaultStreamTextResult = class {
     messageMetadata,
     sendReasoning,
     sendSources,
-    experimental_sendFinish,
-    experimental_sendStart,
+    sendFinish,
+    sendStart,
     onError,
     ...init
   } = {}) {
@@ -7473,8 +7546,8 @@ var DefaultStreamTextResult = class {
         messageMetadata,
         sendReasoning,
         sendSources,
-        experimental_sendFinish,
-        experimental_sendStart,
+        sendFinish,
+        sendStart,
         onError
       }),
       ...init
@@ -7494,8 +7567,8 @@ var DefaultStreamTextResult = class {
     messageMetadata,
     sendReasoning,
     sendSources,
-    experimental_sendFinish,
-    experimental_sendStart,
+    sendFinish,
+    sendStart,
     onError,
     ...init
   } = {}) {
@@ -7507,8 +7580,8 @@ var DefaultStreamTextResult = class {
         messageMetadata,
         sendReasoning,
         sendSources,
-        experimental_sendFinish,
-        experimental_sendStart,
+        sendFinish,
+        sendStart,
         onError
       }),
       ...init
@@ -8555,6 +8628,7 @@ export {
   NoSuchToolError,
   output_exports as Output,
   RetryError,
+  SerialJobExecutor,
   TextStreamChatTransport,
   ToolCallRepairError,
   ToolExecutionError,
@@ -8580,7 +8654,7 @@ export {
   createUIMessageStream,
   createUIMessageStreamResponse,
   customProvider,
-  defaultChatStore,
+  defaultChatStoreOptions,
   defaultSettingsMiddleware,
   embed,
   embedMany,
@@ -8601,7 +8675,6 @@ export {
   isAssistantMessageWithCompletedToolCalls,
   isDeepEqualData,
   jsonSchema2 as jsonSchema,
-  maxSteps,
   modelMessageSchema,
   parsePartialJson,
   pipeTextStreamToResponse,
@@ -8610,6 +8683,7 @@ export {
   simulateReadableStream,
   simulateStreamingMiddleware,
   smoothStream,
+  stepCountIs,
   streamObject,
   streamText,
   systemModelMessageSchema,