ai 4.0.32 → 4.0.33

package/dist/index.mjs

@@ -2714,12 +2714,27 @@ function createStitchableStream() {
       innerStreamReaders.push(innerStream.getReader());
       waitForNewStream.resolve();
     },
+    /**
+     * Gracefully close the outer stream. This will let the inner streams
+     * finish processing and then close the outer stream.
+     */
     close: () => {
       isClosed = true;
       waitForNewStream.resolve();
       if (innerStreamReaders.length === 0) {
         controller == null ? void 0 : controller.close();
       }
+    },
+    /**
+     * 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();
     }
   };
 }
@@ -4419,7 +4434,7 @@ function streamText({
     tools,
     toolChoice,
     toolCallStreaming,
-    transform: transform == null ? void 0 : transform({ tools }),
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4693,7 +4708,14 @@ var DefaultStreamTextResult = class {
     this.closeStream = stitchableStream.close;
     let stream = stitchableStream.stream;
     if (transform) {
-      stream = stream.pipeThrough(transform);
+      stream = stream.pipeThrough(
+        transform({
+          tools,
+          stopStream() {
+            stitchableStream.terminate();
+          }
+        })
+      );
     }
     this.baseStream = stream.pipeThrough(createOutputTransformStream(output)).pipeThrough(eventProcessor);
     const { maxRetries, retry } = prepareRetries({