ai 6.0.0-beta.81 → 6.0.0-beta.82

package/dist/index.mjs

@@ -775,7 +775,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.81" : "0.0.0-test";
+var VERSION = true ? "6.0.0-beta.82" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -9534,12 +9534,27 @@ function smoothStream({
   };
 }
 
+// src/middleware/default-embedding-settings-middleware.ts
+function defaultEmbeddingSettingsMiddleware({
+  settings
+}) {
+  return {
+    specificationVersion: "v3",
+    transformParams: async ({ params }) => {
+      return mergeObjects(
+        settings,
+        params
+      );
+    }
+  };
+}
+
 // src/middleware/default-settings-middleware.ts
 function defaultSettingsMiddleware({
   settings
 }) {
   return {
-    middlewareVersion: "v3",
+    specificationVersion: "v3",
     transformParams: async ({ params }) => {
       return mergeObjects(settings, params);
     }
@@ -9573,7 +9588,7 @@ function extractReasoningMiddleware({
   const openingTag = `<${tagName}>`;
   const closingTag = `</${tagName}>`;
   return {
-    middlewareVersion: "v3",
+    specificationVersion: "v3",
     wrapGenerate: async ({ doGenerate }) => {
       const { content, ...rest } = await doGenerate();
       const transformedContent = [];
@@ -9719,7 +9734,7 @@ function extractReasoningMiddleware({
 // src/middleware/simulate-streaming-middleware.ts
 function simulateStreamingMiddleware() {
   return {
-    middlewareVersion: "v3",
+    specificationVersion: "v3",
     wrapStream: async ({ doGenerate }) => {
       const result = await doGenerate();
       let id = 0;
@@ -9840,6 +9855,54 @@ var doWrap = ({
   };
 };
 
+// src/middleware/wrap-embedding-model.ts
+var wrapEmbeddingModel = ({
+  model,
+  middleware: middlewareArg,
+  modelId,
+  providerId
+}) => {
+  return asArray(middlewareArg).reverse().reduce((wrappedModel, middleware) => {
+    return doWrap2({ model: wrappedModel, middleware, modelId, providerId });
+  }, model);
+};
+var doWrap2 = ({
+  model,
+  middleware: {
+    transformParams,
+    wrapEmbed,
+    overrideProvider,
+    overrideModelId,
+    overrideMaxEmbeddingsPerCall,
+    overrideSupportsParallelCalls
+  },
+  modelId,
+  providerId
+}) => {
+  var _a16, _b, _c, _d;
+  async function doTransform({
+    params
+  }) {
+    return transformParams ? await transformParams({ params, model }) : params;
+  }
+  return {
+    specificationVersion: "v3",
+    provider: (_a16 = providerId != null ? providerId : overrideProvider == null ? void 0 : overrideProvider({ model })) != null ? _a16 : model.provider,
+    modelId: (_b = modelId != null ? modelId : overrideModelId == null ? void 0 : overrideModelId({ model })) != null ? _b : model.modelId,
+    maxEmbeddingsPerCall: (_c = overrideMaxEmbeddingsPerCall == null ? void 0 : overrideMaxEmbeddingsPerCall({ model })) != null ? _c : model.maxEmbeddingsPerCall,
+    supportsParallelCalls: (_d = overrideSupportsParallelCalls == null ? void 0 : overrideSupportsParallelCalls({ model })) != null ? _d : model.supportsParallelCalls,
+    async doEmbed(params) {
+      const transformedParams = await doTransform({ params });
+      const doEmbed = async () => model.doEmbed(transformedParams);
+      return wrapEmbed ? wrapEmbed({
+        doEmbed,
+        params: transformedParams,
+        model
+      }) : doEmbed();
+    }
+  };
+};
+
 // src/model/as-image-model-v3.ts
 function asImageModelV3(model) {
   if (model.specificationVersion === "v3") {
@@ -11158,6 +11221,7 @@ export {
   createUIMessageStream,
   createUIMessageStreamResponse,
   customProvider,
+  defaultEmbeddingSettingsMiddleware,
   defaultSettingsMiddleware,
   dynamicTool,
   embed,
@@ -11208,6 +11272,7 @@ export {
   uiMessageChunkSchema,
   userModelMessageSchema,
   validateUIMessages,
+  wrapEmbeddingModel,
   wrapLanguageModel,
   wrapProvider,
   zodSchema3 as zodSchema