ai 3.2.28 → 3.2.30

package/dist/index.mjs

@@ -1587,6 +1587,94 @@ function toResponseMessages({
 }
 var experimental_generateText = generateText;
 
+// core/util/merge-streams.ts
+function mergeStreams(stream1, stream2) {
+  const reader1 = stream1.getReader();
+  const reader2 = stream2.getReader();
+  let lastRead1 = void 0;
+  let lastRead2 = void 0;
+  let stream1Done = false;
+  let stream2Done = false;
+  async function readStream1(controller) {
+    try {
+      if (lastRead1 == null) {
+        lastRead1 = reader1.read();
+      }
+      const result = await lastRead1;
+      lastRead1 = void 0;
+      if (!result.done) {
+        controller.enqueue(result.value);
+      } else {
+        controller.close();
+      }
+    } catch (error) {
+      controller.error(error);
+    }
+  }
+  async function readStream2(controller) {
+    try {
+      if (lastRead2 == null) {
+        lastRead2 = reader2.read();
+      }
+      const result = await lastRead2;
+      lastRead2 = void 0;
+      if (!result.done) {
+        controller.enqueue(result.value);
+      } else {
+        controller.close();
+      }
+    } catch (error) {
+      controller.error(error);
+    }
+  }
+  return new ReadableStream({
+    async pull(controller) {
+      try {
+        if (stream1Done) {
+          await readStream2(controller);
+          return;
+        }
+        if (stream2Done) {
+          await readStream1(controller);
+          return;
+        }
+        if (lastRead1 == null) {
+          lastRead1 = reader1.read();
+        }
+        if (lastRead2 == null) {
+          lastRead2 = reader2.read();
+        }
+        const { result, reader } = await Promise.race([
+          lastRead1.then((result2) => ({ result: result2, reader: reader1 })),
+          lastRead2.then((result2) => ({ result: result2, reader: reader2 }))
+        ]);
+        if (!result.done) {
+          controller.enqueue(result.value);
+        }
+        if (reader === reader1) {
+          lastRead1 = void 0;
+          if (result.done) {
+            await readStream2(controller);
+            stream1Done = true;
+          }
+        } else {
+          lastRead2 = void 0;
+          if (result.done) {
+            stream2Done = true;
+            await readStream1(controller);
+          }
+        }
+      } catch (error) {
+        controller.error(error);
+      }
+    },
+    cancel() {
+      reader1.cancel();
+      reader2.cancel();
+    }
+  });
+}
+
 // core/generate-text/run-tools-transformation.ts
 import { NoSuchToolError as NoSuchToolError2 } from "@ai-sdk/provider";
 import { generateId } from "@ai-sdk/ui-utils";
@@ -2052,7 +2140,7 @@ var StreamTextResult = class {
           await callbacks.onFinal(aggregatedResponse);
       }
     });
-    const streamDataTransformer = new TransformStream({
+    const streamPartsTransformer = new TransformStream({
       transform: async (chunk, controller) => {
         const chunkType = chunk.type;
         switch (chunkType) {
@@ -2108,7 +2196,7 @@ var StreamTextResult = class {
         }
       }
     });
-    return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamDataTransformer).pipeThrough(new TextEncoderStream());
+    return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamPartsTransformer).pipeThrough(new TextEncoderStream());
   }
   /**
   Writes stream data output to a Node.js response-like object.
@@ -2176,12 +2264,27 @@ var StreamTextResult = class {
   Converts the result to a streamed response object with a stream data part stream.
   It can be used with the `useChat` and `useCompletion` hooks.
   
-  @param init Optional headers.
+  @param options An object with an init property (ResponseInit) and a data property. 
+  You can also pass in a ResponseInit directly (deprecated).
   
   @return A response object.
      */
-  toAIStreamResponse(init) {
-    return new StreamingTextResponse(this.toAIStream(), init);
+  toAIStreamResponse(options) {
+    var _a;
+    const init = options == null ? void 0 : "init" in options ? options.init : {
+      headers: "headers" in options ? options.headers : void 0,
+      status: "status" in options ? options.status : void 0,
+      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();
+    return new Response(stream, {
+      status: (_a = init == null ? void 0 : init.status) != null ? _a : 200,
+      statusText: init == null ? void 0 : init.statusText,
+      headers: prepareResponseHeaders(init, {
+        contentType: "text/plain; charset=utf-8"
+      })
+    });
   }
   /**
   Creates a simple text stream response.
@@ -2625,7 +2728,7 @@ function readableFromAsyncIterable(iterable) {
 
 // streams/stream-data.ts
 import { formatStreamPart as formatStreamPart2 } from "@ai-sdk/ui-utils";
-var StreamData = class {
+var StreamData2 = class {
   constructor() {
     this.encoder = new TextEncoder();
     this.controller = null;
@@ -2696,7 +2799,7 @@ function createStreamDataTransformer() {
     }
   });
 }
-var experimental_StreamData = class extends StreamData {
+var experimental_StreamData = class extends StreamData2 {
 };
 
 // streams/anthropic-stream.ts
@@ -3466,94 +3569,6 @@ async function ReplicateStream(res, cb, options) {
   );
 }
 
-// core/util/merge-streams.ts
-function mergeStreams(stream1, stream2) {
-  const reader1 = stream1.getReader();
-  const reader2 = stream2.getReader();
-  let lastRead1 = void 0;
-  let lastRead2 = void 0;
-  let stream1Done = false;
-  let stream2Done = false;
-  async function readStream1(controller) {
-    try {
-      if (lastRead1 == null) {
-        lastRead1 = reader1.read();
-      }
-      const result = await lastRead1;
-      lastRead1 = void 0;
-      if (!result.done) {
-        controller.enqueue(result.value);
-      } else {
-        controller.close();
-      }
-    } catch (error) {
-      controller.error(error);
-    }
-  }
-  async function readStream2(controller) {
-    try {
-      if (lastRead2 == null) {
-        lastRead2 = reader2.read();
-      }
-      const result = await lastRead2;
-      lastRead2 = void 0;
-      if (!result.done) {
-        controller.enqueue(result.value);
-      } else {
-        controller.close();
-      }
-    } catch (error) {
-      controller.error(error);
-    }
-  }
-  return new ReadableStream({
-    async pull(controller) {
-      try {
-        if (stream1Done) {
-          readStream2(controller);
-          return;
-        }
-        if (stream2Done) {
-          readStream1(controller);
-          return;
-        }
-        if (lastRead1 == null) {
-          lastRead1 = reader1.read();
-        }
-        if (lastRead2 == null) {
-          lastRead2 = reader2.read();
-        }
-        const { result, reader } = await Promise.race([
-          lastRead1.then((result2) => ({ result: result2, reader: reader1 })),
-          lastRead2.then((result2) => ({ result: result2, reader: reader2 }))
-        ]);
-        if (!result.done) {
-          controller.enqueue(result.value);
-        }
-        if (reader === reader1) {
-          lastRead1 = void 0;
-          if (result.done) {
-            readStream2(controller);
-            stream1Done = true;
-          }
-        } else {
-          lastRead2 = void 0;
-          if (result.done) {
-            stream2Done = true;
-            readStream1(controller);
-          }
-        }
-      } catch (error) {
-        controller.error(error);
-      }
-    },
-    cancel() {
-      reader1.cancel();
-      reader2.cancel();
-    }
-  });
-}
-
 // streams/stream-to-response.ts
 function streamToResponse(res, response, init, data) {
   var _a;
@@ -3637,7 +3652,7 @@ export {
   OpenAIStream,
   ReplicateStream,
   RetryError2 as RetryError,
-  StreamData,
+  StreamData2 as StreamData,
   StreamObjectResult,
   StreamTextResult,
   StreamingTextResponse,