ai 5.0.230 → 5.0.232

package/dist/index.mjs

@@ -727,7 +727,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.230" : "0.0.0-test";
+var VERSION = true ? "5.0.232" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -1631,7 +1631,8 @@ function recordSpan({
   tracer,
   attributes,
   fn,
-  endWhenDone = true
+  endWhenDone = true,
+  endOnError = endWhenDone
 }) {
   return tracer.startActiveSpan(name16, { attributes }, async (span) => {
     try {
@@ -1644,7 +1645,9 @@ function recordSpan({
       try {
         recordErrorOnSpan(span, error);
       } finally {
-        span.end();
+        if (endOnError) {
+          span.end();
+        }
       }
       throw error;
     }
@@ -4221,9 +4224,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();
@@ -4233,6 +4247,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();
   }
 }
@@ -5186,6 +5202,7 @@ var DefaultStreamTextResult = class {
               }),
               tracer,
               endWhenDone: false,
+              endOnError: true,
               fn: async (doStreamSpan2) => {
                 return {
                   startTimestampMs: now2(),
@@ -7701,6 +7718,7 @@ var DefaultStreamObjectResult = class {
       }),
       tracer,
       endWhenDone: false,
+      endOnError: true,
       fn: async (rootSpan) => {
         const standardizedPrompt = await standardizePrompt({
           system,
@@ -7772,6 +7790,7 @@ var DefaultStreamObjectResult = class {
             }),
             tracer,
             endWhenDone: false,
+            endOnError: true,
             fn: async (doStreamSpan2) => ({
               startTimestampMs: now2(),
               doStreamSpan: doStreamSpan2,
@@ -9307,7 +9326,8 @@ var HttpChatTransport = class {
     const response = await fetch2(api, {
       method: "GET",
       headers,
-      credentials
+      credentials,
+      signal: options.abortSignal
     });
     if (response.status === 204) {
       return null;
@@ -9362,6 +9382,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
@@ -9501,12 +9522,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;
@@ -9559,7 +9577,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;
@@ -9572,7 +9602,7 @@ var AbstractChat = class {
           lastMessage: this.state.snapshot(lastMessage),
           messageId: this.generateId()
         }),
-        abortController: new AbortController()
+        abortController
       };
       activeResponse = response;
       response.abortController.signal.addEventListener("abort", () => {
@@ -9583,12 +9613,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;
@@ -9606,11 +9647,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) {
@@ -9622,8 +9669,8 @@ var AbstractChat = class {
                 this.state.pushMessage(response.state.message);
               }
             }
-          })
-        )
+          });
+        })
       );
       await consumeStream({
         stream: processUIMessageStream({
@@ -9637,15 +9684,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;
@@ -9659,7 +9720,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,
@@ -9674,11 +9735,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