ai 3.3.28 → 3.3.30

package/dist/index.mjs

@@ -2079,6 +2079,55 @@ function now() {
 
 // core/generate-object/stream-object.ts
 import { createIdGenerator as createIdGenerator2 } from "@ai-sdk/provider-utils";
+
+// core/util/prepare-outgoing-http-headers.ts
+function prepareOutgoingHttpHeaders(init, {
+  contentType,
+  dataStreamVersion
+}) {
+  const headers = {};
+  if ((init == null ? void 0 : init.headers) != null) {
+    for (const [key, value] of Object.entries(init.headers)) {
+      headers[key] = value;
+    }
+  }
+  if (headers["Content-Type"] == null) {
+    headers["Content-Type"] = contentType;
+  }
+  if (dataStreamVersion !== void 0) {
+    headers["X-Vercel-AI-Data-Stream"] = dataStreamVersion;
+  }
+  return headers;
+}
+
+// core/util/write-to-server-response.ts
+function writeToServerResponse({
+  response,
+  status,
+  statusText,
+  headers,
+  stream
+}) {
+  response.writeHead(status != null ? status : 200, statusText, headers);
+  const reader = stream.getReader();
+  const read = async () => {
+    try {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done)
+          break;
+        response.write(value);
+      }
+    } catch (error) {
+      throw error;
+    } finally {
+      response.end();
+    }
+  };
+  read();
+}
+
+// core/generate-object/stream-object.ts
 var originalGenerateId2 = createIdGenerator2({ prefix: "aiobj-", length: 24 });
 async function streamObject({
   model,
@@ -2557,27 +2606,15 @@ var DefaultStreamObjectResult = class {
     });
   }
   pipeTextStreamToResponse(response, init) {
-    var _a11;
-    response.writeHead((_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200, {
-      "Content-Type": "text/plain; charset=utf-8",
-      ...init == null ? void 0 : init.headers
+    writeToServerResponse({
+      response,
+      status: init == null ? void 0 : init.status,
+      statusText: init == null ? void 0 : init.statusText,
+      headers: prepareOutgoingHttpHeaders(init, {
+        contentType: "text/plain; charset=utf-8"
+      }),
+      stream: this.textStream.pipeThrough(new TextEncoderStream())
     });
-    const reader = this.textStream.pipeThrough(new TextEncoderStream()).getReader();
-    const read = async () => {
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done)
-            break;
-          response.write(value);
-        }
-      } catch (error) {
-        throw error;
-      } finally {
-        response.end();
-      }
-    };
-    read();
   }
   toTextStreamResponse(init) {
     var _a11;
@@ -3946,9 +3983,9 @@ var DefaultStreamTextResult = class {
     });
   }
   toAIStream(callbacks = {}) {
-    return this.toDataStream({ callbacks });
+    return this.toDataStreamInternal({ callbacks });
   }
-  toDataStream({
+  toDataStreamInternal({
     callbacks = {},
     getErrorMessage: getErrorMessage4 = () => ""
     // mask error messages for safety by default
@@ -4056,55 +4093,45 @@ var DefaultStreamTextResult = class {
   pipeAIStreamToResponse(response, init) {
     return this.pipeDataStreamToResponse(response, init);
   }
-  pipeDataStreamToResponse(response, init) {
-    var _a11;
-    response.writeHead((_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200, {
-      "Content-Type": "text/plain; charset=utf-8",
-      ...init == null ? void 0 : init.headers
-    });
-    const reader = this.toDataStream().getReader();
-    const read = async () => {
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done)
-            break;
-          response.write(value);
-        }
-      } catch (error) {
-        throw error;
-      } finally {
-        response.end();
-      }
+  pipeDataStreamToResponse(response, options) {
+    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
     };
-    read();
+    const data = options == null ? void 0 : "data" in options ? options.data : void 0;
+    const getErrorMessage4 = options == null ? void 0 : "getErrorMessage" in options ? options.getErrorMessage : void 0;
+    writeToServerResponse({
+      response,
+      status: init == null ? void 0 : init.status,
+      statusText: init == null ? void 0 : init.statusText,
+      headers: prepareOutgoingHttpHeaders(init, {
+        contentType: "text/plain; charset=utf-8",
+        dataStreamVersion: "v1"
+      }),
+      stream: this.toDataStream({ data, getErrorMessage: getErrorMessage4 })
+    });
   }
   pipeTextStreamToResponse(response, init) {
-    var _a11;
-    response.writeHead((_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200, {
-      "Content-Type": "text/plain; charset=utf-8",
-      ...init == null ? void 0 : init.headers
+    writeToServerResponse({
+      response,
+      status: init == null ? void 0 : init.status,
+      statusText: init == null ? void 0 : init.statusText,
+      headers: prepareOutgoingHttpHeaders(init, {
+        contentType: "text/plain; charset=utf-8"
+      }),
+      stream: this.textStream.pipeThrough(new TextEncoderStream())
     });
-    const reader = this.textStream.pipeThrough(new TextEncoderStream()).getReader();
-    const read = async () => {
-      try {
-        while (true) {
-          const { done, value } = await reader.read();
-          if (done)
-            break;
-          response.write(value);
-        }
-      } catch (error) {
-        throw error;
-      } finally {
-        response.end();
-      }
-    };
-    read();
   }
   toAIStreamResponse(options) {
     return this.toDataStreamResponse(options);
   }
+  toDataStream(options) {
+    const stream = this.toDataStreamInternal({
+      getErrorMessage: options == null ? void 0 : options.getErrorMessage
+    });
+    return (options == null ? void 0 : options.data) ? mergeStreams(options == null ? void 0 : options.data.stream, stream) : stream;
+  }
   toDataStreamResponse(options) {
     var _a11;
     const init = options == null ? void 0 : "init" in options ? options.init : {
@@ -4114,8 +4141,7 @@ var DefaultStreamTextResult = class {
     };
     const data = options == null ? void 0 : "data" in options ? options.data : void 0;
     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, {
+    return new Response(this.toDataStream({ data, getErrorMessage: getErrorMessage4 }), {
       status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
       statusText: init == null ? void 0 : init.statusText,
       headers: prepareResponseHeaders(init, {