@llmgateway/ai-sdk-provider 3.2.1 → 3.4.0

package/dist/index.d.mts

@@ -1,4 +1,4 @@
-import { LanguageModelV2, LanguageModelV2CallOptions, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2ResponseMetadata, SharedV2Headers, LanguageModelV2StreamPart } from '@ai-sdk/provider';
+import { LanguageModelV2, LanguageModelV2CallOptions, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2ResponseMetadata, SharedV2Headers, LanguageModelV2StreamPart, ImageModelV3, ImageModelV3CallOptions, SharedV3Warning } from '@ai-sdk/provider';
 export { LanguageModelV2, LanguageModelV2Prompt } from '@ai-sdk/provider';
 import { models, Provider } from '@llmgateway/models';
 
@@ -132,6 +132,9 @@ type LLMGatewayCompletionSettings = {
     suffix?: string;
 } & LLMGatewaySharedSettings;
 
+type LLMGatewayImageModelId = string;
+type LLMGatewayImageSettings = {};
+
 type LLMGatewayChatConfig = {
     provider: string;
     compatibility: 'strict' | 'compatible';
@@ -209,6 +212,34 @@ declare class LLMGatewayCompletionLanguageModel implements LanguageModelV2 {
     doStream(options: LanguageModelV2CallOptions): Promise<Awaited<ReturnType<LanguageModelV2['doStream']>>>;
 }
 
+type LLMGatewayImageConfig = {
+    provider: string;
+    headers: () => Record<string, string | undefined>;
+    url: (options: {
+        modelId: string;
+        path: string;
+    }) => string;
+    fetch?: typeof fetch;
+};
+declare class LLMGatewayImageModel implements ImageModelV3 {
+    readonly specificationVersion: "v3";
+    readonly provider: string;
+    readonly modelId: LLMGatewayImageModelId;
+    readonly maxImagesPerCall: number | undefined;
+    readonly settings: LLMGatewayImageSettings;
+    private readonly config;
+    constructor(modelId: LLMGatewayImageModelId, settings: LLMGatewayImageSettings, config: LLMGatewayImageConfig);
+    doGenerate(options: ImageModelV3CallOptions): Promise<{
+        images: Array<string>;
+        warnings: Array<SharedV3Warning>;
+        response: {
+            timestamp: Date;
+            modelId: string;
+            headers: Record<string, string> | undefined;
+        };
+    }>;
+}
+
 interface LLMGatewayProvider extends LanguageModelV2 {
     (modelId: LLMGatewayChatModelId, settings?: LLMGatewayCompletionSettings): LLMGatewayCompletionLanguageModel;
     (modelId: LLMGatewayChatModelId, settings?: LLMGatewayChatSettings): LLMGatewayChatLanguageModel;
@@ -222,6 +253,14 @@ interface LLMGatewayProvider extends LanguageModelV2 {
   Creates an LLMGateway completion model for text generation.
      */
     completion(modelId: LLMGatewayChatModelId, settings?: LLMGatewayCompletionSettings): LLMGatewayCompletionLanguageModel;
+    /**
+  Creates an LLMGateway image model for image generation.
+     */
+    image(modelId: LLMGatewayImageModelId, settings?: LLMGatewayImageSettings): LLMGatewayImageModel;
+    /**
+  Creates an LLMGateway image model for image generation.
+     */
+    imageModel(modelId: LLMGatewayImageModelId, settings?: LLMGatewayImageSettings): ImageModelV3;
 }
 interface LLMGatewayProviderSettings {
     /**

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV2, LanguageModelV2CallOptions, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2ResponseMetadata, SharedV2Headers, LanguageModelV2StreamPart } from '@ai-sdk/provider';
+import { LanguageModelV2, LanguageModelV2CallOptions, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2ResponseMetadata, SharedV2Headers, LanguageModelV2StreamPart, ImageModelV3, ImageModelV3CallOptions, SharedV3Warning } from '@ai-sdk/provider';
 export { LanguageModelV2, LanguageModelV2Prompt } from '@ai-sdk/provider';
 import { models, Provider } from '@llmgateway/models';
 
@@ -132,6 +132,9 @@ type LLMGatewayCompletionSettings = {
     suffix?: string;
 } & LLMGatewaySharedSettings;
 
+type LLMGatewayImageModelId = string;
+type LLMGatewayImageSettings = {};
+
 type LLMGatewayChatConfig = {
     provider: string;
     compatibility: 'strict' | 'compatible';
@@ -209,6 +212,34 @@ declare class LLMGatewayCompletionLanguageModel implements LanguageModelV2 {
     doStream(options: LanguageModelV2CallOptions): Promise<Awaited<ReturnType<LanguageModelV2['doStream']>>>;
 }
 
+type LLMGatewayImageConfig = {
+    provider: string;
+    headers: () => Record<string, string | undefined>;
+    url: (options: {
+        modelId: string;
+        path: string;
+    }) => string;
+    fetch?: typeof fetch;
+};
+declare class LLMGatewayImageModel implements ImageModelV3 {
+    readonly specificationVersion: "v3";
+    readonly provider: string;
+    readonly modelId: LLMGatewayImageModelId;
+    readonly maxImagesPerCall: number | undefined;
+    readonly settings: LLMGatewayImageSettings;
+    private readonly config;
+    constructor(modelId: LLMGatewayImageModelId, settings: LLMGatewayImageSettings, config: LLMGatewayImageConfig);
+    doGenerate(options: ImageModelV3CallOptions): Promise<{
+        images: Array<string>;
+        warnings: Array<SharedV3Warning>;
+        response: {
+            timestamp: Date;
+            modelId: string;
+            headers: Record<string, string> | undefined;
+        };
+    }>;
+}
+
 interface LLMGatewayProvider extends LanguageModelV2 {
     (modelId: LLMGatewayChatModelId, settings?: LLMGatewayCompletionSettings): LLMGatewayCompletionLanguageModel;
     (modelId: LLMGatewayChatModelId, settings?: LLMGatewayChatSettings): LLMGatewayChatLanguageModel;
@@ -222,6 +253,14 @@ interface LLMGatewayProvider extends LanguageModelV2 {
   Creates an LLMGateway completion model for text generation.
      */
     completion(modelId: LLMGatewayChatModelId, settings?: LLMGatewayCompletionSettings): LLMGatewayCompletionLanguageModel;
+    /**
+  Creates an LLMGateway image model for image generation.
+     */
+    image(modelId: LLMGatewayImageModelId, settings?: LLMGatewayImageSettings): LLMGatewayImageModel;
+    /**
+  Creates an LLMGateway image model for image generation.
+     */
+    imageModel(modelId: LLMGatewayImageModelId, settings?: LLMGatewayImageSettings): ImageModelV3;
 }
 interface LLMGatewayProviderSettings {
     /**

package/dist/index.js

@@ -3618,6 +3618,79 @@ var LLMGateway = class {
   }
 };
 
+// src/image/index.ts
+var import_v47 = require("zod/v4");
+var LLMGatewayImageResponseSchema = import_v47.z.object({
+  data: import_v47.z.array(
+    import_v47.z.object({
+      b64_json: import_v47.z.string().optional(),
+      url: import_v47.z.string().optional(),
+      revised_prompt: import_v47.z.string().optional()
+    })
+  )
+});
+var LLMGatewayImageModel = class {
+  constructor(modelId, settings, config) {
+    this.specificationVersion = "v3";
+    this.maxImagesPerCall = void 0;
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+    this.provider = config.provider;
+  }
+  async doGenerate(options) {
+    const warnings = [];
+    if (options.seed != null) {
+      warnings.push({
+        type: "unsupported",
+        feature: "seed"
+      });
+    }
+    const body = {
+      model: this.modelId,
+      prompt: options.prompt,
+      n: options.n,
+      response_format: "b64_json"
+    };
+    if (options.size != null) {
+      body.size = options.size;
+    }
+    if (options.aspectRatio != null) {
+      body.aspect_ratio = options.aspectRatio;
+    }
+    const { value: response, responseHeaders } = await postJsonToApi({
+      url: this.config.url({
+        path: "/images/generations",
+        modelId: this.modelId
+      }),
+      headers: combineHeaders(this.config.headers(), options.headers),
+      body,
+      failedResponseHandler: llmgatewayFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler(
+        LLMGatewayImageResponseSchema
+      ),
+      abortSignal: options.abortSignal,
+      fetch: this.config.fetch
+    });
+    return {
+      images: response.data.map((item) => {
+        if (item.b64_json) {
+          return item.b64_json;
+        }
+        throw new Error(
+          "Expected b64_json in response but got url. Set response_format to b64_json."
+        );
+      }),
+      warnings,
+      response: {
+        timestamp: /* @__PURE__ */ new Date(),
+        modelId: this.modelId,
+        headers: responseHeaders
+      }
+    };
+  }
+};
+
 // src/provider.ts
 function createLLMGateway(options = {}) {
   var _a16, _b16, _c, _d;
@@ -3646,6 +3719,12 @@ function createLLMGateway(options = {}) {
     fetch: options.fetch,
     extraBody: options.extraBody
   });
+  const createImageModel = (modelId, settings = {}) => new LLMGatewayImageModel(modelId, settings, {
+    provider: "llmgateway.image",
+    url: ({ path }) => `${baseURL}${path}`,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
   const createLanguageModel = (modelId, settings) => {
     if (new.target) {
       throw new Error(
@@ -3664,6 +3743,8 @@ function createLLMGateway(options = {}) {
   provider.languageModel = createLanguageModel;
   provider.chat = createChatModel;
   provider.completion = createCompletionModel;
+  provider.image = createImageModel;
+  provider.imageModel = createImageModel;
   return provider;
 }
 var llmgateway = createLLMGateway({