ai 4.1.14 → 4.1.16

package/dist/index.mjs

@@ -5701,6 +5701,16 @@ function extractReasoningMiddleware({
 
 // core/middleware/wrap-language-model.ts
 var wrapLanguageModel = ({
+  model,
+  middleware: middlewareArg,
+  modelId,
+  providerId
+}) => {
+  return asArray(middlewareArg).reverse().reduce((wrappedModel, middleware) => {
+    return doWrap({ model: wrappedModel, middleware, modelId, providerId });
+  }, model);
+};
+var doWrap = ({
   model,
   middleware: { transformParams, wrapGenerate, wrapStream },
   modelId,
@@ -5824,9 +5834,10 @@ function appendResponseMessages({
 
 // core/registry/custom-provider.ts
 import { NoSuchModelError as NoSuchModelError2 } from "@ai-sdk/provider";
-function experimental_customProvider({
+function customProvider({
   languageModels,
   textEmbeddingModels,
+  imageModels,
   fallbackProvider
 }) {
   return {
@@ -5847,9 +5858,19 @@ function experimental_customProvider({
         return fallbackProvider.textEmbeddingModel(modelId);
       }
       throw new NoSuchModelError2({ modelId, modelType: "textEmbeddingModel" });
+    },
+    imageModel(modelId) {
+      if (imageModels != null && modelId in imageModels) {
+        return imageModels[modelId];
+      }
+      if (fallbackProvider == null ? void 0 : fallbackProvider.imageModel) {
+        return fallbackProvider.imageModel(modelId);
+      }
+      throw new NoSuchModelError2({ modelId, modelType: "imageModel" });
     }
   };
 }
+var experimental_customProvider = customProvider;
 
 // core/registry/no-such-provider-error.ts
 import { AISDKError as AISDKError15, NoSuchModelError as NoSuchModelError3 } from "@ai-sdk/provider";
@@ -5889,7 +5910,10 @@ var DefaultProviderRegistry = class {
   constructor() {
     this.providers = {};
   }
-  registerProvider({ id, provider }) {
+  registerProvider({
+    id,
+    provider
+  }) {
     this.providers[id] = provider;
   }
   getProvider(id) {
@@ -5937,6 +5961,16 @@ var DefaultProviderRegistry = class {
     }
     return model;
   }
+  imageModel(id) {
+    var _a15;
+    const [providerId, modelId] = this.splitId(id, "imageModel");
+    const provider = this.getProvider(providerId);
+    const model = (_a15 = provider.imageModel) == null ? void 0 : _a15.call(provider, modelId);
+    if (model == null) {
+      throw new NoSuchModelError4({ modelId: id, modelType: "imageModel" });
+    }
+    return model;
+  }
   /**
    * @deprecated Use `textEmbeddingModel` instead.
    */
@@ -6384,6 +6418,7 @@ export {
   createDataStream,
   createDataStreamResponse,
   createIdGenerator5 as createIdGenerator,
+  customProvider,
   embed,
   embedMany,
   experimental_createProviderRegistry,