ai 5.0.0-beta.23 → 5.0.0-beta.24

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # ai
 
+## 5.0.0-beta.24
+
+### Patch Changes
+
+- add5ac1: feat (ai): make streamText toUIMessageStream async iterable
+- ff1c81a: feat (ai): add streamText onAbort callback
+- e4c8647: feat (ui): allow asynchronous onFinish in createUIMessageStream
+- 383cbfa: feat (ai): add isAborted to onFinish callback for ui message streams
+- Updated dependencies [57edfcb]
+- Updated dependencies [383cbfa]
+  - @ai-sdk/provider-utils@3.0.0-beta.5
+  - @ai-sdk/gateway@1.0.0-beta.10
+
 ## 5.0.0-beta.23
 
 ### Patch Changes

package/dist/bin/ai.js

@@ -336,6 +336,9 @@ var uiMessageChunkSchema = import_v43.z.union([
     type: import_v43.z.literal("finish"),
     messageMetadata: import_v43.z.unknown().optional()
   }),
+  import_v43.z.strictObject({
+    type: import_v43.z.literal("abort")
+  }),
   import_v43.z.strictObject({
     type: import_v43.z.literal("message-metadata"),
     messageMetadata: import_v43.z.unknown()
@@ -1079,6 +1082,7 @@ function handleUIMessageStreamFinish({
   } else {
     messageId = lastMessage.id;
   }
+  let isAborted = false;
   const idInjectedStream = stream.pipeThrough(
     new TransformStream({
       transform(chunk, controller) {
@@ -1088,6 +1092,9 @@ function handleUIMessageStreamFinish({
             startChunk.messageId = messageId;
           }
         }
+        if (chunk.type === "abort") {
+          isAborted = true;
+        }
         controller.enqueue(chunk);
       }
     })
@@ -1113,9 +1120,10 @@ function handleUIMessageStreamFinish({
       transform(chunk, controller) {
         controller.enqueue(chunk);
       },
-      flush() {
+      async flush() {
         const isContinuation = state.message.id === (lastMessage == null ? void 0 : lastMessage.id);
-        onFinish({
+        await onFinish({
+          isAborted,
           isContinuation,
           responseMessage: state.message,
           messages: [
@@ -1214,6 +1222,13 @@ function createStitchableStream() {
   let controller = null;
   let isClosed = false;
   let waitForNewStream = createResolvablePromise();
+  const terminate = () => {
+    isClosed = true;
+    waitForNewStream.resolve();
+    innerStreamReaders.forEach((reader) => reader.cancel());
+    innerStreamReaders = [];
+    controller == null ? void 0 : controller.close();
+  };
   const processPull = async () => {
     if (isClosed && innerStreamReaders.length === 0) {
       controller == null ? void 0 : controller.close();
@@ -1239,9 +1254,7 @@ function createStitchableStream() {
     } catch (error) {
       controller == null ? void 0 : controller.error(error);
       innerStreamReaders.shift();
-      if (isClosed && innerStreamReaders.length === 0) {
-        controller == null ? void 0 : controller.close();
-      }
+      terminate();
     }
   };
   return {
@@ -1280,13 +1293,7 @@ function createStitchableStream() {
      * Immediately close the outer stream. This will cancel all inner streams
      * and close the outer stream.
      */
-    terminate: () => {
-      isClosed = true;
-      waitForNewStream.resolve();
-      innerStreamReaders.forEach((reader) => reader.cancel());
-      innerStreamReaders = [];
-      controller == null ? void 0 : controller.close();
-    }
+    terminate
   };
 }
 
@@ -2646,6 +2653,30 @@ function addTokenCounts(tokenCount1, tokenCount2) {
   return tokenCount1 == null && tokenCount2 == null ? void 0 : (tokenCount1 != null ? tokenCount1 : 0) + (tokenCount2 != null ? tokenCount2 : 0);
 }
 
+// src/util/filter-stream-errors.ts
+function filterStreamErrors(readable, onError) {
+  return new ReadableStream({
+    async start(controller) {
+      const reader = readable.getReader();
+      try {
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) {
+            controller.close();
+            break;
+          }
+          controller.enqueue(value);
+        }
+      } catch (error) {
+        await onError({ error, controller });
+      }
+    },
+    cancel(reason) {
+      return readable.cancel(reason);
+    }
+  });
+}
+
 // src/generate-text/run-tools-transformation.ts
 var import_provider_utils11 = require("@ai-sdk/provider-utils");
 
@@ -3165,6 +3196,7 @@ function streamText({
     console.error(error);
   },
   onFinish,
+  onAbort,
   onStepFinish,
   _internal: {
     now: now2 = now,
@@ -3196,6 +3228,7 @@ function streamText({
     onChunk,
     onError,
     onFinish,
+    onAbort,
     onStepFinish,
     now: now2,
     currentDate,
@@ -3294,6 +3327,7 @@ var DefaultStreamTextResult = class {
     onChunk,
     onError,
     onFinish,
+    onAbort,
     onStepFinish
   }) {
     this._totalUsage = new DelayedPromise();
@@ -3301,7 +3335,6 @@ var DefaultStreamTextResult = class {
     this._steps = new DelayedPromise();
     this.output = output;
     this.includeRawChunks = includeRawChunks;
-    this.generateId = generateId2;
     let stepFinish;
     let recordedContent = [];
     const recordedResponseMessages = [];
@@ -3504,6 +3537,15 @@ var DefaultStreamTextResult = class {
     this.addStream = stitchableStream.addStream;
     this.closeStream = stitchableStream.close;
     let stream = stitchableStream.stream;
+    stream = filterStreamErrors(stream, ({ error, controller }) => {
+      if ((0, import_provider_utils12.isAbortError)(error) && (abortSignal == null ? void 0 : abortSignal.aborted)) {
+        onAbort == null ? void 0 : onAbort({ steps: recordedSteps });
+        controller.enqueue({ type: "abort" });
+        controller.close();
+      } else {
+        controller.error(error);
+      }
+    });
     stream = stream.pipeThrough(
       new TransformStream({
         start(controller) {
@@ -4246,6 +4288,10 @@ var DefaultStreamTextResult = class {
               }
               break;
             }
+            case "abort": {
+              controller.enqueue(part);
+              break;
+            }
             case "tool-input-end": {
               break;
             }
@@ -4266,13 +4312,15 @@ var DefaultStreamTextResult = class {
         }
       })
     );
-    return handleUIMessageStreamFinish({
-      stream: baseStream,
-      messageId: responseMessageId != null ? responseMessageId : generateMessageId == null ? void 0 : generateMessageId(),
-      originalMessages,
-      onFinish,
-      onError
-    });
+    return createAsyncIterableStream(
+      handleUIMessageStreamFinish({
+        stream: baseStream,
+        messageId: responseMessageId != null ? responseMessageId : generateMessageId == null ? void 0 : generateMessageId(),
+        originalMessages,
+        onFinish,
+        onError
+      })
+    );
   }
   pipeUIMessageStreamToResponse(response, {
     originalMessages,