ai 4.0.31 → 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();
     }
   };
 }
@@ -3317,6 +3332,7 @@ var ToolExecutionError = class extends AISDKError9 {
   constructor({
     toolArgs,
     toolName,
+    toolCallId,
     cause,
     message = `Error executing tool ${toolName}: ${getErrorMessage2(cause)}`
   }) {
@@ -3324,6 +3340,7 @@ var ToolExecutionError = class extends AISDKError9 {
     this[_a9] = true;
     this.toolArgs = toolArgs;
     this.toolName = toolName;
+    this.toolCallId = toolCallId;
   }
   static isInstance(error) {
     return AISDKError9.hasMarker(error, marker9);
@@ -3946,6 +3963,7 @@ async function executeTools({
             return result2;
           } catch (error) {
             throw new ToolExecutionError({
+              toolCallId,
               toolName,
               toolArgs: args,
               cause: error
@@ -4304,6 +4322,7 @@ function runToolsTransformation({
                     toolResultsStreamController.enqueue({
                       type: "error",
                       error: new ToolExecutionError({
+                        toolCallId: toolCall.toolCallId,
                         toolName: toolCall.toolName,
                         toolArgs: toolCall.args,
                         cause: error
@@ -4415,7 +4434,7 @@ function streamText({
     tools,
     toolChoice,
     toolCallStreaming,
-    transform: transform == null ? void 0 : transform({ tools }),
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4689,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({