ai 4.0.31 → 4.0.33

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # ai
 
+## 4.0.33
+
+### Patch Changes
+
+- 5510ee7: feat (ai/core): add stopStream option to streamText transforms
+
+## 4.0.32
+
+### Patch Changes
+
+- de66619: feat (ai/core): add tool call id to ToolExecution error
+
 ## 4.0.31
 
 ### 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.
@@ -2404,10 +2408,12 @@ declare class ToolExecutionError extends AISDKError {
     private readonly [symbol$5];
     readonly toolName: string;
     readonly toolArgs: JSONValue;
-    constructor({ toolArgs, toolName, cause, message, }: {
+    readonly toolCallId: string;
+    constructor({ toolArgs, toolName, toolCallId, cause, message, }: {
         message?: string;
         toolArgs: JSONValue;
         toolName: string;
+        toolCallId: string;
         cause: unknown;
     });
     static isInstance(error: unknown): error is ToolExecutionError;

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.
@@ -2404,10 +2408,12 @@ declare class ToolExecutionError extends AISDKError {
     private readonly [symbol$5];
     readonly toolName: string;
     readonly toolArgs: JSONValue;
-    constructor({ toolArgs, toolName, cause, message, }: {
+    readonly toolCallId: string;
+    constructor({ toolArgs, toolName, toolCallId, cause, message, }: {
         message?: string;
         toolArgs: JSONValue;
         toolName: string;
+        toolCallId: string;
         cause: unknown;
     });
     static isInstance(error: unknown): error is ToolExecutionError;

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();
     }
   };
 }
@@ -3368,6 +3383,7 @@ var ToolExecutionError = class extends import_provider12.AISDKError {
   constructor({
     toolArgs,
     toolName,
+    toolCallId,
     cause,
     message = `Error executing tool ${toolName}: ${(0, import_provider12.getErrorMessage)(cause)}`
   }) {
@@ -3375,6 +3391,7 @@ var ToolExecutionError = class extends import_provider12.AISDKError {
     this[_a9] = true;
     this.toolArgs = toolArgs;
     this.toolName = toolName;
+    this.toolCallId = toolCallId;
   }
   static isInstance(error) {
     return import_provider12.AISDKError.hasMarker(error, marker9);
@@ -3997,6 +4014,7 @@ async function executeTools({
             return result2;
           } catch (error) {
             throw new ToolExecutionError({
+              toolCallId,
               toolName,
               toolArgs: args,
               cause: error
@@ -4340,6 +4358,7 @@ function runToolsTransformation({
                     toolResultsStreamController.enqueue({
                       type: "error",
                       error: new ToolExecutionError({
+                        toolCallId: toolCall.toolCallId,
                         toolName: toolCall.toolName,
                         toolArgs: toolCall.args,
                         cause: error
@@ -4451,7 +4470,7 @@ function streamText({
     tools,
     toolChoice,
     toolCallStreaming,
-    transform: transform == null ? void 0 : transform({ tools }),
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4725,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({