ai 4.1.14 → 4.1.15

package/dist/index.mjs

@@ -5824,9 +5824,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 +5848,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 +5900,10 @@ var DefaultProviderRegistry = class {
   constructor() {
     this.providers = {};
   }
-  registerProvider({ id, provider }) {
+  registerProvider({
+    id,
+    provider
+  }) {
     this.providers[id] = provider;
   }
   getProvider(id) {
@@ -5937,6 +5951,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 +6408,7 @@ export {
   createDataStream,
   createDataStreamResponse,
   createIdGenerator5 as createIdGenerator,
+  customProvider,
   embed,
   embedMany,
   experimental_createProviderRegistry,