ai 4.1.29 → 4.1.31

package/dist/index.mjs

@@ -2144,9 +2144,7 @@ var arrayOutputStrategy = (schema) => {
                 }
                 case "text-delta":
                 case "finish":
-                  break;
                 case "error":
-                  controller.error(chunk.error);
                   break;
                 default: {
                   const _exhaustiveCheck = chunk;
@@ -2916,6 +2914,7 @@ function streamObject({
   experimental_telemetry: telemetry,
   experimental_providerMetadata,
   providerOptions = experimental_providerMetadata,
+  onError,
   onFinish,
   _internal: {
     generateId: generateId3 = originalGenerateId2,
@@ -2950,6 +2949,7 @@ function streamObject({
     schemaDescription,
     providerOptions,
     mode,
+    onError,
     onFinish,
     generateId: generateId3,
     currentDate,
@@ -2972,6 +2972,7 @@ var DefaultStreamObjectResult = class {
     schemaDescription,
     providerOptions,
     mode,
+    onError,
     onFinish,
     generateId: generateId3,
     currentDate,
@@ -2983,7 +2984,6 @@ var DefaultStreamObjectResult = class {
     this.warningsPromise = new DelayedPromise();
     this.requestPromise = new DelayedPromise();
     this.responsePromise = new DelayedPromise();
-    this.stitchableStream = createStitchableStream();
     const { maxRetries, retry } = prepareRetries({
       maxRetries: maxRetriesArg
     });
@@ -2995,6 +2995,16 @@ var DefaultStreamObjectResult = class {
     });
     const tracer = getTracer(telemetry);
     const self = this;
+    const stitchableStream = createStitchableStream();
+    const eventProcessor = new TransformStream({
+      transform(chunk, controller) {
+        controller.enqueue(chunk);
+        if (chunk.type === "error") {
+          onError == null ? void 0 : onError({ error: chunk.error });
+        }
+      }
+    });
+    this.baseStream = stitchableStream.stream.pipeThrough(eventProcessor);
     recordSpan({
       name: "ai.streamObject",
       attributes: selectTelemetryAttributes({
@@ -3338,25 +3348,26 @@ var DefaultStreamObjectResult = class {
                   experimental_providerMetadata: providerMetadata
                 }));
               } catch (error2) {
-                controller.error(error2);
+                controller.enqueue({ type: "error", error: error2 });
               } finally {
                 rootSpan.end();
               }
             }
           })
         );
-        self.stitchableStream.addStream(transformedStream);
+        stitchableStream.addStream(transformedStream);
       }
     }).catch((error) => {
-      self.stitchableStream.addStream(
+      stitchableStream.addStream(
         new ReadableStream({
           start(controller) {
-            controller.error(error);
+            controller.enqueue({ type: "error", error });
+            controller.close();
           }
         })
       );
     }).finally(() => {
-      self.stitchableStream.close();
+      stitchableStream.close();
     });
     this.outputStrategy = outputStrategy;
   }
@@ -3380,7 +3391,7 @@ var DefaultStreamObjectResult = class {
   }
   get partialObjectStream() {
     return createAsyncIterableStream(
-      this.stitchableStream.stream.pipeThrough(
+      this.baseStream.pipeThrough(
         new TransformStream({
           transform(chunk, controller) {
             switch (chunk.type) {
@@ -3389,9 +3400,7 @@ var DefaultStreamObjectResult = class {
                 break;
               case "text-delta":
               case "finish":
-                break;
               case "error":
-                controller.error(chunk.error);
                 break;
               default: {
                 const _exhaustiveCheck = chunk;
@@ -3404,13 +3413,11 @@ var DefaultStreamObjectResult = class {
     );
   }
   get elementStream() {
-    return this.outputStrategy.createElementStream(
-      this.stitchableStream.stream
-    );
+    return this.outputStrategy.createElementStream(this.baseStream);
   }
   get textStream() {
     return createAsyncIterableStream(
-      this.stitchableStream.stream.pipeThrough(
+      this.baseStream.pipeThrough(
         new TransformStream({
           transform(chunk, controller) {
             switch (chunk.type) {
@@ -3419,9 +3426,7 @@ var DefaultStreamObjectResult = class {
                 break;
               case "object":
               case "finish":
-                break;
               case "error":
-                controller.error(chunk.error);
                 break;
               default: {
                 const _exhaustiveCheck = chunk;
@@ -3434,7 +3439,7 @@ var DefaultStreamObjectResult = class {
     );
   }
   get fullStream() {
-    return createAsyncIterableStream(this.stitchableStream.stream);
+    return createAsyncIterableStream(this.baseStream);
   }
   pipeTextStreamToResponse(response, init) {
     writeToServerResponse({