ai 6.0.11 → 6.0.12

package/dist/index.mjs

@@ -898,7 +898,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.11" : "0.0.0-test";
+var VERSION = true ? "6.0.12" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -10586,6 +10586,52 @@ var doWrap2 = ({
   };
 };
 
+// src/middleware/wrap-image-model.ts
+var wrapImageModel = ({
+  model,
+  middleware: middlewareArg,
+  modelId,
+  providerId
+}) => {
+  return [...asArray(middlewareArg)].reverse().reduce((wrappedModel, middleware) => {
+    return doWrap3({ model: wrappedModel, middleware, modelId, providerId });
+  }, model);
+};
+var doWrap3 = ({
+  model,
+  middleware: {
+    transformParams,
+    wrapGenerate,
+    overrideProvider,
+    overrideModelId,
+    overrideMaxImagesPerCall
+  },
+  modelId,
+  providerId
+}) => {
+  var _a16, _b, _c;
+  async function doTransform({ params }) {
+    return transformParams ? await transformParams({ params, model }) : params;
+  }
+  const maxImagesPerCallRaw = (_a16 = overrideMaxImagesPerCall == null ? void 0 : overrideMaxImagesPerCall({ model })) != null ? _a16 : model.maxImagesPerCall;
+  const maxImagesPerCall = maxImagesPerCallRaw instanceof Function ? maxImagesPerCallRaw.bind(model) : maxImagesPerCallRaw;
+  return {
+    specificationVersion: "v3",
+    provider: (_b = providerId != null ? providerId : overrideProvider == null ? void 0 : overrideProvider({ model })) != null ? _b : model.provider,
+    modelId: (_c = modelId != null ? modelId : overrideModelId == null ? void 0 : overrideModelId({ model })) != null ? _c : model.modelId,
+    maxImagesPerCall,
+    async doGenerate(params) {
+      const transformedParams = await doTransform({ params });
+      const doGenerate = async () => model.doGenerate(transformedParams);
+      return wrapGenerate ? wrapGenerate({
+        doGenerate,
+        params: transformedParams,
+        model
+      }) : doGenerate();
+    }
+  };
+};
+
 // src/model/as-provider-v3.ts
 function asProviderV3(provider) {
   if ("specificationVersion" in provider && provider.specificationVersion === "v3") {
@@ -10607,7 +10653,8 @@ function asProviderV3(provider) {
 // src/middleware/wrap-provider.ts
 function wrapProvider({
   provider,
-  languageModelMiddleware
+  languageModelMiddleware,
+  imageModelMiddleware
 }) {
   const providerV3 = asProviderV3(provider);
   return {
@@ -10617,7 +10664,13 @@ function wrapProvider({
       middleware: languageModelMiddleware
     }),
     embeddingModel: providerV3.embeddingModel,
-    imageModel: providerV3.imageModel,
+    imageModel: (modelId) => {
+      let model = providerV3.imageModel(modelId);
+      if (imageModelMiddleware != null) {
+        model = wrapImageModel({ model, middleware: imageModelMiddleware });
+      }
+      return model;
+    },
     transcriptionModel: providerV3.transcriptionModel,
     speechModel: providerV3.speechModel,
     rerankingModel: providerV3.rerankingModel
@@ -10729,11 +10782,13 @@ import {
 } from "@ai-sdk/provider";
 function createProviderRegistry(providers, {
   separator = ":",
-  languageModelMiddleware
+  languageModelMiddleware,
+  imageModelMiddleware
 } = {}) {
   const registry = new DefaultProviderRegistry({
     separator,
-    languageModelMiddleware
+    languageModelMiddleware,
+    imageModelMiddleware
   });
   for (const [id, provider] of Object.entries(providers)) {
     registry.registerProvider({ id, provider });
@@ -10744,11 +10799,13 @@ var experimental_createProviderRegistry = createProviderRegistry;
 var DefaultProviderRegistry = class {
   constructor({
     separator,
-    languageModelMiddleware
+    languageModelMiddleware,
+    imageModelMiddleware
   }) {
     this.providers = {};
     this.separator = separator;
     this.languageModelMiddleware = languageModelMiddleware;
+    this.imageModelMiddleware = imageModelMiddleware;
   }
   registerProvider({
     id,
@@ -10814,10 +10871,16 @@ var DefaultProviderRegistry = class {
     var _a16;
     const [providerId, modelId] = this.splitId(id, "imageModel");
     const provider = this.getProvider(providerId, "imageModel");
-    const model = (_a16 = provider.imageModel) == null ? void 0 : _a16.call(provider, modelId);
+    let model = (_a16 = provider.imageModel) == null ? void 0 : _a16.call(provider, modelId);
     if (model == null) {
       throw new NoSuchModelError4({ modelId: id, modelType: "imageModel" });
     }
+    if (this.imageModelMiddleware != null) {
+      model = wrapImageModel({
+        model,
+        middleware: this.imageModelMiddleware
+      });
+    }
     return model;
   }
   transcriptionModel(id) {
@@ -11986,6 +12049,7 @@ export {
   userModelMessageSchema,
   validateUIMessages,
   wrapEmbeddingModel,
+  wrapImageModel,
   wrapLanguageModel,
   wrapProvider,
   zodSchema3 as zodSchema