ai 5.0.230 → 5.0.231

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # ai
 
+## 5.0.231
+
+### Patch Changes
+
+- 2709314: Stop pending and active resumed chat streams after cancellation, and prevent overlapping resumptions from applying stale updates.
+- Updated dependencies [b474448]
+  - @ai-sdk/gateway@2.0.130
+
 ## 5.0.230
 
 ### Patch Changes

package/dist/index.d.mts

@@ -3094,9 +3094,10 @@ functionality that can be fully encapsulated in the provider.
  * @param {ReadableStream} stream - The ReadableStream to be consumed.
  * @returns {Promise<void>} A promise that resolves when the stream is fully consumed.
  */
-declare function consumeStream({ stream, onError, }: {
+declare function consumeStream({ stream, onError, abortSignal, }: {
     stream: ReadableStream;
     onError?: (error: unknown) => void;
+    abortSignal?: AbortSignal;
 }): Promise<void>;
 
 /**
@@ -3983,6 +3984,7 @@ interface ChatTransport<UI_MESSAGE extends UIMessage> {
      *
      * @param options - Configuration object containing:
      * @param options.chatId - Unique identifier for the chat session to reconnect to
+     * @param options.abortSignal - Signal to abort the reconnection request if needed
      * @param options.headers - Additional HTTP headers to include in the reconnection request
      * @param options.body - Additional JSON properties to include in the request body
      * @param options.metadata - Custom metadata to attach to the request
@@ -3996,6 +3998,8 @@ interface ChatTransport<UI_MESSAGE extends UIMessage> {
     reconnectToStream: (options: {
         /** Unique identifier for the chat session to reconnect to */
         chatId: string;
+        /** Signal to abort the reconnection request if needed */
+        abortSignal?: AbortSignal;
     } & ChatRequestOptions) => Promise<ReadableStream<UIMessageChunk> | null>;
 }
 
@@ -4113,6 +4117,7 @@ declare abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
     private onData?;
     private sendAutomaticallyWhen?;
     private activeResponse;
+    private activeResumeRequest;
     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.d.ts

@@ -3094,9 +3094,10 @@ functionality that can be fully encapsulated in the provider.
  * @param {ReadableStream} stream - The ReadableStream to be consumed.
  * @returns {Promise<void>} A promise that resolves when the stream is fully consumed.
  */
-declare function consumeStream({ stream, onError, }: {
+declare function consumeStream({ stream, onError, abortSignal, }: {
     stream: ReadableStream;
     onError?: (error: unknown) => void;
+    abortSignal?: AbortSignal;
 }): Promise<void>;
 
 /**
@@ -3983,6 +3984,7 @@ interface ChatTransport<UI_MESSAGE extends UIMessage> {
      *
      * @param options - Configuration object containing:
      * @param options.chatId - Unique identifier for the chat session to reconnect to
+     * @param options.abortSignal - Signal to abort the reconnection request if needed
      * @param options.headers - Additional HTTP headers to include in the reconnection request
      * @param options.body - Additional JSON properties to include in the request body
      * @param options.metadata - Custom metadata to attach to the request
@@ -3996,6 +3998,8 @@ interface ChatTransport<UI_MESSAGE extends UIMessage> {
     reconnectToStream: (options: {
         /** Unique identifier for the chat session to reconnect to */
         chatId: string;
+        /** Signal to abort the reconnection request if needed */
+        abortSignal?: AbortSignal;
     } & ChatRequestOptions) => Promise<ReadableStream<UIMessageChunk> | null>;
 }
 
@@ -4113,6 +4117,7 @@ declare abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
     private onData?;
     private sendAutomaticallyWhen?;
     private activeResponse;
+    private activeResumeRequest;
     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

@@ -816,7 +816,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "5.0.230" : "0.0.0-test";
+var VERSION = true ? "5.0.231" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -4290,9 +4290,20 @@ function createAsyncIterableStream(source) {
 // src/util/consume-stream.ts
 async function consumeStream({
   stream,
-  onError
+  onError,
+  abortSignal
 }) {
   const reader = stream.getReader();
+  let cancelPromise;
+  const cancelOnAbort = () => {
+    cancelPromise != null ? cancelPromise : cancelPromise = reader.cancel().catch(() => {
+    });
+  };
+  if (abortSignal == null ? void 0 : abortSignal.aborted) {
+    cancelOnAbort();
+  } else {
+    abortSignal == null ? void 0 : abortSignal.addEventListener("abort", cancelOnAbort, { once: true });
+  }
   try {
     while (true) {
       const { done } = await reader.read();
@@ -4302,6 +4313,8 @@ async function consumeStream({
   } catch (error) {
     onError == null ? void 0 : onError(error);
   } finally {
+    await cancelPromise;
+    abortSignal == null ? void 0 : abortSignal.removeEventListener("abort", cancelOnAbort);
     reader.releaseLock();
   }
 }
@@ -9331,7 +9344,8 @@ var HttpChatTransport = class {
     const response = await fetch2(api, {
       method: "GET",
       headers,
-      credentials
+      credentials,
+      signal: options.abortSignal
     });
     if (response.status === 204) {
       return null;
@@ -9386,6 +9400,7 @@ var AbstractChat = class {
     sendAutomaticallyWhen
   }) {
     this.activeResponse = void 0;
+    this.activeResumeRequest = void 0;
     this.jobExecutor = new SerialJobExecutor();
     /**
      * Appends or replaces a user message to the chat list. This triggers the API call to fetch
@@ -9525,12 +9540,9 @@ var AbstractChat = class {
      * Abort the current request immediately, keep the generated tokens if any.
      */
     this.stop = async () => {
-      var _a16;
-      if (this.status !== "streaming" && this.status !== "submitted")
-        return;
-      if ((_a16 = this.activeResponse) == null ? void 0 : _a16.abortController) {
-        this.activeResponse.abortController.abort();
-      }
+      var _a16, _b;
+      (_a16 = this.activeResumeRequest) == null ? void 0 : _a16.abortController.abort();
+      (_b = this.activeResponse) == null ? void 0 : _b.abortController.abort();
     };
     this.id = id;
     this.transport = transport;
@@ -9583,7 +9595,19 @@ var AbstractChat = class {
     body,
     messageId
   }) {
-    var _a16, _b, _c;
+    var _a16, _b, _c, _d;
+    const abortController = new AbortController();
+    const activeResumeRequest = trigger === "resume-stream" ? { abortController } : void 0;
+    if (activeResumeRequest) {
+      (_a16 = this.activeResumeRequest) == null ? void 0 : _a16.abortController.abort();
+      this.activeResumeRequest = activeResumeRequest;
+    }
+    const isCurrentRequest = () => activeResumeRequest == null || this.activeResumeRequest === activeResumeRequest;
+    const clearActiveResumeRequest = () => {
+      if (this.activeResumeRequest === activeResumeRequest) {
+        this.activeResumeRequest = void 0;
+      }
+    };
     this.setStatus({ status: "submitted", error: void 0 });
     const lastMessage = this.lastMessage;
     let isAbort = false;
@@ -9596,7 +9620,7 @@ var AbstractChat = class {
           lastMessage: this.state.snapshot(lastMessage),
           messageId: this.generateId()
         }),
-        abortController: new AbortController()
+        abortController
       };
       activeResponse = response;
       response.abortController.signal.addEventListener("abort", () => {
@@ -9607,12 +9631,23 @@ var AbstractChat = class {
       if (trigger === "resume-stream") {
         const reconnect = await this.transport.reconnectToStream({
           chatId: this.id,
+          abortSignal: abortController.signal,
           metadata,
           headers,
           body
         });
+        if (abortController.signal.aborted || !isCurrentRequest()) {
+          await (reconnect == null ? void 0 : reconnect.cancel().catch(() => {
+          }));
+          if (isCurrentRequest()) {
+            this.setStatus({ status: "ready" });
+          }
+          return;
+        }
         if (reconnect == null) {
-          this.setStatus({ status: "ready" });
+          if (isCurrentRequest()) {
+            this.setStatus({ status: "ready" });
+          }
           return;
         }
         stream = reconnect;
@@ -9630,11 +9665,17 @@ var AbstractChat = class {
       }
       const runUpdateMessageJob = (job) => (
         // serialize the job execution to avoid race conditions:
-        this.jobExecutor.run(
-          () => job({
+        this.jobExecutor.run(() => {
+          if (response.abortController.signal.aborted) {
+            return Promise.resolve();
+          }
+          return job({
             state: response.state,
             write: () => {
               var _a17;
+              if (response.abortController.signal.aborted) {
+                return;
+              }
               this.setStatus({ status: "streaming" });
               const replaceLastMessage = response.state.message.id === ((_a17 = this.lastMessage) == null ? void 0 : _a17.id);
               if (replaceLastMessage) {
@@ -9646,8 +9687,8 @@ var AbstractChat = class {
                 this.state.pushMessage(response.state.message);
               }
             }
-          })
-        )
+          });
+        })
       );
       await consumeStream({
         stream: processUIMessageStream({
@@ -9661,15 +9702,29 @@ var AbstractChat = class {
             throw error;
           }
         }),
+        abortSignal: response.abortController.signal,
         onError: (error) => {
           throw error;
         }
       });
-      this.setStatus({ status: "ready" });
+      if (isAbort) {
+        if (isCurrentRequest()) {
+          this.setStatus({ status: "ready" });
+        }
+        return null;
+      }
+      if (isCurrentRequest()) {
+        this.setStatus({ status: "ready" });
+      }
     } catch (err) {
       if (isAbort || err.name === "AbortError") {
         isAbort = true;
-        this.setStatus({ status: "ready" });
+        if (isCurrentRequest()) {
+          this.setStatus({ status: "ready" });
+        }
+        return null;
+      }
+      if (!isCurrentRequest()) {
         return null;
       }
       isError = true;
@@ -9683,7 +9738,7 @@ var AbstractChat = class {
     } finally {
       try {
         if (activeResponse) {
-          (_a16 = this.onFinish) == null ? void 0 : _a16.call(this, {
+          (_b = this.onFinish) == null ? void 0 : _b.call(this, {
             message: activeResponse.state.message,
             messages: this.state.messages,
             isAbort,
@@ -9698,11 +9753,12 @@ var AbstractChat = class {
       if (this.activeResponse === activeResponse) {
         this.activeResponse = void 0;
       }
+      clearActiveResumeRequest();
     }
-    if (((_b = this.sendAutomaticallyWhen) == null ? void 0 : _b.call(this, { messages: this.state.messages })) && !isError) {
+    if (((_c = this.sendAutomaticallyWhen) == null ? void 0 : _c.call(this, { messages: this.state.messages })) && !isError) {
       await this.makeRequest({
         trigger: "submit-message",
-        messageId: (_c = this.lastMessage) == null ? void 0 : _c.id,
+        messageId: (_d = this.lastMessage) == null ? void 0 : _d.id,
         metadata,
         headers,
         body