ai 3.3.27 → 3.3.29

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;
@@ -4056,51 +4093,35 @@ 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: data ? mergeStreams(data.stream, this.toDataStream({ getErrorMessage: getErrorMessage4 })) : this.toDataStream({ 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);
@@ -4136,6 +4157,39 @@ var DefaultStreamTextResult = class {
 };
 var experimental_streamText = streamText;
 
+// core/middleware/wrap-language-model.ts
+var experimental_wrapLanguageModel = ({
+  model,
+  middleware: { transformParams, wrapGenerate, wrapStream },
+  modelId,
+  providerId
+}) => {
+  async function doTransform({
+    params,
+    type
+  }) {
+    return transformParams ? await transformParams({ params, type }) : params;
+  }
+  return {
+    specificationVersion: "v1",
+    provider: providerId != null ? providerId : model.provider,
+    modelId: modelId != null ? modelId : model.modelId,
+    defaultObjectGenerationMode: model.defaultObjectGenerationMode,
+    supportsImageUrls: model.supportsImageUrls,
+    supportsStructuredOutputs: model.supportsStructuredOutputs,
+    async doGenerate(params) {
+      const transformedParams = await doTransform({ params, type: "generate" });
+      const doGenerate = async () => model.doGenerate(transformedParams);
+      return wrapGenerate ? wrapGenerate({ doGenerate, params: transformedParams, model }) : doGenerate();
+    },
+    async doStream(params) {
+      const transformedParams = await doTransform({ params, type: "stream" });
+      const doStream = async () => model.doStream(transformedParams);
+      return wrapStream ? wrapStream({ doStream, params: transformedParams, model }) : doStream();
+    }
+  };
+};
+
 // core/prompt/attachments-to-parts.ts
 function attachmentsToParts(attachments) {
   var _a11, _b, _c;
@@ -5585,6 +5639,7 @@ export {
   experimental_generateText,
   experimental_streamObject,
   experimental_streamText,
+  experimental_wrapLanguageModel,
   formatStreamPart,
   generateId2 as generateId,
   generateObject,