ai 6.0.248 → 6.0.250

package/dist/index.mjs

@@ -1193,7 +1193,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.248" : "0.0.0-test";
+var VERSION = true ? "6.0.250" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -2429,7 +2429,8 @@ async function recordSpan({
   tracer,
   attributes,
   fn,
-  endWhenDone = true
+  endWhenDone = true,
+  endOnError = endWhenDone
 }) {
   return tracer.startActiveSpan(
     name22,
@@ -2446,7 +2447,9 @@ async function recordSpan({
         try {
           recordErrorOnSpan(span, error);
         } finally {
-          span.end();
+          if (endOnError) {
+            span.end();
+          }
         }
         throw error;
       }
@@ -6517,9 +6520,19 @@ function createAsyncIterableStream(source) {
 // src/util/consume-stream.ts
 async function consumeStream({
   stream,
-  onError
+  onError,
+  abortSignal
 }) {
   const reader = stream.getReader();
+  const cancelOnAbort = () => {
+    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();
@@ -6529,6 +6542,7 @@ async function consumeStream({
   } catch (error) {
     onError == null ? void 0 : onError(error);
   } finally {
+    abortSignal == null ? void 0 : abortSignal.removeEventListener("abort", cancelOnAbort);
     reader.releaseLock();
   }
 }
@@ -7818,6 +7832,7 @@ var DefaultStreamTextResult = class {
                 }),
                 tracer,
                 endWhenDone: false,
+                endOnError: true,
                 fn: async (doStreamSpan2) => ({
                   startTimestampMs: now2(),
                   // get before the call
@@ -11347,6 +11362,7 @@ var DefaultStreamObjectResult = class {
       }),
       tracer,
       endWhenDone: false,
+      endOnError: true,
       fn: async (rootSpan) => {
         const standardizedPrompt = await standardizePrompt({
           system,
@@ -11418,6 +11434,7 @@ var DefaultStreamObjectResult = class {
             }),
             tracer,
             endWhenDone: false,
+            endOnError: true,
             fn: async (doStreamSpan2) => ({
               startTimestampMs: now2(),
               doStreamSpan: doStreamSpan2,
@@ -13653,7 +13670,8 @@ var HttpChatTransport = class {
     const response = await fetch2(api, {
       method: "GET",
       headers,
-      credentials
+      credentials,
+      signal: options.abortSignal
     });
     if (response.status === 204) {
       return null;
@@ -13708,6 +13726,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
@@ -13877,12 +13896,9 @@ var AbstractChat = class {
      * Abort the current request immediately, keep the generated tokens if any.
      */
     this.stop = async () => {
-      var _a22;
-      if (this.status !== "streaming" && this.status !== "submitted")
-        return;
-      if ((_a22 = this.activeResponse) == null ? void 0 : _a22.abortController) {
-        this.activeResponse.abortController.abort();
-      }
+      var _a22, _b;
+      (_a22 = this.activeResumeRequest) == null ? void 0 : _a22.abortController.abort();
+      (_b = this.activeResponse) == null ? void 0 : _b.abortController.abort();
     };
     this.id = id;
     this.transport = transport;
@@ -13946,25 +13962,60 @@ var AbstractChat = class {
     body,
     messageId
   }) {
-    var _a22, _b;
+    var _a22, _b, _c;
+    const abortController = new AbortController();
+    const activeResumeRequest = trigger === "resume-stream" ? { abortController } : void 0;
+    if (activeResumeRequest) {
+      (_a22 = this.activeResumeRequest) == null ? void 0 : _a22.abortController.abort();
+      this.activeResumeRequest = activeResumeRequest;
+    }
+    const isCurrentRequest = () => activeResumeRequest == null || this.activeResumeRequest === activeResumeRequest;
+    const clearActiveResumeRequest = () => {
+      if (this.activeResumeRequest === activeResumeRequest) {
+        this.activeResumeRequest = void 0;
+      }
+    };
     let resumeStream;
     if (trigger === "resume-stream") {
       try {
         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" });
+          }
+          clearActiveResumeRequest();
+          return;
+        }
         if (reconnect == null) {
+          this.setStatus({ status: "ready" });
+          clearActiveResumeRequest();
           return;
         }
         resumeStream = reconnect;
       } catch (err) {
+        if (abortController.signal.aborted || err.name === "AbortError") {
+          if (isCurrentRequest()) {
+            this.setStatus({ status: "ready" });
+          }
+          clearActiveResumeRequest();
+          return;
+        }
+        if (!isCurrentRequest()) {
+          return;
+        }
         if (this.onError && err instanceof Error) {
           this.onError(err);
         }
         this.setStatus({ status: "error", error: err });
+        clearActiveResumeRequest();
         return;
       }
     }
@@ -13977,10 +14028,10 @@ var AbstractChat = class {
     try {
       const response = {
         state: createStreamingUIMessageState({
-          lastMessage: trigger === "regenerate-message" ? void 0 : this.state.snapshot(lastMessage),
+          lastMessage: trigger === "resume-stream" || trigger === "regenerate-message" ? void 0 : this.state.snapshot(lastMessage),
           messageId: this.generateId()
         }),
-        abortController: new AbortController()
+        abortController
       };
       activeResponse = response;
       response.abortController.signal.addEventListener("abort", () => {
@@ -14004,11 +14055,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 _a23;
+              if (response.abortController.signal.aborted) {
+                return;
+              }
               this.setStatus({ status: "streaming" });
               const replaceLastMessage = response.state.message.id === ((_a23 = this.lastMessage) == null ? void 0 : _a23.id);
               if (replaceLastMessage) {
@@ -14020,8 +14077,8 @@ var AbstractChat = class {
                 this.state.pushMessage(response.state.message);
               }
             }
-          })
-        )
+          });
+        })
       );
       await consumeStream({
         stream: processUIMessageStream({
@@ -14035,15 +14092,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;
@@ -14057,7 +14128,7 @@ var AbstractChat = class {
     } finally {
       try {
         if (activeResponse) {
-          (_a22 = this.onFinish) == null ? void 0 : _a22.call(this, {
+          (_b = this.onFinish) == null ? void 0 : _b.call(this, {
             message: activeResponse.state.message,
             messages: this.state.messages,
             isAbort,
@@ -14072,11 +14143,12 @@ var AbstractChat = class {
       if (this.activeResponse === activeResponse) {
         this.activeResponse = void 0;
       }
+      clearActiveResumeRequest();
     }
     if (!isError && await this.shouldSendAutomatically()) {
       await this.makeRequest({
         trigger: "submit-message",
-        messageId: (_b = this.lastMessage) == null ? void 0 : _b.id,
+        messageId: (_c = this.lastMessage) == null ? void 0 : _c.id,
         metadata,
         headers,
         body