ai 5.0.260 → 5.0.261

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # ai
 
+## 5.0.261
+
+### Patch Changes
+
+- 69879fe: fix(ai): prevent unhandled rejections when UI message stream reading stops early
+
 ## 5.0.260
 
 ### Patch Changes

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.260" : "0.0.0-test";
+var VERSION = true ? "5.0.261" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -10675,9 +10675,15 @@ function readUIMessageStream({
   var _a16;
   let controller;
   let hasErrored = false;
+  let isCancelled = false;
+  const abortController = new AbortController();
   const outputStream = new ReadableStream({
     start(controllerParam) {
       controller = controllerParam;
+    },
+    cancel() {
+      isCancelled = true;
+      abortController.abort();
     }
   });
   const state = createStreamingUIMessageState({
@@ -10698,15 +10704,18 @@ function readUIMessageStream({
         return job({
           state,
           write: () => {
-            controller == null ? void 0 : controller.enqueue(createUIMessageSnapshot(state.message));
+            if (!isCancelled) {
+              controller == null ? void 0 : controller.enqueue(createUIMessageSnapshot(state.message));
+            }
           }
         });
       },
       onError: handleError
     }),
-    onError: handleError
+    onError: handleError,
+    abortSignal: abortController.signal
   }).finally(() => {
-    if (!hasErrored) {
+    if (!hasErrored && !isCancelled) {
       controller == null ? void 0 : controller.close();
     }
   });