@llmgateway/ai-sdk-provider 3.2.1 → 3.3.0

package/dist/internal/index.d.mts

@@ -1,4 +1,4 @@
-import { LanguageModelV2, LanguageModelV2CallOptions, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2ResponseMetadata, SharedV2Headers, LanguageModelV2StreamPart } from '@ai-sdk/provider';
+import { LanguageModelV2, LanguageModelV2CallOptions, ImageModelV3, ImageModelV3CallOptions, SharedV3Warning, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2ResponseMetadata, SharedV2Headers, LanguageModelV2StreamPart } from '@ai-sdk/provider';
 export { LanguageModelV2, LanguageModelV2Prompt } from '@ai-sdk/provider';
 import { models, Provider } from '@llmgateway/models';
 export { Model, ModelDefinition, Provider, ProviderModelMapping, StabilityLevel, models } from '@llmgateway/models';
@@ -93,6 +93,9 @@ type LLMGatewayCompletionSettings = {
     suffix?: string;
 } & LLMGatewaySharedSettings;
 
+type LLMGatewayImageModelId = string;
+type LLMGatewayImageSettings = {};
+
 type LLMGatewayCompletionConfig = {
     provider: string;
     compatibility: 'strict' | 'compatible';
@@ -118,6 +121,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;
+        };
+    }>;
+}
+
 declare const providers: [{
     readonly id: "llmgateway";
     readonly name: "LLM Gateway";
@@ -399,4 +430,4 @@ declare class LLMGatewayChatLanguageModel implements LanguageModelV2 {
     }>;
 }
 
-export { LLMGatewayChatLanguageModel, type LLMGatewayChatModelId, type LLMGatewayChatSettings, LLMGatewayCompletionLanguageModel, type LLMGatewayCompletionSettings, type LLMGatewayProviderOptions, type LLMGatewaySharedSettings, type LLMGatewayUsageAccounting, type ProviderId, providers };
+export { LLMGatewayChatLanguageModel, type LLMGatewayChatModelId, type LLMGatewayChatSettings, LLMGatewayCompletionLanguageModel, type LLMGatewayCompletionSettings, LLMGatewayImageModel, type LLMGatewayImageModelId, type LLMGatewayImageSettings, type LLMGatewayProviderOptions, type LLMGatewaySharedSettings, type LLMGatewayUsageAccounting, type ProviderId, providers };

package/dist/internal/index.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV2, LanguageModelV2CallOptions, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2ResponseMetadata, SharedV2Headers, LanguageModelV2StreamPart } from '@ai-sdk/provider';
+import { LanguageModelV2, LanguageModelV2CallOptions, ImageModelV3, ImageModelV3CallOptions, SharedV3Warning, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2Usage, LanguageModelV2CallWarning, LanguageModelV2ResponseMetadata, SharedV2Headers, LanguageModelV2StreamPart } from '@ai-sdk/provider';
 export { LanguageModelV2, LanguageModelV2Prompt } from '@ai-sdk/provider';
 import { models, Provider } from '@llmgateway/models';
 export { Model, ModelDefinition, Provider, ProviderModelMapping, StabilityLevel, models } from '@llmgateway/models';
@@ -93,6 +93,9 @@ type LLMGatewayCompletionSettings = {
     suffix?: string;
 } & LLMGatewaySharedSettings;
 
+type LLMGatewayImageModelId = string;
+type LLMGatewayImageSettings = {};
+
 type LLMGatewayCompletionConfig = {
     provider: string;
     compatibility: 'strict' | 'compatible';
@@ -118,6 +121,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;
+        };
+    }>;
+}
+
 declare const providers: [{
     readonly id: "llmgateway";
     readonly name: "LLM Gateway";
@@ -399,4 +430,4 @@ declare class LLMGatewayChatLanguageModel implements LanguageModelV2 {
     }>;
 }
 
-export { LLMGatewayChatLanguageModel, type LLMGatewayChatModelId, type LLMGatewayChatSettings, LLMGatewayCompletionLanguageModel, type LLMGatewayCompletionSettings, type LLMGatewayProviderOptions, type LLMGatewaySharedSettings, type LLMGatewayUsageAccounting, type ProviderId, providers };
+export { LLMGatewayChatLanguageModel, type LLMGatewayChatModelId, type LLMGatewayChatSettings, LLMGatewayCompletionLanguageModel, type LLMGatewayCompletionSettings, LLMGatewayImageModel, type LLMGatewayImageModelId, type LLMGatewayImageSettings, type LLMGatewayProviderOptions, type LLMGatewaySharedSettings, type LLMGatewayUsageAccounting, type ProviderId, providers };

package/dist/internal/index.js

@@ -61,6 +61,7 @@ var index_exports = {};
 __export(index_exports, {
   LLMGatewayChatLanguageModel: () => LLMGatewayChatLanguageModel,
   LLMGatewayCompletionLanguageModel: () => LLMGatewayCompletionLanguageModel,
+  LLMGatewayImageModel: () => LLMGatewayImageModel,
   models: () => import_models.models,
   providers: () => providers
 });
@@ -3543,6 +3544,82 @@ var LLMGatewayCompletionLanguageModel = 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.aspectRatio != null) {
+      warnings.push({
+        type: "unsupported",
+        feature: "aspectRatio"
+      });
+    }
+    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;
+    }
+    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/models/index.ts
 var import_models = require("@llmgateway/models");
 
@@ -3758,6 +3835,7 @@ var providers = [
 0 && (module.exports = {
   LLMGatewayChatLanguageModel,
   LLMGatewayCompletionLanguageModel,
+  LLMGatewayImageModel,
   models,
   providers
 });