ai 5.0.0-canary.1 → 5.0.0-canary.2

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # ai
 
+## 5.0.0-canary.2
+
+### Patch Changes
+
+- bd398e4: fix (core): improve error handling in streamText's consumeStream method
+
 ## 5.0.0-canary.1
 
 ### Minor Changes

package/dist/index.d.mts

@@ -2539,6 +2539,9 @@ type DataStreamOptions = {
      */
     experimental_sendStart?: boolean;
 };
+type ConsumeStreamOptions = {
+    onError?: (error: unknown) => void;
+};
 /**
 A result object for accessing different stream types and additional information.
  */
@@ -2659,8 +2662,10 @@ interface StreamTextResult<TOOLS extends ToolSet, PARTIAL_OUTPUT> {
   This is useful to force the stream to finish.
   It effectively removes the backpressure and allows the stream to finish,
   triggering the `onFinish` callback and the promise resolution.
+  
+  If an error occurs, it is passed to the optional `onError` callback.
     */
-    consumeStream(): Promise<void>;
+    consumeStream(options?: ConsumeStreamOptions): Promise<void>;
     /**
     Converts the result to a data stream.
   

package/dist/index.d.ts

@@ -2539,6 +2539,9 @@ type DataStreamOptions = {
      */
     experimental_sendStart?: boolean;
 };
+type ConsumeStreamOptions = {
+    onError?: (error: unknown) => void;
+};
 /**
 A result object for accessing different stream types and additional information.
  */
@@ -2659,8 +2662,10 @@ interface StreamTextResult<TOOLS extends ToolSet, PARTIAL_OUTPUT> {
   This is useful to force the stream to finish.
   It effectively removes the backpressure and allows the stream to finish,
   triggering the `onFinish` callback and the promise resolution.
+  
+  If an error occurs, it is passed to the optional `onError` callback.
     */
-    consumeStream(): Promise<void>;
+    consumeStream(options?: ConsumeStreamOptions): Promise<void>;
     /**
     Converts the result to a data stream.
   

package/dist/index.js

@@ -4751,6 +4751,25 @@ function asArray(value) {
   return value === void 0 ? [] : Array.isArray(value) ? value : [value];
 }
 
+// util/consume-stream.ts
+async function consumeStream({
+  stream,
+  onError
+}) {
+  const reader = stream.getReader();
+  try {
+    while (true) {
+      const { done } = await reader.read();
+      if (done)
+        break;
+    }
+  } catch (error) {
+    onError == null ? void 0 : onError(error);
+  } finally {
+    reader.releaseLock();
+  }
+}
+
 // core/util/merge-streams.ts
 function mergeStreams(stream1, stream2) {
   const reader1 = stream1.getReader();
@@ -5992,9 +6011,15 @@ var DefaultStreamTextResult = class {
       )
     );
   }
-  async consumeStream() {
-    const stream = this.fullStream;
-    for await (const part of stream) {
+  async consumeStream(options) {
+    var _a17;
+    try {
+      await consumeStream({
+        stream: this.fullStream,
+        onError: options == null ? void 0 : options.onError
+      });
+    } catch (error) {
+      (_a17 = options == null ? void 0 : options.onError) == null ? void 0 : _a17.call(options, error);
     }
   }
   get experimental_partialOutputStream() {