ai 7.0.119 → 7.0.120

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # ai
 
+## 7.0.120
+
+### Patch Changes
+
+- b032d70: fix(ai): preserve tool metadata from tool output chunks
+- e21b98b: fix(ai): continue active UI message parts when resuming after a disconnect
+- Updated dependencies [e361d39]
+  - @ai-sdk/gateway@4.0.98
+
 ## 7.0.119
 
 ### Patch Changes

package/dist/index.d.ts

@@ -5816,6 +5816,7 @@ declare abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
     private pendingApprovalMessageId;
     private activeResponse;
     private activeResumeRequest;
+    private resumableStreamState;
     private jobExecutor;
     constructor({ generateId, id, transport, messageMetadataSchema, dataPartSchemas, state, onError, onToolCall, onFinish, onData, sendAutomaticallyWhen, }: Omit<ChatInit<UI_MESSAGE>, 'messages'> & {
         state: ChatState<UI_MESSAGE>;

package/dist/index.js

@@ -1328,7 +1328,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "7.0.119" : "0.0.0-test";
+var VERSION = true ? "7.0.120" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -7979,7 +7979,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: chunk.toolMetadata ?? toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -7995,7 +7995,7 @@ function processUIMessageStream({
                     preliminary: chunk.preliminary,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: chunk.toolMetadata ?? toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -8016,7 +8016,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: chunk.toolMetadata ?? toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -8032,7 +8032,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: chunk.toolMetadata ?? toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -22007,6 +22007,7 @@ var AbstractChat = class {
     this.pendingMessagePreparations = /* @__PURE__ */ new Set();
     this.activeResponse = void 0;
     this.activeResumeRequest = void 0;
+    this.resumableStreamState = void 0;
     this.jobExecutor = new SerialJobExecutor();
     /**
      * Appends or replaces a user message to the chat list. This triggers the API call to fetch
@@ -22301,6 +22302,9 @@ var AbstractChat = class {
     body,
     messageId
   }) {
+    if (trigger !== "resume-stream") {
+      this.resumableStreamState = void 0;
+    }
     const abortController = new AbortController();
     const activeResumeRequest = trigger === "resume-stream" ? { abortController } : void 0;
     if (activeResumeRequest) {
@@ -22328,12 +22332,14 @@ var AbstractChat = class {
           });
           if (isCurrentRequest()) {
             this.setStatus({ status: "ready" });
+            this.resumableStreamState = void 0;
           }
           clearActiveResumeRequest();
           return;
         }
         if (reconnect == null) {
           this.setStatus({ status: "ready" });
+          this.resumableStreamState = void 0;
           clearActiveResumeRequest();
           return;
         }
@@ -22342,6 +22348,7 @@ var AbstractChat = class {
         if (abortController.signal.aborted || err.name === "AbortError") {
           if (isCurrentRequest()) {
             this.setStatus({ status: "ready" });
+            this.resumableStreamState = void 0;
           }
           clearActiveResumeRequest();
           return;
@@ -22368,7 +22375,7 @@ var AbstractChat = class {
     let activeResponse;
     try {
       const response = {
-        state: createStreamingUIMessageState({
+        state: trigger === "resume-stream" && this.resumableStreamState != null ? this.resumableStreamState : createStreamingUIMessageState({
           lastMessage: trigger === "resume-stream" || trigger === "regenerate-message" ? void 0 : this.state.snapshot(responseMessage),
           messageId: this.generateId()
         }),
@@ -22446,17 +22453,26 @@ var AbstractChat = class {
       if (isAbort) {
         if (isCurrentRequest()) {
           this.setStatus({ status: "ready" });
+          if (this.resumableStreamState === response.state) {
+            this.resumableStreamState = void 0;
+          }
         }
         return null;
       }
       if (isCurrentRequest()) {
         this.setStatus({ status: "ready" });
+        if (this.resumableStreamState === response.state) {
+          this.resumableStreamState = void 0;
+        }
       }
     } catch (err) {
       if (isAbort || err.name === "AbortError") {
         isAbort = true;
         if (isCurrentRequest()) {
           this.setStatus({ status: "ready" });
+          if (this.resumableStreamState === activeResponse?.state) {
+            this.resumableStreamState = void 0;
+          }
         }
         return null;
       }
@@ -22467,6 +22483,11 @@ var AbstractChat = class {
       if (err instanceof TypeError && (err.message.toLowerCase().includes("fetch") || err.message.toLowerCase().includes("network"))) {
         isDisconnect = true;
       }
+      if (isDisconnect) {
+        this.resumableStreamState = activeResponse?.state;
+      } else if (this.resumableStreamState === activeResponse?.state) {
+        this.resumableStreamState = void 0;
+      }
       if (this.onError && err instanceof Error) {
         this.onError(err);
       }