ai 5.0.0-alpha.4 → 5.0.0-alpha.6

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,
@@ -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,38 +1635,6 @@ 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,
@@ -1723,18 +1692,14 @@ var ChatStore = class {
     transport,
     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 = maxSteps;
@@ -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 = 1,
+  dataPartSchemas,
   chats
 }) {
-  return new ChatStore({
+  return () => ({
     transport: new DefaultChatTransport({
       api,
       fetch: fetch2,
@@ -2346,11 +2311,57 @@ function defaultChatStore({
   });
 }
 
+// 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");
@@ -4451,11 +4525,17 @@ function validateObjectGenerationInput({
   }
 }
 
+// core/prompt/resolve-language-model.ts
+var import_gateway = require("@ai-sdk/gateway");
+function resolveLanguageModel(model) {
+  return typeof model === "string" ? import_gateway.gateway.languageModel(model) : model;
+}
+
 // core/generate-object/generate-object.ts
 var originalGenerateId = (0, import_provider_utils15.createIdGenerator)({ prefix: "aiobj", size: 24 });
 async function generateObject(options) {
   const {
-    model,
+    model: modelArg,
     output = "object",
     system,
     prompt,
@@ -4472,6 +4552,7 @@ async function generateObject(options) {
     } = {},
     ...settings
   } = options;
+  const model = resolveLanguageModel(modelArg);
   const enumValues = "enum" in options ? options.enum : void 0;
   const {
     schema: inputSchema,
@@ -4550,7 +4631,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,
@@ -4931,7 +5012,7 @@ function streamObject(options) {
 }
 var DefaultStreamObjectResult = class {
   constructor({
-    model,
+    model: modelArg,
     headers,
     telemetry,
     settings,
@@ -4956,6 +5037,7 @@ var DefaultStreamObjectResult = class {
     this._warnings = new DelayedPromise();
     this._request = new DelayedPromise();
     this._response = new DelayedPromise();
+    const model = resolveLanguageModel(modelArg);
     const { maxRetries, retry } = prepareRetries({
       maxRetries: maxRetriesArg
     });
@@ -5052,7 +5134,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,
@@ -5769,7 +5851,7 @@ var originalGenerateId3 = (0, import_provider_utils19.createIdGenerator)({
   size: 24
 });
 async function generateText({
-  model,
+  model: modelArg,
   tools,
   toolChoice,
   system,
@@ -5794,6 +5876,7 @@ async function generateText({
   onStepFinish,
   ...settings
 }) {
+  const model = resolveLanguageModel(modelArg);
   const stopConditions = asArray(stopWhen);
   const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
   const callSettings = prepareCallSettings(settings);
@@ -5830,7 +5913,7 @@ async function generateText({
     }),
     tracer,
     fn: async (span) => {
-      var _a17, _b, _c, _d;
+      var _a17, _b, _c, _d, _e;
       const callSettings2 = prepareCallSettings(settings);
       let currentModelResponse;
       let currentToolCalls = [];
@@ -5849,16 +5932,18 @@ async function generateText({
         }));
         const promptMessages = await convertToLanguageModelPrompt({
           prompt: {
-            system: initialPrompt.system,
+            system: (_a17 = prepareStepResult == null ? void 0 : prepareStepResult.system) != null ? _a17 : initialPrompt.system,
             messages: stepInputMessages
           },
           supportedUrls: await model.supportedUrls
         });
-        const stepModel = (_a17 = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _a17 : model;
+        const stepModel = resolveLanguageModel(
+          (_b = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _b : 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
+          toolChoice: (_c = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _c : toolChoice,
+          activeTools: (_d = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _d : activeTools
         });
         currentModelResponse = await retry(
           () => {
@@ -5878,7 +5963,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:
@@ -5901,7 +5986,7 @@ async function generateText({
               }),
               tracer,
               fn: async (span2) => {
-                var _a19, _b2, _c2, _d2, _e, _f, _g, _h;
+                var _a19, _b2, _c2, _d2, _e2, _f, _g, _h;
                 const result = await stepModel.doGenerate({
                   ...callSettings2,
                   tools: stepTools,
@@ -5915,7 +6000,7 @@ async function generateText({
                 const responseData = {
                   id: (_b2 = (_a19 = result.response) == null ? void 0 : _a19.id) != null ? _b2 : generateId3(),
                   timestamp: (_d2 = (_c2 = result.response) == null ? void 0 : _c2.timestamp) != null ? _d2 : currentDate(),
-                  modelId: (_f = (_e = result.response) == null ? void 0 : _e.modelId) != null ? _f : stepModel.modelId,
+                  modelId: (_f = (_e2 = result.response) == null ? void 0 : _e2.modelId) != null ? _f : stepModel.modelId,
                   headers: (_g = result.response) == null ? void 0 : _g.headers,
                   body: (_h = result.response) == null ? void 0 : _h.body
                 };
@@ -5991,7 +6076,7 @@ async function generateText({
           usage: currentModelResponse.usage,
           warnings: currentModelResponse.warnings,
           providerMetadata: currentModelResponse.providerMetadata,
-          request: (_d = currentModelResponse.request) != null ? _d : {},
+          request: (_e = currentModelResponse.request) != null ? _e : {},
           response: {
             ...currentModelResponse.response,
             // deep clone msgs to avoid mutating past messages in multi-step:
@@ -6583,7 +6668,7 @@ function streamText({
   ...settings
 }) {
   return new DefaultStreamTextResult({
-    model,
+    model: resolveLanguageModel(model),
     telemetry,
     headers,
     settings,
@@ -6898,7 +6983,7 @@ var DefaultStreamTextResult = class {
           responseMessages,
           usage
         }) {
-          var _a17, _b, _c;
+          var _a17, _b, _c, _d;
           stepFinish = new DelayedPromise();
           const initialPrompt = await standardizePrompt({
             system,
@@ -6916,16 +7001,18 @@ var DefaultStreamTextResult = class {
           }));
           const promptMessages = await convertToLanguageModelPrompt({
             prompt: {
-              system: initialPrompt.system,
+              system: (_a17 = prepareStepResult == null ? void 0 : prepareStepResult.system) != null ? _a17 : initialPrompt.system,
               messages: stepInputMessages
             },
             supportedUrls: await model.supportedUrls
           });
-          const stepModel = (_a17 = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _a17 : model;
+          const stepModel = resolveLanguageModel(
+            (_b = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _b : 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
+            toolChoice: (_c = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _c : toolChoice,
+            activeTools: (_d = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _d : activeTools
           });
           const {
             result: { stream: stream2, response, request },
@@ -6947,7 +7034,7 @@ var DefaultStreamTextResult = class {
                   "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:
@@ -7031,7 +7118,7 @@ var DefaultStreamTextResult = class {
             streamWithToolResults.pipeThrough(
               new TransformStream({
                 async transform(chunk, controller) {
-                  var _a18, _b2, _c2, _d;
+                  var _a18, _b2, _c2, _d2;
                   if (chunk.type === "stream-start") {
                     warnings = chunk.warnings;
                     return;
@@ -7108,7 +7195,7 @@ var DefaultStreamTextResult = class {
                       doStreamSpan.addEvent("ai.stream.finish");
                       doStreamSpan.setAttributes({
                         "ai.response.msToFinish": msToFinish,
-                        "ai.response.avgOutputTokensPerSecond": 1e3 * ((_d = stepUsage.outputTokens) != null ? _d : 0) / msToFinish
+                        "ai.response.avgOutputTokensPerSecond": 1e3 * ((_d2 = stepUsage.outputTokens) != null ? _d2 : 0) / msToFinish
                       });
                       break;
                     }
@@ -7358,14 +7445,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) => {
@@ -7471,7 +7558,7 @@ var DefaultStreamTextResult = class {
               break;
             }
             case "start": {
-              if (experimental_sendStart) {
+              if (sendStart) {
                 const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "start",
@@ -7482,7 +7569,7 @@ var DefaultStreamTextResult = class {
               break;
             }
             case "finish": {
-              if (experimental_sendFinish) {
+              if (sendFinish) {
                 const metadata = messageMetadata == null ? void 0 : messageMetadata({ part });
                 controller.enqueue({
                   type: "finish",
@@ -7499,38 +7586,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,
@@ -7539,8 +7600,8 @@ var DefaultStreamTextResult = class {
     messageMetadata,
     sendReasoning,
     sendSources,
-    experimental_sendFinish,
-    experimental_sendStart,
+    sendFinish,
+    sendStart,
     onError,
     ...init
   } = {}) {
@@ -7553,8 +7614,8 @@ var DefaultStreamTextResult = class {
         messageMetadata,
         sendReasoning,
         sendSources,
-        experimental_sendFinish,
-        experimental_sendStart,
+        sendFinish,
+        sendStart,
         onError
       }),
       ...init
@@ -7574,8 +7635,8 @@ var DefaultStreamTextResult = class {
     messageMetadata,
     sendReasoning,
     sendSources,
-    experimental_sendFinish,
-    experimental_sendStart,
+    sendFinish,
+    sendStart,
     onError,
     ...init
   } = {}) {
@@ -7587,8 +7648,8 @@ var DefaultStreamTextResult = class {
         messageMetadata,
         sendReasoning,
         sendSources,
-        experimental_sendFinish,
-        experimental_sendStart,
+        sendFinish,
+        sendStart,
         onError
       }),
       ...init
@@ -8636,6 +8697,7 @@ var DefaultTranscriptionResult = class {
   NoSuchToolError,
   Output,
   RetryError,
+  SerialJobExecutor,
   TextStreamChatTransport,
   ToolCallRepairError,
   ToolExecutionError,
@@ -8661,7 +8723,7 @@ var DefaultTranscriptionResult = class {
   createUIMessageStream,
   createUIMessageStreamResponse,
   customProvider,
-  defaultChatStore,
+  defaultChatStoreOptions,
   defaultSettingsMiddleware,
   embed,
   embedMany,