ai 3.3.4 → 3.3.5

package/dist/index.d.mts

@@ -1554,6 +1554,7 @@ declare class DefaultStreamTextResult<TOOLS extends Record<string, CoreTool>> im
     get textStream(): AsyncIterableStream<string>;
     get fullStream(): AsyncIterableStream<TextStreamPart<TOOLS>>;
     toAIStream(callbacks?: AIStreamCallbacksAndOptions): ReadableStream<Uint8Array>;
+    private toDataStream;
     pipeAIStreamToResponse(response: ServerResponse$1, init?: {
         headers?: Record<string, string>;
         status?: number;
@@ -1573,6 +1574,7 @@ declare class DefaultStreamTextResult<TOOLS extends Record<string, CoreTool>> im
     toDataStreamResponse(options?: ResponseInit | {
         init?: ResponseInit;
         data?: StreamData;
+        getErrorMessage?: (error: unknown) => string;
     }): Response;
     toTextStreamResponse(init?: ResponseInit): Response;
 }

package/dist/index.d.ts

@@ -1554,6 +1554,7 @@ declare class DefaultStreamTextResult<TOOLS extends Record<string, CoreTool>> im
     get textStream(): AsyncIterableStream<string>;
     get fullStream(): AsyncIterableStream<TextStreamPart<TOOLS>>;
     toAIStream(callbacks?: AIStreamCallbacksAndOptions): ReadableStream<Uint8Array>;
+    private toDataStream;
     pipeAIStreamToResponse(response: ServerResponse$1, init?: {
         headers?: Record<string, string>;
         status?: number;
@@ -1573,6 +1574,7 @@ declare class DefaultStreamTextResult<TOOLS extends Record<string, CoreTool>> im
     toDataStreamResponse(options?: ResponseInit | {
         init?: ResponseInit;
         data?: StreamData;
+        getErrorMessage?: (error: unknown) => string;
     }): Response;
     toTextStreamResponse(init?: ResponseInit): Response;
 }

package/dist/index.js

@@ -3159,6 +3159,13 @@ var DefaultStreamTextResult = class {
     });
   }
   toAIStream(callbacks = {}) {
+    return this.toDataStream({ callbacks });
+  }
+  toDataStream({
+    callbacks = {},
+    getErrorMessage: getErrorMessage4 = () => ""
+    // mask error messages for safety by default
+  } = {}) {
     let aggregatedResponse = "";
     const callbackTransformer = new TransformStream({
       async start() {
@@ -3225,7 +3232,7 @@ var DefaultStreamTextResult = class {
             break;
           case "error":
             controller.enqueue(
-              (0, import_ui_utils6.formatStreamPart)("error", JSON.stringify(chunk.error))
+              (0, import_ui_utils6.formatStreamPart)("error", getErrorMessage4(chunk.error))
             );
             break;
           case "finish":
@@ -3257,7 +3264,7 @@ var DefaultStreamTextResult = class {
       "Content-Type": "text/plain; charset=utf-8",
       ...init == null ? void 0 : init.headers
     });
-    const reader = this.toAIStream().getReader();
+    const reader = this.toDataStream().getReader();
     const read = async () => {
       try {
         while (true) {
@@ -3308,7 +3315,8 @@ var DefaultStreamTextResult = class {
       statusText: "statusText" in options ? options.statusText : void 0
     };
     const data = options == null ? void 0 : "data" in options ? options.data : void 0;
-    const stream = data ? mergeStreams(data.stream, this.toAIStream()) : this.toAIStream();
+    const getErrorMessage4 = options == null ? void 0 : "getErrorMessage" in options ? options.getErrorMessage : void 0;
+    const stream = data ? mergeStreams(data.stream, this.toDataStream({ getErrorMessage: getErrorMessage4 })) : this.toDataStream({ getErrorMessage: getErrorMessage4 });
     return new Response(stream, {
       status: (_a9 = init == null ? void 0 : init.status) != null ? _a9 : 200,
       statusText: init == null ? void 0 : init.statusText,