ai 4.0.32 → 4.0.34

package/dist/index.mjs

@@ -1907,7 +1907,7 @@ var arrayOutputStrategy = (schema) => {
     // be able to generate an array directly:
     // possible future optimization: use arrays directly when model supports grammar-guided generation
     jsonSchema: {
-      $schema: "http://json-schema.org/draft-07/schema#",
+      $schema: "https://json-schema.org/draft/2019-09/schema#",
       type: "object",
       properties: {
         elements: { type: "array", items: itemSchema }
@@ -2019,7 +2019,7 @@ var enumOutputStrategy = (enumValues) => {
     // be able to generate an enum value directly:
     // possible future optimization: use enums directly when model supports top-level enums
     jsonSchema: {
-      $schema: "http://json-schema.org/draft-07/schema#",
+      $schema: "https://json-schema.org/draft/2019-09/schema#",
       type: "object",
       properties: {
         result: { type: "string", enum: enumValues }
@@ -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({