ai 5.0.272 → 5.0.274

package/dist/index.js

@@ -817,7 +817,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "5.0.272" : "0.0.0-test";
+var VERSION = true ? "5.0.274" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -3818,7 +3818,8 @@ function processUIMessageStream({
   runUpdateMessageJob,
   onError,
   onToolCall,
-  onData
+  onData,
+  resetStateOnMessageIdChange = false
 }) {
   return stream.pipeThrough(
     new TransformStream({
@@ -4235,6 +4236,16 @@ function processUIMessageStream({
               break;
             }
             case "start": {
+              if (resetStateOnMessageIdChange && chunk.messageId != null && chunk.messageId !== state.message.id) {
+                Object.assign(
+                  state,
+                  createStreamingUIMessageState({
+                    lastMessage: void 0,
+                    messageId: chunk.messageId
+                  }),
+                  { finishReason: void 0 }
+                );
+              }
               if (chunk.messageId != null) {
                 state.message.id = chunk.messageId;
               }
@@ -6388,6 +6399,20 @@ var DefaultStreamTextResult = class {
   }
 };
 
+// src/util/get-own.ts
+function getOwn(obj, key) {
+  return obj != null && Object.hasOwn(obj, key) ? obj[key] : void 0;
+}
+
+// src/ui/unavailable-tool.ts
+function markToolPartAsUnavailable(part) {
+  part.dynamic = false;
+  return part;
+}
+function isToolPartFromUnavailableTool(part) {
+  return part.type === "dynamic-tool" && part.dynamic === false;
+}
+
 // src/ui/convert-to-model-messages.ts
 function convertToModelMessages(messages, options) {
   const modelMessages = [];
@@ -6399,6 +6424,25 @@ function convertToModelMessages(messages, options) {
       )
     }));
   }
+  function createModelOutput({
+    toolPart,
+    toolName,
+    output,
+    errorMode
+  }) {
+    const tool2 = getOwn(options == null ? void 0 : options.tools, toolName);
+    if (errorMode === "none" && tool2 == null && isToolPartFromUnavailableTool(toolPart)) {
+      return {
+        type: "text",
+        value: "Tool output omitted because the tool is no longer available."
+      };
+    }
+    return createToolModelOutput({
+      output,
+      tool: tool2,
+      errorMode
+    });
+  }
   for (const message of messages) {
     switch (message.role) {
       case "system": {
@@ -6452,7 +6496,7 @@ function convertToModelMessages(messages, options) {
       case "assistant": {
         if (message.parts != null) {
           let processBlock2 = function() {
-            var _a16, _b, _c;
+            var _a16, _b;
             if (block.length === 0) {
               return;
             }
@@ -6504,9 +6548,10 @@ function convertToModelMessages(messages, options) {
                       type: "tool-result",
                       toolCallId: part.toolCallId,
                       toolName,
-                      output: createToolModelOutput({
+                      output: createModelOutput({
+                        toolPart: part,
+                        toolName,
                         output: part.state === "output-error" ? part.errorText : part.output,
-                        tool: (_b = options == null ? void 0 : options.tools) == null ? void 0 : _b[toolName],
                         errorMode: part.state === "output-error" ? "json" : "none"
                       }),
                       ...part.callProviderMetadata != null ? { providerOptions: part.callProviderMetadata } : {}
@@ -6514,7 +6559,7 @@ function convertToModelMessages(messages, options) {
                   }
                 }
               } else if (isDataUIPart(part)) {
-                const dataPart = (_c = options == null ? void 0 : options.convertDataPart) == null ? void 0 : _c.call(
+                const dataPart = (_b = options == null ? void 0 : options.convertDataPart) == null ? void 0 : _b.call(
                   options,
                   part
                 );
@@ -6537,7 +6582,6 @@ function convertToModelMessages(messages, options) {
               modelMessages.push({
                 role: "tool",
                 content: toolParts.map((toolPart) => {
-                  var _a17;
                   switch (toolPart.state) {
                     case "output-error":
                     case "output-available": {
@@ -6546,9 +6590,10 @@ function convertToModelMessages(messages, options) {
                         type: "tool-result",
                         toolCallId: toolPart.toolCallId,
                         toolName,
-                        output: createToolModelOutput({
+                        output: createModelOutput({
+                          toolPart,
+                          toolName,
                           output: toolPart.state === "output-error" ? toolPart.errorText : toolPart.output,
-                          tool: (_a17 = options == null ? void 0 : options.tools) == null ? void 0 : _a17[toolName],
                           errorMode: toolPart.state === "output-error" ? "text" : "none"
                         }),
                         ...toolPart.callProviderMetadata != null ? { providerOptions: toolPart.callProviderMetadata } : {}
@@ -6741,11 +6786,6 @@ var DefaultEmbedResult = class {
   }
 };
 
-// src/util/get-own.ts
-function getOwn(obj, key) {
-  return obj != null && Object.hasOwn(obj, key) ? obj[key] : void 0;
-}
-
 // src/util/set-own.ts
 function setOwn(object2, key, value) {
   Object.defineProperty(object2, key, {
@@ -8007,17 +8047,24 @@ var SerialJobExecutor = class {
   constructor() {
     this.queue = [];
     this.isProcessing = false;
+    this.idleWaiters = [];
   }
   async processQueue() {
     if (this.isProcessing) {
       return;
     }
     this.isProcessing = true;
-    while (this.queue.length > 0) {
-      await this.queue[0]();
-      this.queue.shift();
+    try {
+      while (this.queue.length > 0) {
+        await this.queue[0]();
+        this.queue.shift();
+      }
+    } finally {
+      this.isProcessing = false;
+      for (const resolve2 of this.idleWaiters.splice(0)) {
+        resolve2();
+      }
     }
-    this.isProcessing = false;
   }
   async run(job) {
     return new Promise((resolve2, reject) => {
@@ -8032,6 +8079,14 @@ var SerialJobExecutor = class {
       void this.processQueue();
     });
   }
+  async waitForIdle() {
+    if (!this.isProcessing && this.queue.length === 0) {
+      return;
+    }
+    await new Promise((resolve2) => {
+      this.idleWaiters.push(resolve2);
+    });
+  }
 };
 
 // src/util/simulate-readable-stream.ts
@@ -9778,7 +9833,9 @@ var import_provider_utils31 = require("@ai-sdk/provider-utils");
 var import_provider_utils30 = require("@ai-sdk/provider-utils");
 function appendPathToUrl(url, path) {
   const queryOrFragmentStart = url.search(/[?#]/);
-  return queryOrFragmentStart === -1 ? `${url}${path}` : `${url.slice(0, queryOrFragmentStart)}${path}${url.slice(queryOrFragmentStart)}`;
+  const urlPath = queryOrFragmentStart === -1 ? url : url.slice(0, queryOrFragmentStart);
+  const suffix = queryOrFragmentStart === -1 ? "" : url.slice(queryOrFragmentStart);
+  return `${urlPath.endsWith("/") && path.startsWith("/") ? urlPath.slice(0, -1) : urlPath}${path}${suffix}`;
 }
 var HttpChatTransport = class {
   constructor({
@@ -9949,6 +10006,8 @@ var AbstractChat = class {
     this.activeResumeRequest = void 0;
     this.resumableStreamState = void 0;
     this.jobExecutor = new SerialJobExecutor();
+    this.activeStopCount = 0;
+    this.stopGeneration = 0;
     /**
      * Appends or replaces a user message to the chat list. This triggers the API call to fetch
      * the assistant's response.
@@ -10039,6 +10098,27 @@ var AbstractChat = class {
     this.resumeStream = async (options = {}) => {
       await this.makeRequest({ trigger: "resume-stream", ...options });
     };
+    /**
+     * Resume the stream if the latest response ended with a resumable network
+     * disconnect. If a response is still active, wait for it to settle first so
+     * disconnects reported shortly after the caller's trigger are handled.
+     *
+     * @internal
+     */
+    this["~resumeStreamIfDisconnected"] = async ({
+      shouldResume = () => true
+    } = {}) => {
+      const activeResponse = this.activeResponse;
+      if (this.resumableStreamState == null && activeResponse != null) {
+        await activeResponse.completionPromise;
+        if (this.resumableStreamState !== activeResponse.state) {
+          return;
+        }
+      }
+      if (this.resumableStreamState != null && shouldResume()) {
+        await this.resumeStream();
+      }
+    };
     /**
      * Clear the error state and set the status to ready if the chat is in an error state.
      */
@@ -10055,15 +10135,21 @@ var AbstractChat = class {
       output,
       errorText
     }) => this.jobExecutor.run(async () => {
-      var _a16, _b;
       const messages = this.state.messages;
-      const lastMessage = messages[messages.length - 1];
-      this.state.replaceMessage(messages.length - 1, {
-        ...lastMessage,
-        parts: lastMessage.parts.map(
-          (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId ? { ...part, state, output, errorText } : part
+      const messageIndex = messages.findIndex(
+        (message) => message.parts.some(
+          (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId
         )
-      });
+      );
+      if (messageIndex !== -1) {
+        const message = messages[messageIndex];
+        this.state.replaceMessage(messageIndex, {
+          ...message,
+          parts: message.parts.map(
+            (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId ? { ...part, state, output, errorText } : part
+          )
+        });
+      }
       if (this.activeResponse) {
         this.activeResponse.state.message.parts = this.activeResponse.state.message.parts.map(
           (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId ? {
@@ -10074,22 +10160,41 @@ var AbstractChat = class {
           } : part
         );
       }
-      if (this.status !== "streaming" && this.status !== "submitted" && ((_a16 = this.sendAutomaticallyWhen) == null ? void 0 : _a16.call(this, { messages: this.state.messages }))) {
-        this.makeRequest({
-          trigger: "submit-message",
-          messageId: (_b = this.lastMessage) == null ? void 0 : _b.id
-        });
+      if (messageIndex !== -1 && this.status !== "streaming" && this.status !== "submitted" && this.sendAutomaticallyWhen) {
+        const shouldSend = await this.shouldSendAutomatically();
+        if (shouldSend) {
+          void this.runAutomaticRequest(
+            () => this.makeRequest({
+              trigger: "submit-message",
+              messageId: messages[messageIndex].id
+            }),
+            shouldSend
+          );
+        }
       }
     });
     /** @deprecated Use addToolOutput */
     this.addToolResult = this.addToolOutput;
     /**
-     * Abort the current request immediately, keep the generated tokens if any.
+     * Abort the current request, keep the generated tokens if any, and wait for
+     * the request pipeline to finish.
      */
     this.stop = async () => {
-      var _a16, _b;
-      (_a16 = this.activeResumeRequest) == null ? void 0 : _a16.abortController.abort();
-      (_b = this.activeResponse) == null ? void 0 : _b.abortController.abort();
+      this.activeStopCount++;
+      this.stopGeneration++;
+      try {
+        const activeResumeRequest = this.activeResumeRequest;
+        const activeResponse = this.activeResponse;
+        activeResumeRequest == null ? void 0 : activeResumeRequest.abortController.abort();
+        activeResponse == null ? void 0 : activeResponse.abortController.abort();
+        await Promise.all([
+          activeResumeRequest == null ? void 0 : activeResumeRequest.completionPromise,
+          activeResponse == null ? void 0 : activeResponse.completionPromise
+        ]);
+        await this.jobExecutor.waitForIdle();
+      } finally {
+        this.activeStopCount--;
+      }
     };
     this.id = id;
     this.transport = transport;
@@ -10135,19 +10240,53 @@ var AbstractChat = class {
   set messages(messages) {
     this.state.messages = messages;
   }
-  async makeRequest({
+  async shouldSendAutomatically() {
+    if (!this.sendAutomaticallyWhen)
+      return false;
+    const result = this.sendAutomaticallyWhen({
+      messages: this.state.messages
+    });
+    if (result && typeof result === "object" && "then" in result) {
+      return await result;
+    }
+    return result;
+  }
+  async runAutomaticRequest(request, shouldSend) {
+    const stopGeneration = this.stopGeneration;
+    const startedWhileStopping = this.activeStopCount > 0;
+    if (!(shouldSend != null ? shouldSend : await this.shouldSendAutomatically())) {
+      return;
+    }
+    if (startedWhileStopping || stopGeneration !== this.stopGeneration || this.activeStopCount > 0) {
+      return;
+    }
+    await request();
+  }
+  async makeRequest(options) {
+    const completion = createResolvablePromise();
+    try {
+      await this.makeRequestImpl({
+        ...options,
+        completionPromise: completion.promise
+      });
+    } finally {
+      completion.resolve();
+    }
+  }
+  async makeRequestImpl({
     trigger,
     metadata,
     headers,
     body,
-    messageId
+    messageId,
+    completionPromise
   }) {
-    var _a16, _b, _c, _d;
+    var _a16, _b;
     if (trigger !== "resume-stream") {
       this.resumableStreamState = void 0;
     }
     const abortController = new AbortController();
-    const activeResumeRequest = trigger === "resume-stream" ? { abortController } : void 0;
+    const activeResumeRequest = trigger === "resume-stream" ? { abortController, completionPromise } : void 0;
     if (activeResumeRequest) {
       (_a16 = this.activeResumeRequest) == null ? void 0 : _a16.abortController.abort();
       this.activeResumeRequest = activeResumeRequest;
@@ -10160,9 +10299,13 @@ var AbstractChat = class {
     };
     this.setStatus({ status: "submitted", error: void 0 });
     const lastMessage = this.lastMessage;
+    const responseMessageIndex = trigger === "submit-message" && messageId != null ? this.state.messages.findIndex((message) => message.id === messageId) : this.state.messages.length - 1;
+    const responseMessage = responseMessageIndex === -1 ? lastMessage : this.state.messages[responseMessageIndex];
+    const originalResponseMessageId = responseMessage == null ? void 0 : responseMessage.id;
     const resumableResponseMessage = trigger === "resume-stream" && (lastMessage == null ? void 0 : lastMessage.role) === "assistant" && lastMessage.parts.some(
       (part) => isToolOrDynamicToolUIPart(part) && part.state === "input-streaming"
     ) ? this.state.snapshot(lastMessage) : void 0;
+    const usesEarlierAssistantMessage = responseMessageIndex !== -1 && responseMessageIndex < this.state.messages.length - 1 && (responseMessage == null ? void 0 : responseMessage.role) === "assistant";
     let isAbort = false;
     let isDisconnect = false;
     let isError = false;
@@ -10170,10 +10313,11 @@ var AbstractChat = class {
     try {
       const response = {
         state: trigger === "resume-stream" && this.resumableStreamState != null ? this.resumableStreamState : createStreamingUIMessageState({
-          lastMessage: trigger === "resume-stream" ? resumableResponseMessage : this.state.snapshot(lastMessage),
+          lastMessage: trigger === "resume-stream" ? resumableResponseMessage : this.state.snapshot(responseMessage),
           messageId: this.generateId()
         }),
-        abortController
+        abortController,
+        completionPromise
       };
       activeResponse = response;
       response.abortController.signal.addEventListener("abort", () => {
@@ -10227,19 +10371,32 @@ var AbstractChat = class {
           return job({
             state: response.state,
             write: ({ updateStatus = true } = {}) => {
-              var _a17;
               if (response.abortController.signal.aborted) {
                 return;
               }
               if (updateStatus) {
                 this.setStatus({ status: "streaming" });
               }
-              const replaceLastMessage = response.state.message.id === ((_a17 = this.lastMessage) == null ? void 0 : _a17.id);
-              if (replaceLastMessage) {
+              const existingMessageIndex = this.state.messages.findLastIndex(
+                (message) => message.id === response.state.message.id
+              );
+              if (existingMessageIndex !== -1) {
                 this.state.replaceMessage(
-                  this.state.messages.length - 1,
+                  existingMessageIndex,
                   response.state.message
                 );
+              } else if (usesEarlierAssistantMessage) {
+                const originalMessageIndex = this.state.messages.findLastIndex(
+                  (message) => message.id === originalResponseMessageId
+                );
+                if (originalMessageIndex !== -1) {
+                  this.state.replaceMessage(
+                    originalMessageIndex,
+                    response.state.message
+                  );
+                } else {
+                  this.state.pushMessage(response.state.message);
+                }
               } else {
                 this.state.pushMessage(response.state.message);
               }
@@ -10325,14 +10482,19 @@ var AbstractChat = class {
         clearActiveResumeRequest();
       }
     }
-    if (((_c = this.sendAutomaticallyWhen) == null ? void 0 : _c.call(this, { messages: this.state.messages })) && !isError) {
-      await this.makeRequest({
-        trigger: "submit-message",
-        messageId: (_d = this.lastMessage) == null ? void 0 : _d.id,
-        metadata,
-        headers,
-        body
-      });
+    if (!isAbort && !isError) {
+      await this.runAutomaticRequest(
+        () => {
+          var _a17;
+          return this.makeRequest({
+            trigger: "submit-message",
+            messageId: (_a17 = this.lastMessage) == null ? void 0 : _a17.id,
+            metadata,
+            headers,
+            body
+          });
+        }
+      );
     }
   }
 };
@@ -10413,20 +10575,20 @@ function asDynamicToolPart(toolPart) {
     ...toolPart.callProviderMetadata === void 0 ? {} : { callProviderMetadata: toolPart.callProviderMetadata }
   };
   if (toolPart.state === "output-available") {
-    return {
+    return markToolPartAsUnavailable({
       ...common,
       state: "output-available",
       input: toolPart.input,
       output: toolPart.output,
       ...toolPart.preliminary === void 0 ? {} : { preliminary: toolPart.preliminary }
-    };
+    });
   }
-  return {
+  return markToolPartAsUnavailable({
     ...common,
     state: "output-error",
     input: toolPart.input,
     errorText: toolPart.errorText
-  };
+  });
 }
 var uiMessagesSchema = (0, import_provider_utils33.lazyValidator)(
   () => (0, import_provider_utils33.zodSchema)(
@@ -10482,6 +10644,7 @@ var uiMessagesSchema = (0, import_provider_utils33.lazyValidator)(
             }),
             import_v48.z.object({
               type: import_v48.z.literal("dynamic-tool"),
+              dynamic: import_v48.z.literal(false).optional(),
               toolName: import_v48.z.string(),
               toolCallId: import_v48.z.string(),
               state: import_v48.z.literal("input-streaming"),
@@ -10493,6 +10656,7 @@ var uiMessagesSchema = (0, import_provider_utils33.lazyValidator)(
             }),
             import_v48.z.object({
               type: import_v48.z.literal("dynamic-tool"),
+              dynamic: import_v48.z.literal(false).optional(),
               toolName: import_v48.z.string(),
               toolCallId: import_v48.z.string(),
               state: import_v48.z.literal("input-available"),
@@ -10504,6 +10668,7 @@ var uiMessagesSchema = (0, import_provider_utils33.lazyValidator)(
             }),
             import_v48.z.object({
               type: import_v48.z.literal("dynamic-tool"),
+              dynamic: import_v48.z.literal(false).optional(),
               toolName: import_v48.z.string(),
               toolCallId: import_v48.z.string(),
               state: import_v48.z.literal("output-available"),
@@ -10516,6 +10681,7 @@ var uiMessagesSchema = (0, import_provider_utils33.lazyValidator)(
             }),
             import_v48.z.object({
               type: import_v48.z.literal("dynamic-tool"),
+              dynamic: import_v48.z.literal(false).optional(),
               toolName: import_v48.z.string(),
               toolCallId: import_v48.z.string(),
               state: import_v48.z.literal("output-error"),
@@ -10715,7 +10881,7 @@ async function safeValidateUIMessages({
           }
           const toolPart = part;
           const toolName = toolPart.type.slice(5);
-          const tool2 = tools[toolName];
+          const tool2 = getOwn(tools, toolName);
           if (!tool2 && (toolPart.state === "output-available" || toolPart.state === "output-error")) {
             message.parts[partIdx] = asDynamicToolPart(toolPart);
             continue;
@@ -10954,6 +11120,7 @@ function readUIMessageStream({
   consumeStream({
     stream: processUIMessageStream({
       stream,
+      resetStateOnMessageIdChange: true,
       runUpdateMessageJob(job) {
         return job({
           state,