ai 4.0.32 → 4.0.33

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # ai
 
+## 4.0.33
+
+### Patch Changes
+
+- 5510ee7: feat (ai/core): add stopStream option to streamText transforms
+
 ## 4.0.32
 
 ### Patch Changes

package/dist/index.d.mts

@@ -2134,9 +2134,13 @@ Enable streaming of tool call deltas as they are generated. Disabled by default.
     experimental_toolCallStreaming?: boolean;
     /**
 Optional transformation that is applied to the stream.
+
+@param stopStream - A function that stops the source stream.
+@param tools - The tools that are accessible to and can be called by the model. The model needs to support calling tools.
      */
     experimental_transform?: (options: {
         tools: TOOLS;
+        stopStream: () => void;
     }) => TransformStream<TextStreamPart<TOOLS>, TextStreamPart<TOOLS>>;
     /**
 Callback that is called for each chunk of the stream. The stream processing will pause until the callback promise is resolved.

package/dist/index.d.ts

@@ -2134,9 +2134,13 @@ Enable streaming of tool call deltas as they are generated. Disabled by default.
     experimental_toolCallStreaming?: boolean;
     /**
 Optional transformation that is applied to the stream.
+
+@param stopStream - A function that stops the source stream.
+@param tools - The tools that are accessible to and can be called by the model. The model needs to support calling tools.
      */
     experimental_transform?: (options: {
         tools: TOOLS;
+        stopStream: () => void;
     }) => TransformStream<TextStreamPart<TOOLS>, TextStreamPart<TOOLS>>;
     /**
 Callback that is called for each chunk of the stream. The stream processing will pause until the callback promise is resolved.

package/dist/index.js

@@ -2765,12 +2765,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();
     }
   };
 }
@@ -4455,7 +4470,7 @@ function streamText({
     tools,
     toolChoice,
     toolCallStreaming,
-    transform: transform == null ? void 0 : transform({ tools }),
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4729,7 +4744,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({