@llmgateway/ai-sdk-provider 3.2.1 → 3.4.0

package/dist/index.mjs

@@ -3585,6 +3585,79 @@ var LLMGateway = class {
   }
 };
 
+// src/image/index.ts
+import { z as z8 } from "zod/v4";
+var LLMGatewayImageResponseSchema = z8.object({
+  data: z8.array(
+    z8.object({
+      b64_json: z8.string().optional(),
+      url: z8.string().optional(),
+      revised_prompt: z8.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;
@@ -3613,6 +3686,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(
@@ -3631,6 +3710,8 @@ function createLLMGateway(options = {}) {
   provider.languageModel = createLanguageModel;
   provider.chat = createChatModel;
   provider.completion = createCompletionModel;
+  provider.image = createImageModel;
+  provider.imageModel = createImageModel;
   return provider;
 }
 var llmgateway = createLLMGateway({