ai 6.0.303 → 6.0.304

package/dist/index.mjs

@@ -1225,7 +1225,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.303" : "0.0.0-test";
+var VERSION = true ? "6.0.304" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -1428,32 +1428,6 @@ async function convertToLanguageModelPrompt({
     download2 != null ? download2 : createDefaultDownloadFunction(void 0, abortSignal),
     supportedUrls
   );
-  const approvalIdToToolCallId = /* @__PURE__ */ new Map();
-  for (const message of prompt.messages) {
-    if (message.role === "assistant" && Array.isArray(message.content)) {
-      for (const part of message.content) {
-        if (part.type === "tool-approval-request" && "approvalId" in part && "toolCallId" in part) {
-          approvalIdToToolCallId.set(
-            part.approvalId,
-            part.toolCallId
-          );
-        }
-      }
-    }
-  }
-  const approvedToolCallIds = /* @__PURE__ */ new Set();
-  for (const message of prompt.messages) {
-    if (message.role === "tool") {
-      for (const part of message.content) {
-        if (part.type === "tool-approval-response") {
-          const toolCallId = approvalIdToToolCallId.get(part.approvalId);
-          if (toolCallId) {
-            approvedToolCallIds.add(toolCallId);
-          }
-        }
-      }
-    }
-  }
   const messages = [
     ...prompt.system != null ? typeof prompt.system === "string" ? [{ role: "system", content: prompt.system }] : asArray(prompt.system).map((message) => ({
       role: "system",
@@ -1506,9 +1480,6 @@ async function convertToLanguageModelPrompt({
       }
       case "user":
       case "system":
-        for (const id of approvedToolCallIds) {
-          toolCallIds.delete(id);
-        }
         if (toolCallIds.size > 0) {
           throw new MissingToolResultsError({
             toolCallIds: Array.from(toolCallIds)
@@ -1517,9 +1488,6 @@ async function convertToLanguageModelPrompt({
         break;
     }
   }
-  for (const id of approvedToolCallIds) {
-    toolCallIds.delete(id);
-  }
   if (toolCallIds.size > 0) {
     throw new MissingToolResultsError({ toolCallIds: Array.from(toolCallIds) });
   }
@@ -6052,13 +6020,15 @@ function processUIMessageStream({
   onError,
   onToolCall,
   onData,
-  resetStateOnMessageIdChange = false
+  resetStateOnMessageIdChange = false,
+  resetStateOnFirstMessageStart = false
 }) {
+  let isFirstMessageStart = true;
   return stream.pipeThrough(
     new TransformStream({
       async transform(chunk, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
-          var _a23, _b, _c, _d, _e, _f, _g, _h;
+          var _a23, _b, _c, _d, _e, _f, _g, _h, _i;
           function getCurrentStepParts() {
             const parts = state.message.parts;
             let currentStepStartIndex = parts.length - 1;
@@ -6565,12 +6535,14 @@ function processUIMessageStream({
               break;
             }
             case "start": {
-              if (resetStateOnMessageIdChange && chunk.messageId != null && chunk.messageId !== state.message.id) {
+              const shouldReset = resetStateOnFirstMessageStart && isFirstMessageStart || resetStateOnMessageIdChange && chunk.messageId != null && chunk.messageId !== state.message.id;
+              isFirstMessageStart = false;
+              if (shouldReset) {
                 Object.assign(
                   state,
                   createStreamingUIMessageState({
                     lastMessage: void 0,
-                    messageId: chunk.messageId
+                    messageId: (_i = chunk.messageId) != null ? _i : state.message.id
                   }),
                   { finishReason: void 0 }
                 );
@@ -6579,7 +6551,7 @@ function processUIMessageStream({
                 state.message.id = chunk.messageId;
               }
               await updateMessageMetadata(chunk.messageMetadata);
-              if (chunk.messageId != null || chunk.messageMetadata != null) {
+              if (shouldReset || chunk.messageId != null || chunk.messageMetadata != null) {
                 write({ updateStatus: false });
               }
               break;
@@ -14635,8 +14607,12 @@ var HttpChatTransport = class {
 
 // src/ui/default-chat-transport.ts
 var DefaultChatTransport = class extends HttpChatTransport {
-  constructor(options = {}) {
+  constructor({
+    resumeStreamIsReplay = true,
+    ...options
+  } = {}) {
     super(options);
+    this.resumeStreamIsReplay = resumeStreamIsReplay;
   }
   processResponseStream(stream) {
     return parseJsonEventStream2({
@@ -14781,6 +14757,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.
      */
@@ -14841,27 +14838,31 @@ var AbstractChat = class {
       options
     }) => this.jobExecutor.run(async () => {
       const messages = this.state.messages;
-      const lastMessage = messages[messages.length - 1];
       const updatePart = (part) => isToolUIPart(part) && part.toolCallId === toolCallId ? { ...part, state, output, errorText } : part;
-      this.state.replaceMessage(messages.length - 1, {
-        ...lastMessage,
-        parts: lastMessage.parts.map(updatePart)
-      });
+      const messageIndex = messages.findIndex(
+        (message) => message.parts.some(
+          (part) => isToolUIPart(part) && part.toolCallId === toolCallId
+        )
+      );
+      if (messageIndex !== -1) {
+        const message = messages[messageIndex];
+        this.state.replaceMessage(messageIndex, {
+          ...message,
+          parts: message.parts.map(updatePart)
+        });
+      }
       if (this.activeResponse) {
         this.activeResponse.state.message.parts = this.activeResponse.state.message.parts.map(updatePart);
       }
-      if (this.status !== "streaming" && this.status !== "submitted" && this.sendAutomaticallyWhen) {
+      if (messageIndex !== -1 && this.status !== "streaming" && this.status !== "submitted" && this.sendAutomaticallyWhen) {
         const shouldSend = await this.shouldSendAutomatically();
         if (shouldSend) {
           void this.runAutomaticRequest(
-            () => {
-              var _a23;
-              return this.makeRequest({
-                trigger: "submit-message",
-                messageId: (_a23 = this.lastMessage) == null ? void 0 : _a23.id,
-                ...options
-              });
-            },
+            () => this.makeRequest({
+              trigger: "submit-message",
+              messageId: messages[messageIndex].id,
+              ...options
+            }),
             shouldSend
           );
         }
@@ -15065,6 +15066,7 @@ var AbstractChat = class {
     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" && (responseMessage == null ? void 0 : responseMessage.role) === "assistant" && responseMessage.parts.some(
       (part) => isToolUIPart(part) && part.state === "input-streaming"
     ) ? this.state.snapshot(responseMessage) : void 0;
@@ -15111,23 +15113,32 @@ var AbstractChat = class {
           return job({
             state: response.state,
             write: ({ updateStatus = true } = {}) => {
-              var _a24;
               if (response.abortController.signal.aborted) {
                 return;
               }
               if (updateStatus) {
                 this.setStatus({ status: "streaming" });
               }
-              if (usesEarlierAssistantMessage) {
+              const existingMessageIndex = this.state.messages.findLastIndex(
+                (message) => message.id === response.state.message.id
+              );
+              if (existingMessageIndex !== -1) {
                 this.state.replaceMessage(
-                  responseMessageIndex,
+                  existingMessageIndex,
                   response.state.message
                 );
-              } else if (response.state.message.id === ((_a24 = this.lastMessage) == null ? void 0 : _a24.id)) {
-                this.state.replaceMessage(
-                  this.state.messages.length - 1,
-                  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);
               }
@@ -15138,6 +15149,8 @@ var AbstractChat = class {
       await consumeStream({
         stream: processUIMessageStream({
           stream,
+          resetStateOnMessageIdChange: trigger === "resume-stream",
+          resetStateOnFirstMessageStart: trigger === "resume-stream" && this.transport.resumeStreamIsReplay,
           onToolCall: this.onToolCall,
           onData: this.onData,
           messageMetadataSchema: this.messageMetadataSchema,