@runpod/ai-sdk-provider 0.1.1 → 0.2.0

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @runpod/ai-sdk-provider
 
+## 0.2.0
+
+### Minor Changes
+
+- 1fec4a8: added support for image gen:
+  - `qwen/qwen-image`
+  - `bytedance/seedream-3.0`
+  - `black-forest-labs/flux-1-kontext-dev`
+  - `black-forest-labs/flux-1-schnell`
+  - `black-forest-labs/flux-1-dev`
+
 ## 0.1.1
 
 ### Patch Changes

package/README.md

@@ -1,6 +1,6 @@
 # Runpod AI SDK Provider
 
-The **Runpod provider** for the [AI SDK](https://ai-sdk.dev/docs) contains language model support for [Runpod's](https://runpod.io) public endpoints.
+The **Runpod provider** for the [AI SDK](https://ai-sdk.dev/docs) contains language model and image generation support for [Runpod's](https://runpod.io) public endpoints.
 
 ## Installation
 
@@ -37,11 +37,23 @@ import { runpod } from '@runpod/ai-sdk-provider';
 
 ## Supported Models
 
+### Language Models
+
 | Model ID                               | Description                                                         |
 | -------------------------------------- | ------------------------------------------------------------------- |
 | `deep-cogito/deep-cogito-v2-llama-70b` | 70B parameter general-purpose LLM with advanced reasoning           |
 | `qwen/qwen3-32b-awq`                   | 32B parameter multilingual model with strong reasoning capabilities |
 
+### Image Models
+
+| Model ID                               | Description                     | Supported Aspect Ratios |
+| -------------------------------------- | ------------------------------- | ----------------------- |
+| `qwen/qwen-image`                      | Text-to-image generation        | 1:1, 4:3, 3:4           |
+| `bytedance/seedream-3.0`               | Advanced text-to-image model    | 1:1, 4:3, 3:4           |
+| `black-forest-labs/flux-1-kontext-dev` | Context-aware image generation  | 1:1, 4:3, 3:4           |
+| `black-forest-labs/flux-1-schnell`     | Fast image generation (4 steps) | 1:1, 4:3, 3:4           |
+| `black-forest-labs/flux-1-dev`         | High-quality image generation   | 1:1, 4:3, 3:4           |
+
 ## Usage Examples
 
 ### Basic Text Generation
@@ -124,6 +136,60 @@ const { object } = await generateObject({
 console.log(object.recipe);
 ```
 
+### Image Generation
+
+```ts
+import { runpod } from '@runpod/ai-sdk-provider';
+import { experimental_generateImage as generateImage } from 'ai';
+
+const { image } = await generateImage({
+  model: runpod.imageModel('qwen/qwen-image'),
+  prompt: 'A fashion-forward woman in Paris wearing a trench coat',
+  aspectRatio: '4:3',
+});
+
+// With additional parameters
+const { image } = await generateImage({
+  model: runpod.imageModel('qwen/qwen-image'),
+  prompt: 'A sunset over mountains',
+  size: '1328x1328',
+  seed: 42,
+  providerOptions: {
+    runpod: {
+      negative_prompt: 'blurry, low quality',
+      enable_safety_checker: true,
+    },
+  },
+});
+```
+
+#### Provider Options
+
+The Runpod provider supports additional options via `providerOptions.runpod`:
+
+| Option                  | Type      | Default | Description                                         |
+| ----------------------- | --------- | ------- | --------------------------------------------------- |
+| `negative_prompt`       | `string`  | -       | Text describing what you don't want in the image    |
+| `enable_safety_checker` | `boolean` | `true`  | Enable content safety filtering                     |
+| `image`                 | `string`  | -       | Input image URL (required for Flux Kontext models)  |
+| `maxPollAttempts`       | `number`  | `60`    | Maximum polling attempts for async image generation |
+| `pollIntervalMillis`    | `number`  | `5000`  | Polling interval in milliseconds (5 seconds)        |
+
+**Notes**:
+
+- The provider uses strict validation for image parameters. Unsupported aspect ratios (like `16:9`, `9:16`, `3:2`, `2:3`) will throw an `InvalidArgumentError` with a clear message about supported alternatives.
+- Model availability may vary. Some models might be temporarily unavailable or require specific parameters.
+- **Verified working models**: All models tested and confirmed working
+  - `qwen/qwen-image` - Original model (60s, JPEG)
+  - `bytedance/seedream-3.0` - Fast model (17s, JPEG)
+  - `black-forest-labs/flux-1-schnell` - Very fast (2s, PNG)
+  - `black-forest-labs/flux-1-dev` - High quality (6s, PNG)
+  - `black-forest-labs/flux-1-kontext-dev` - Context-aware with input images (38s, PNG)
+- The provider automatically handles different parameter formats:
+  - **Qwen/Seedream**: `size` parameter, `result` response field
+  - **Flux standard**: `width/height` parameters, `image_url` response field
+  - **Flux Kontext**: `size` parameter + `image` input, `image_url` response field
+
 ## Links
 
 - [Runpod](https://runpod.io) - Cloud platform for AI compute

package/dist/index.d.mts

@@ -1,4 +1,4 @@
-import { LanguageModelV2 } from '@ai-sdk/provider';
+import { LanguageModelV2, ImageModelV2 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 export { OpenAICompatibleErrorData as RunPodErrorData } from '@ai-sdk/openai-compatible';
 
@@ -6,6 +6,8 @@ type RunPodChatModelId = 'deep-cogito/deep-cogito-v2-llama-70b' | 'qwen/qwen3-32
 
 type RunPodCompletionModelId = 'deep-cogito/deep-cogito-v2-llama-70b' | 'qwen/qwen3-32b-awq' | (string & {});
 
+type RunpodImageModelId = 'qwen/qwen-image' | 'bytedance/seedream-3.0' | 'black-forest-labs/flux-1-kontext-dev' | 'black-forest-labs/flux-1-schnell' | 'black-forest-labs/flux-1-dev';
+
 interface RunPodProviderSettings {
     /**
   RunPod API key.
@@ -38,8 +40,12 @@ interface RunPodProvider {
   Creates a completion model for text generation.
   */
     completionModel(modelId: RunPodCompletionModelId): LanguageModelV2;
+    /**
+  Creates an image model for image generation.
+  */
+    imageModel(modelId: RunpodImageModelId): ImageModelV2;
 }
 declare function createRunPod(options?: RunPodProviderSettings): RunPodProvider;
 declare const runpod: RunPodProvider;
 
-export { type RunPodChatModelId, type RunPodCompletionModelId, type RunPodProvider, type RunPodProviderSettings, createRunPod, runpod };
+export { type RunPodChatModelId, type RunPodCompletionModelId, type RunPodProvider, type RunPodProviderSettings, type RunpodImageModelId, createRunPod, runpod };

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV2 } from '@ai-sdk/provider';
+import { LanguageModelV2, ImageModelV2 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 export { OpenAICompatibleErrorData as RunPodErrorData } from '@ai-sdk/openai-compatible';
 
@@ -6,6 +6,8 @@ type RunPodChatModelId = 'deep-cogito/deep-cogito-v2-llama-70b' | 'qwen/qwen3-32
 
 type RunPodCompletionModelId = 'deep-cogito/deep-cogito-v2-llama-70b' | 'qwen/qwen3-32b-awq' | (string & {});
 
+type RunpodImageModelId = 'qwen/qwen-image' | 'bytedance/seedream-3.0' | 'black-forest-labs/flux-1-kontext-dev' | 'black-forest-labs/flux-1-schnell' | 'black-forest-labs/flux-1-dev';
+
 interface RunPodProviderSettings {
     /**
   RunPod API key.
@@ -38,8 +40,12 @@ interface RunPodProvider {
   Creates a completion model for text generation.
   */
     completionModel(modelId: RunPodCompletionModelId): LanguageModelV2;
+    /**
+  Creates an image model for image generation.
+  */
+    imageModel(modelId: RunpodImageModelId): ImageModelV2;
 }
 declare function createRunPod(options?: RunPodProviderSettings): RunPodProvider;
 declare const runpod: RunPodProvider;
 
-export { type RunPodChatModelId, type RunPodCompletionModelId, type RunPodProvider, type RunPodProviderSettings, createRunPod, runpod };
+export { type RunPodChatModelId, type RunPodCompletionModelId, type RunPodProvider, type RunPodProviderSettings, type RunpodImageModelId, createRunPod, runpod };

package/dist/index.js

@@ -27,18 +27,307 @@ module.exports = __toCommonJS(index_exports);
 
 // src/runpod-provider.ts
 var import_openai_compatible = require("@ai-sdk/openai-compatible");
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
+
+// src/runpod-image-model.ts
 var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_provider = require("@ai-sdk/provider");
+var import_zod = require("zod");
+var SUPPORTED_ASPECT_RATIOS = {
+  "1:1": "1328*1328",
+  // ✅ Native support
+  "4:3": "1472*1140",
+  // ✅ Native support
+  "3:4": "1140*1472"
+  // ✅ Native support
+};
+var SUPPORTED_SIZES = /* @__PURE__ */ new Set([
+  // Native aspect ratio sizes
+  "1328*1328",
+  // 1:1
+  "1472*1140",
+  // 4:3
+  "1140*1472",
+  // 3:4
+  // Additional validated sizes
+  "512*512",
+  "768*768",
+  "1024*1024",
+  "512*768",
+  "768*512",
+  "1024*768",
+  "768*1024"
+]);
+var RunpodImageModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v2";
+    this.maxImagesPerCall = 1;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate({
+    prompt,
+    n = 1,
+    size,
+    aspectRatio,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    const warnings = [];
+    let runpodSize;
+    if (size) {
+      const runpodSizeCandidate = size.replace("x", "*");
+      if (!SUPPORTED_SIZES.has(runpodSizeCandidate)) {
+        throw new import_provider.InvalidArgumentError({
+          argument: "size",
+          message: `Size ${size} is not supported by Runpod. Supported sizes: ${Array.from(
+            SUPPORTED_SIZES
+          ).map((s) => s.replace("*", "x")).join(", ")}`
+        });
+      }
+      runpodSize = runpodSizeCandidate;
+    } else if (aspectRatio) {
+      if (!SUPPORTED_ASPECT_RATIOS[aspectRatio]) {
+        throw new import_provider.InvalidArgumentError({
+          argument: "aspectRatio",
+          message: `Aspect ratio ${aspectRatio} is not supported by Runpod. Supported aspect ratios: ${Object.keys(SUPPORTED_ASPECT_RATIOS).join(", ")}`
+        });
+      }
+      runpodSize = SUPPORTED_ASPECT_RATIOS[aspectRatio];
+    } else {
+      runpodSize = "1328*1328";
+    }
+    if (n > 1) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "n",
+        details: "Runpod image models only support generating 1 image at a time. Using n=1."
+      });
+    }
+    const currentDate = this.config._internal?.currentDate?.() ?? /* @__PURE__ */ new Date();
+    const inputPayload = this.buildInputPayload(
+      prompt,
+      runpodSize,
+      seed,
+      providerOptions.runpod
+    );
+    const { value: response, responseHeaders } = await (0, import_provider_utils.postJsonToApi)({
+      url: `${this.config.baseURL}/runsync`,
+      headers: (0, import_provider_utils.combineHeaders)(this.config.headers(), headers),
+      body: {
+        input: inputPayload
+      },
+      failedResponseHandler: (0, import_provider_utils.createJsonErrorResponseHandler)({
+        errorSchema: runpodImageErrorSchema,
+        errorToMessage: (data) => data.error ?? "Unknown error"
+      }),
+      successfulResponseHandler: (0, import_provider_utils.createJsonResponseHandler)(
+        runpodImageResponseSchema
+      ),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    const typedResponse = response;
+    if (typedResponse.status === "COMPLETED" && (typedResponse.output?.result || typedResponse.output?.image_url)) {
+      const imageUrl = typedResponse.output.result || typedResponse.output.image_url;
+      const imageData = await this.downloadImage(imageUrl, abortSignal);
+      return {
+        images: [imageData],
+        warnings,
+        response: {
+          timestamp: currentDate,
+          modelId: this.modelId,
+          headers: responseHeaders
+        },
+        providerMetadata: {
+          runpod: {
+            images: [
+              {
+                url: imageUrl,
+                cost: typedResponse.output?.cost
+              }
+            ]
+          }
+        }
+      };
+    } else if (typedResponse.status === "IN_QUEUE" || typedResponse.status === "IN_PROGRESS") {
+      const pollOptions = {
+        maxAttempts: providerOptions.runpod?.maxPollAttempts,
+        pollIntervalMillis: providerOptions.runpod?.pollIntervalMillis
+      };
+      const imageUrl = await this.pollForCompletion(
+        typedResponse.id,
+        abortSignal,
+        pollOptions
+      );
+      const imageData = await this.downloadImage(imageUrl, abortSignal);
+      return {
+        images: [imageData],
+        warnings,
+        response: {
+          timestamp: currentDate,
+          modelId: this.modelId,
+          headers: responseHeaders
+        },
+        providerMetadata: {
+          runpod: {
+            images: [
+              {
+                url: imageUrl,
+                jobId: typedResponse.id
+              }
+            ]
+          }
+        }
+      };
+    } else {
+      throw new Error(`Unexpected response status: ${typedResponse.status}`);
+    }
+  }
+  async downloadImage(imageUrl, abortSignal) {
+    const { value: imageData } = await (0, import_provider_utils.getFromApi)({
+      url: imageUrl,
+      successfulResponseHandler: (0, import_provider_utils.createBinaryResponseHandler)(),
+      failedResponseHandler: (0, import_provider_utils.createJsonErrorResponseHandler)({
+        errorSchema: runpodImageErrorSchema,
+        errorToMessage: (data) => data.error ?? "Failed to download image"
+      }),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    return imageData;
+  }
+  async pollForCompletion(jobId, abortSignal, pollOptions) {
+    const maxAttempts = pollOptions?.maxAttempts ?? 60;
+    const pollInterval = pollOptions?.pollIntervalMillis ?? 5e3;
+    for (let attempt = 0; attempt < maxAttempts; attempt++) {
+      if (abortSignal?.aborted) {
+        throw new Error("Image generation was aborted");
+      }
+      const { value: statusResponse } = await (0, import_provider_utils.getFromApi)({
+        url: `${this.config.baseURL}/status/${jobId}`,
+        headers: this.config.headers(),
+        successfulResponseHandler: (0, import_provider_utils.createJsonResponseHandler)(
+          runpodImageStatusSchema
+        ),
+        failedResponseHandler: (0, import_provider_utils.createJsonErrorResponseHandler)({
+          errorSchema: runpodImageErrorSchema,
+          errorToMessage: (data) => data.error ?? "Failed to check job status"
+        }),
+        abortSignal,
+        fetch: this.config.fetch
+      });
+      const typedStatusResponse = statusResponse;
+      if (typedStatusResponse.status === "COMPLETED" && (typedStatusResponse.output?.result || typedStatusResponse.output?.image_url)) {
+        return typedStatusResponse.output.result || typedStatusResponse.output.image_url;
+      }
+      if (typedStatusResponse.status === "FAILED") {
+        throw new Error(
+          `Image generation failed: ${typedStatusResponse.error || "Unknown error"}`
+        );
+      }
+      await new Promise((resolve) => setTimeout(resolve, pollInterval));
+    }
+    throw new Error(
+      `Image generation timed out after ${maxAttempts} attempts (${maxAttempts * pollInterval / 1e3}s)`
+    );
+  }
+  buildInputPayload(prompt, runpodSize, seed, runpodOptions) {
+    const isFluxModel = this.modelId.includes("flux") || this.modelId.includes("black-forest-labs");
+    if (isFluxModel) {
+      const isKontext = this.modelId.includes("kontext");
+      if (isKontext) {
+        return {
+          prompt,
+          negative_prompt: runpodOptions?.negative_prompt ?? "",
+          seed: seed ?? -1,
+          num_inference_steps: 28,
+          guidance: 2,
+          size: runpodSize,
+          output_format: "png",
+          enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,
+          ...runpodOptions
+          // This will include the 'image' parameter if provided
+        };
+      } else {
+        const [width, height] = runpodSize.split("*").map(Number);
+        return {
+          prompt,
+          negative_prompt: runpodOptions?.negative_prompt ?? "",
+          seed: seed ?? -1,
+          num_inference_steps: this.modelId.includes("schnell") ? 4 : 28,
+          guidance: this.modelId.includes("schnell") ? 7 : 2,
+          width,
+          height,
+          image_format: "png",
+          ...runpodOptions
+        };
+      }
+    }
+    return {
+      prompt,
+      negative_prompt: runpodOptions?.negative_prompt ?? "",
+      size: runpodSize,
+      seed: seed ?? -1,
+      enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,
+      ...runpodOptions
+    };
+  }
+};
+var runpodImageResponseSchema = import_zod.z.object({
+  id: import_zod.z.string(),
+  status: import_zod.z.enum(["COMPLETED", "IN_QUEUE", "IN_PROGRESS", "FAILED"]),
+  delayTime: import_zod.z.number().optional(),
+  executionTime: import_zod.z.number().optional(),
+  output: import_zod.z.object({
+    cost: import_zod.z.number().optional(),
+    result: import_zod.z.string().optional(),
+    // URL to the generated image (Qwen format)
+    image_url: import_zod.z.string().optional()
+    // URL to the generated image (Flux format)
+  }).optional()
+  // Optional for IN_QUEUE/IN_PROGRESS responses
+});
+var runpodImageStatusSchema = import_zod.z.object({
+  id: import_zod.z.string(),
+  status: import_zod.z.enum(["COMPLETED", "IN_QUEUE", "IN_PROGRESS", "FAILED"]),
+  output: import_zod.z.object({
+    cost: import_zod.z.number().optional(),
+    result: import_zod.z.string().optional(),
+    image_url: import_zod.z.string().optional()
+  }).optional(),
+  error: import_zod.z.string().optional()
+  // Error message if FAILED
+});
+var runpodImageErrorSchema = import_zod.z.object({
+  error: import_zod.z.string(),
+  message: import_zod.z.string().optional()
+});
+
+// src/runpod-provider.ts
 var MODEL_ID_TO_ENDPOINT_URL = {
   "deep-cogito/deep-cogito-v2-llama-70b": "https://api.runpod.ai/v2/deep-cogito-v2-llama-70b/openai/v1",
   "qwen/qwen3-32b-awq": "https://api.runpod.ai/v2/qwen3-32b-awq/openai/v1"
 };
+var IMAGE_MODEL_ID_TO_ENDPOINT_URL = {
+  "qwen/qwen-image": "https://api.runpod.ai/v2/qwen-image-t2i",
+  "bytedance/seedream-3.0": "https://api.runpod.ai/v2/seedream-3-0-t2i",
+  "black-forest-labs/flux-1-kontext-dev": "https://api.runpod.ai/v2/black-forest-labs-flux-1-kontext-dev",
+  "black-forest-labs/flux-1-schnell": "https://api.runpod.ai/v2/black-forest-labs-flux-1-schnell",
+  "black-forest-labs/flux-1-dev": "https://api.runpod.ai/v2/black-forest-labs-flux-1-dev"
+};
 var MODEL_ID_TO_OPENAI_NAME = {
   "deep-cogito/deep-cogito-v2-llama-70b": "deepcogito/cogito-v2-preview-llama-70B",
   "qwen/qwen3-32b-awq": "Qwen/Qwen3-32B-AWQ"
 };
 function createRunPod(options = {}) {
   const getHeaders = () => ({
-    Authorization: `Bearer ${(0, import_provider_utils.loadApiKey)({
+    Authorization: `Bearer ${(0, import_provider_utils2.loadApiKey)({
       apiKey: options.apiKey,
       environmentVariableName: "RUNPOD_API_KEY",
       description: "RunPod"
@@ -56,7 +345,7 @@ function createRunPod(options = {}) {
     }
     return {
       provider: `runpod.${modelType}`,
-      url: ({ path }) => `${(0, import_provider_utils.withoutTrailingSlash)(baseURL)}${path}`,
+      url: ({ path }) => `${(0, import_provider_utils2.withoutTrailingSlash)(baseURL)}${path}`,
       headers: getHeaders,
       fetch: options.fetch
     };
@@ -75,10 +364,27 @@ function createRunPod(options = {}) {
       getModelConfig(modelId, "completion")
     );
   };
+  const createImageModel = (modelId) => {
+    const baseURL = IMAGE_MODEL_ID_TO_ENDPOINT_URL[modelId];
+    if (!baseURL) {
+      throw new Error(
+        `Unsupported Runpod image model: ${modelId}. Supported models: ${Object.keys(
+          IMAGE_MODEL_ID_TO_ENDPOINT_URL
+        ).join(", ")}`
+      );
+    }
+    return new RunpodImageModel(modelId, {
+      provider: "runpod.image",
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    });
+  };
   const provider = (modelId) => createChatModel(modelId);
   provider.completionModel = createCompletionModel;
   provider.languageModel = createChatModel;
   provider.chatModel = createChatModel;
+  provider.imageModel = createImageModel;
   return provider;
 }
 var runpod = createRunPod();

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/runpod-provider.ts"],"sourcesContent":["export { createRunPod, runpod } from './runpod-provider';\nexport type { RunPodProvider, RunPodProviderSettings } from './runpod-provider';\nexport type { RunPodChatModelId } from './runpod-chat-options';\nexport type { RunPodCompletionModelId } from './runpod-completion-options';\nexport type { OpenAICompatibleErrorData as RunPodErrorData } from '@ai-sdk/openai-compatible';\n","import { LanguageModelV2 } from '@ai-sdk/provider';\nimport {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n} from '@ai-sdk/openai-compatible';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { RunPodChatModelId } from './runpod-chat-options';\nimport { RunPodCompletionModelId } from './runpod-completion-options';\n\nexport interface RunPodProviderSettings {\n  /**\nRunPod API key.\n*/\n  apiKey?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n*/\n  fetch?: FetchFunction;\n}\n\nexport interface RunPodProvider {\n  /**\nCreates a model for text generation.\n*/\n  (modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a chat model for text generation.\n*/\n  chatModel(modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a chat model for text generation.\n*/\n  languageModel(modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a completion model for text generation.\n*/\n  completionModel(modelId: RunPodCompletionModelId): LanguageModelV2;\n}\n\n// Mapping of RunPod model IDs to their endpoint URLs\nconst MODEL_ID_TO_ENDPOINT_URL: Record<string, string> = {\n  'deep-cogito/deep-cogito-v2-llama-70b':\n    'https://api.runpod.ai/v2/deep-cogito-v2-llama-70b/openai/v1',\n  'qwen/qwen3-32b-awq': 'https://api.runpod.ai/v2/qwen3-32b-awq/openai/v1',\n};\n\n// Mapping of RunPod model IDs to their OpenAI model names\nconst MODEL_ID_TO_OPENAI_NAME: Record<string, string> = {\n  'deep-cogito/deep-cogito-v2-llama-70b':\n    'deepcogito/cogito-v2-preview-llama-70B',\n  'qwen/qwen3-32b-awq': 'Qwen/Qwen3-32B-AWQ',\n};\n\nexport function createRunPod(\n  options: RunPodProviderSettings = {}\n): RunPodProvider {\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'RUNPOD_API_KEY',\n      description: 'RunPod',\n    })}`,\n    ...options.headers,\n  });\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const getModelConfig = (\n    modelId: string,\n    modelType: string\n  ): CommonModelConfig => {\n    const baseURL = MODEL_ID_TO_ENDPOINT_URL[modelId];\n    if (!baseURL) {\n      throw new Error(\n        `Unsupported RunPod model: ${modelId}. Supported models: ${Object.keys(\n          MODEL_ID_TO_ENDPOINT_URL\n        ).join(', ')}`\n      );\n    }\n\n    return {\n      provider: `runpod.${modelType}`,\n      url: ({ path }) => `${withoutTrailingSlash(baseURL)}${path}`,\n      headers: getHeaders,\n      fetch: options.fetch,\n    };\n  };\n\n  const createChatModel = (modelId: RunPodChatModelId) => {\n    const openaiModelName = MODEL_ID_TO_OPENAI_NAME[modelId] || modelId;\n    return new OpenAICompatibleChatLanguageModel(\n      openaiModelName,\n      getModelConfig(modelId, 'chat')\n    );\n  };\n\n  const createCompletionModel = (modelId: RunPodCompletionModelId) => {\n    const openaiModelName = MODEL_ID_TO_OPENAI_NAME[modelId] || modelId;\n    return new OpenAICompatibleCompletionLanguageModel(\n      openaiModelName,\n      getModelConfig(modelId, 'completion')\n    );\n  };\n\n  const provider = (modelId: RunPodChatModelId) => createChatModel(modelId);\n\n  provider.completionModel = createCompletionModel;\n  provider.languageModel = createChatModel;\n  provider.chatModel = createChatModel;\n\n  return provider;\n}\n\nexport const runpod = createRunPod();\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,+BAGO;AACP,4BAIO;AA2CP,IAAM,2BAAmD;AAAA,EACvD,wCACE;AAAA,EACF,sBAAsB;AACxB;AAGA,IAAM,0BAAkD;AAAA,EACtD,wCACE;AAAA,EACF,sBAAsB;AACxB;AAEO,SAAS,aACd,UAAkC,CAAC,GACnB;AAChB,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,kCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AASA,QAAM,iBAAiB,CACrB,SACA,cACsB;AACtB,UAAM,UAAU,yBAAyB,OAAO;AAChD,QAAI,CAAC,SAAS;AACZ,YAAM,IAAI;AAAA,QACR,6BAA6B,OAAO,uBAAuB,OAAO;AAAA,UAChE;AAAA,QACF,EAAE,KAAK,IAAI,CAAC;AAAA,MACd;AAAA,IACF;AAEA,WAAO;AAAA,MACL,UAAU,UAAU,SAAS;AAAA,MAC7B,KAAK,CAAC,EAAE,KAAK,MAAM,OAAG,4CAAqB,OAAO,CAAC,GAAG,IAAI;AAAA,MAC1D,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,IACjB;AAAA,EACF;AAEA,QAAM,kBAAkB,CAAC,YAA+B;AACtD,UAAM,kBAAkB,wBAAwB,OAAO,KAAK;AAC5D,WAAO,IAAI;AAAA,MACT;AAAA,MACA,eAAe,SAAS,MAAM;AAAA,IAChC;AAAA,EACF;AAEA,QAAM,wBAAwB,CAAC,YAAqC;AAClE,UAAM,kBAAkB,wBAAwB,OAAO,KAAK;AAC5D,WAAO,IAAI;AAAA,MACT;AAAA,MACA,eAAe,SAAS,YAAY;AAAA,IACtC;AAAA,EACF;AAEA,QAAM,WAAW,CAAC,YAA+B,gBAAgB,OAAO;AAExE,WAAS,kBAAkB;AAC3B,WAAS,gBAAgB;AACzB,WAAS,YAAY;AAErB,SAAO;AACT;AAEO,IAAM,SAAS,aAAa;","names":[]}
+{"version":3,"sources":["../src/index.ts","../src/runpod-provider.ts","../src/runpod-image-model.ts"],"sourcesContent":["export { createRunPod, runpod } from './runpod-provider';\nexport type { RunPodProvider, RunPodProviderSettings } from './runpod-provider';\nexport type { RunPodChatModelId } from './runpod-chat-options';\nexport type { RunPodCompletionModelId } from './runpod-completion-options';\nexport type { RunpodImageModelId } from './runpod-image-options';\nexport type { OpenAICompatibleErrorData as RunPodErrorData } from '@ai-sdk/openai-compatible';\n","import { LanguageModelV2, ImageModelV2 } from '@ai-sdk/provider';\nimport {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n} from '@ai-sdk/openai-compatible';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { RunPodChatModelId } from './runpod-chat-options';\nimport { RunPodCompletionModelId } from './runpod-completion-options';\nimport { RunpodImageModelId } from './runpod-image-options';\nimport { RunpodImageModel } from './runpod-image-model';\n\nexport interface RunPodProviderSettings {\n  /**\nRunPod API key.\n*/\n  apiKey?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n*/\n  fetch?: FetchFunction;\n}\n\nexport interface RunPodProvider {\n  /**\nCreates a model for text generation.\n*/\n  (modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a chat model for text generation.\n*/\n  chatModel(modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a chat model for text generation.\n*/\n  languageModel(modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a completion model for text generation.\n*/\n  completionModel(modelId: RunPodCompletionModelId): LanguageModelV2;\n\n  /**\nCreates an image model for image generation.\n*/\n  imageModel(modelId: RunpodImageModelId): ImageModelV2;\n}\n\n// Mapping of RunPod model IDs to their endpoint URLs\nconst MODEL_ID_TO_ENDPOINT_URL: Record<string, string> = {\n  'deep-cogito/deep-cogito-v2-llama-70b':\n    'https://api.runpod.ai/v2/deep-cogito-v2-llama-70b/openai/v1',\n  'qwen/qwen3-32b-awq': 'https://api.runpod.ai/v2/qwen3-32b-awq/openai/v1',\n};\n\n// Mapping of RunPod image model IDs to their endpoint URLs\nconst IMAGE_MODEL_ID_TO_ENDPOINT_URL: Record<string, string> = {\n  'qwen/qwen-image': 'https://api.runpod.ai/v2/qwen-image-t2i',\n  'bytedance/seedream-3.0': 'https://api.runpod.ai/v2/seedream-3-0-t2i',\n  'black-forest-labs/flux-1-kontext-dev':\n    'https://api.runpod.ai/v2/black-forest-labs-flux-1-kontext-dev',\n  'black-forest-labs/flux-1-schnell':\n    'https://api.runpod.ai/v2/black-forest-labs-flux-1-schnell',\n  'black-forest-labs/flux-1-dev':\n    'https://api.runpod.ai/v2/black-forest-labs-flux-1-dev',\n};\n\n// Mapping of RunPod model IDs to their OpenAI model names\nconst MODEL_ID_TO_OPENAI_NAME: Record<string, string> = {\n  'deep-cogito/deep-cogito-v2-llama-70b':\n    'deepcogito/cogito-v2-preview-llama-70B',\n  'qwen/qwen3-32b-awq': 'Qwen/Qwen3-32B-AWQ',\n};\n\nexport function createRunPod(\n  options: RunPodProviderSettings = {}\n): RunPodProvider {\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'RUNPOD_API_KEY',\n      description: 'RunPod',\n    })}`,\n    ...options.headers,\n  });\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const getModelConfig = (\n    modelId: string,\n    modelType: string\n  ): CommonModelConfig => {\n    const baseURL = MODEL_ID_TO_ENDPOINT_URL[modelId];\n    if (!baseURL) {\n      throw new Error(\n        `Unsupported RunPod model: ${modelId}. Supported models: ${Object.keys(\n          MODEL_ID_TO_ENDPOINT_URL\n        ).join(', ')}`\n      );\n    }\n\n    return {\n      provider: `runpod.${modelType}`,\n      url: ({ path }) => `${withoutTrailingSlash(baseURL)}${path}`,\n      headers: getHeaders,\n      fetch: options.fetch,\n    };\n  };\n\n  const createChatModel = (modelId: RunPodChatModelId) => {\n    const openaiModelName = MODEL_ID_TO_OPENAI_NAME[modelId] || modelId;\n    return new OpenAICompatibleChatLanguageModel(\n      openaiModelName,\n      getModelConfig(modelId, 'chat')\n    );\n  };\n\n  const createCompletionModel = (modelId: RunPodCompletionModelId) => {\n    const openaiModelName = MODEL_ID_TO_OPENAI_NAME[modelId] || modelId;\n    return new OpenAICompatibleCompletionLanguageModel(\n      openaiModelName,\n      getModelConfig(modelId, 'completion')\n    );\n  };\n\n  const createImageModel = (modelId: RunpodImageModelId) => {\n    const baseURL = IMAGE_MODEL_ID_TO_ENDPOINT_URL[modelId];\n    if (!baseURL) {\n      throw new Error(\n        `Unsupported Runpod image model: ${modelId}. Supported models: ${Object.keys(\n          IMAGE_MODEL_ID_TO_ENDPOINT_URL\n        ).join(', ')}`\n      );\n    }\n\n    return new RunpodImageModel(modelId, {\n      provider: 'runpod.image',\n      baseURL,\n      headers: getHeaders,\n      fetch: options.fetch,\n    });\n  };\n\n  const provider = (modelId: RunPodChatModelId) => createChatModel(modelId);\n\n  provider.completionModel = createCompletionModel;\n  provider.languageModel = createChatModel;\n  provider.chatModel = createChatModel;\n  provider.imageModel = createImageModel;\n\n  return provider;\n}\n\nexport const runpod = createRunPod();\n","import { ImageModelV2, ImageModelV2CallWarning } from '@ai-sdk/provider';\r\nimport {\r\n  combineHeaders,\r\n  createJsonResponseHandler,\r\n  createJsonErrorResponseHandler,\r\n  createBinaryResponseHandler,\r\n  FetchFunction,\r\n  postJsonToApi,\r\n  getFromApi,\r\n} from '@ai-sdk/provider-utils';\r\nimport { InvalidArgumentError } from '@ai-sdk/provider';\r\nimport { z } from 'zod';\r\nimport { RunpodImageModelId } from './runpod-image-options';\r\n\r\ninterface RunpodImageModelConfig {\r\n  provider: string;\r\n  baseURL: string;\r\n  headers: () => Record<string, string>;\r\n  fetch?: FetchFunction;\r\n  _internal?: {\r\n    currentDate?: () => Date;\r\n  };\r\n}\r\n\r\n// Runpod supported aspect ratios (only validated working sizes)\r\nconst SUPPORTED_ASPECT_RATIOS: Record<string, string> = {\r\n  '1:1': '1328*1328', // ✅ Native support\r\n  '4:3': '1472*1140', // ✅ Native support\r\n  '3:4': '1140*1472', // ✅ Native support\r\n};\r\n\r\n// Runpod supported sizes (validated working sizes)\r\nconst SUPPORTED_SIZES = new Set([\r\n  // Native aspect ratio sizes\r\n  '1328*1328', // 1:1\r\n  '1472*1140', // 4:3\r\n  '1140*1472', // 3:4\r\n  // Additional validated sizes\r\n  '512*512',\r\n  '768*768',\r\n  '1024*1024',\r\n  '512*768',\r\n  '768*512',\r\n  '1024*768',\r\n  '768*1024',\r\n]);\r\n\r\nexport class RunpodImageModel implements ImageModelV2 {\r\n  readonly specificationVersion = 'v2';\r\n  readonly maxImagesPerCall = 1;\r\n\r\n  get provider(): string {\r\n    return this.config.provider;\r\n  }\r\n\r\n  constructor(\r\n    readonly modelId: RunpodImageModelId,\r\n    private config: RunpodImageModelConfig\r\n  ) {}\r\n\r\n  async doGenerate({\r\n    prompt,\r\n    n = 1,\r\n    size,\r\n    aspectRatio,\r\n    seed,\r\n    providerOptions,\r\n    headers,\r\n    abortSignal,\r\n  }: Parameters<ImageModelV2['doGenerate']>[0]): Promise<\r\n    Awaited<ReturnType<ImageModelV2['doGenerate']>>\r\n  > {\r\n    const warnings: Array<ImageModelV2CallWarning> = [];\r\n\r\n    // Determine the size to use\r\n    let runpodSize: string;\r\n\r\n    if (size) {\r\n      // Convert AI SDK format \"1328x1328\" to Runpod format \"1328*1328\"\r\n      const runpodSizeCandidate = size.replace('x', '*');\r\n\r\n      // Validate size is supported\r\n      if (!SUPPORTED_SIZES.has(runpodSizeCandidate)) {\r\n        throw new InvalidArgumentError({\r\n          argument: 'size',\r\n          message: `Size ${size} is not supported by Runpod. Supported sizes: ${Array.from(\r\n            SUPPORTED_SIZES\r\n          )\r\n            .map((s) => s.replace('*', 'x'))\r\n            .join(', ')}`,\r\n        });\r\n      }\r\n\r\n      runpodSize = runpodSizeCandidate;\r\n    } else if (aspectRatio) {\r\n      // Validate aspect ratio is supported\r\n      if (!SUPPORTED_ASPECT_RATIOS[aspectRatio]) {\r\n        throw new InvalidArgumentError({\r\n          argument: 'aspectRatio',\r\n          message: `Aspect ratio ${aspectRatio} is not supported by Runpod. Supported aspect ratios: ${Object.keys(SUPPORTED_ASPECT_RATIOS).join(', ')}`,\r\n        });\r\n      }\r\n\r\n      // Use supported aspect ratio mapping\r\n      runpodSize = SUPPORTED_ASPECT_RATIOS[aspectRatio];\r\n    } else {\r\n      // Default to square format\r\n      runpodSize = '1328*1328';\r\n    }\r\n\r\n    // Handle multiple images warning\r\n    if (n > 1) {\r\n      warnings.push({\r\n        type: 'unsupported-setting',\r\n        setting: 'n',\r\n        details:\r\n          'Runpod image models only support generating 1 image at a time. Using n=1.',\r\n      });\r\n    }\r\n\r\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\r\n\r\n    // Runpod uses a different request format - /runsync endpoint with input wrapper\r\n    const inputPayload = this.buildInputPayload(\r\n      prompt,\r\n      runpodSize,\r\n      seed,\r\n      providerOptions.runpod\r\n    );\r\n\r\n    const { value: response, responseHeaders } = await postJsonToApi({\r\n      url: `${this.config.baseURL}/runsync`,\r\n      headers: combineHeaders(this.config.headers(), headers),\r\n      body: {\r\n        input: inputPayload,\r\n      },\r\n      failedResponseHandler: createJsonErrorResponseHandler({\r\n        errorSchema: runpodImageErrorSchema as any,\r\n        errorToMessage: (data: any) => data.error ?? 'Unknown error',\r\n      }),\r\n      successfulResponseHandler: createJsonResponseHandler(\r\n        runpodImageResponseSchema as any\r\n      ),\r\n      abortSignal,\r\n      fetch: this.config.fetch,\r\n    });\r\n\r\n    // Handle both sync and async responses from Runpod\r\n    const typedResponse = response as any;\r\n    if (\r\n      typedResponse.status === 'COMPLETED' &&\r\n      (typedResponse.output?.result || typedResponse.output?.image_url)\r\n    ) {\r\n      // Sync response - image is ready\r\n      // Different models use different response formats: result vs image_url\r\n      const imageUrl =\r\n        typedResponse.output.result || typedResponse.output.image_url;\r\n      const imageData = await this.downloadImage(imageUrl, abortSignal);\r\n\r\n      return {\r\n        images: [imageData],\r\n        warnings,\r\n        response: {\r\n          timestamp: currentDate,\r\n          modelId: this.modelId,\r\n          headers: responseHeaders,\r\n        },\r\n        providerMetadata: {\r\n          runpod: {\r\n            images: [\r\n              {\r\n                url: imageUrl,\r\n                cost: typedResponse.output?.cost,\r\n              },\r\n            ],\r\n          },\r\n        },\r\n      };\r\n    } else if (\r\n      typedResponse.status === 'IN_QUEUE' ||\r\n      typedResponse.status === 'IN_PROGRESS'\r\n    ) {\r\n      // Async response - need to poll for completion\r\n      const pollOptions = {\r\n        maxAttempts: providerOptions.runpod?.maxPollAttempts as number,\r\n        pollIntervalMillis: providerOptions.runpod\r\n          ?.pollIntervalMillis as number,\r\n      };\r\n      const imageUrl = await this.pollForCompletion(\r\n        typedResponse.id,\r\n        abortSignal,\r\n        pollOptions\r\n      );\r\n      const imageData = await this.downloadImage(imageUrl, abortSignal);\r\n\r\n      return {\r\n        images: [imageData],\r\n        warnings,\r\n        response: {\r\n          timestamp: currentDate,\r\n          modelId: this.modelId,\r\n          headers: responseHeaders,\r\n        },\r\n        providerMetadata: {\r\n          runpod: {\r\n            images: [\r\n              {\r\n                url: imageUrl,\r\n                jobId: typedResponse.id,\r\n              },\r\n            ],\r\n          },\r\n        },\r\n      };\r\n    } else {\r\n      throw new Error(`Unexpected response status: ${typedResponse.status}`);\r\n    }\r\n  }\r\n\r\n  private async downloadImage(\r\n    imageUrl: string,\r\n    abortSignal?: AbortSignal\r\n  ): Promise<Uint8Array> {\r\n    const { value: imageData } = await getFromApi({\r\n      url: imageUrl,\r\n      successfulResponseHandler: createBinaryResponseHandler(),\r\n      failedResponseHandler: createJsonErrorResponseHandler({\r\n        errorSchema: runpodImageErrorSchema as any,\r\n        errorToMessage: (data: any) => data.error ?? 'Failed to download image',\r\n      }),\r\n      abortSignal,\r\n      fetch: this.config.fetch,\r\n    });\r\n    return imageData;\r\n  }\r\n\r\n  private async pollForCompletion(\r\n    jobId: string,\r\n    abortSignal?: AbortSignal,\r\n    pollOptions?: { maxAttempts?: number; pollIntervalMillis?: number }\r\n  ): Promise<string> {\r\n    const maxAttempts = pollOptions?.maxAttempts ?? 60; // 5 minutes with 5-second intervals\r\n    const pollInterval = pollOptions?.pollIntervalMillis ?? 5000; // 5 seconds\r\n\r\n    for (let attempt = 0; attempt < maxAttempts; attempt++) {\r\n      if (abortSignal?.aborted) {\r\n        throw new Error('Image generation was aborted');\r\n      }\r\n\r\n      const { value: statusResponse } = await getFromApi({\r\n        url: `${this.config.baseURL}/status/${jobId}`,\r\n        headers: this.config.headers(),\r\n        successfulResponseHandler: createJsonResponseHandler(\r\n          runpodImageStatusSchema as any\r\n        ),\r\n        failedResponseHandler: createJsonErrorResponseHandler({\r\n          errorSchema: runpodImageErrorSchema as any,\r\n          errorToMessage: (data: any) =>\r\n            data.error ?? 'Failed to check job status',\r\n        }),\r\n        abortSignal,\r\n        fetch: this.config.fetch,\r\n      });\r\n\r\n      const typedStatusResponse = statusResponse as any;\r\n      if (\r\n        typedStatusResponse.status === 'COMPLETED' &&\r\n        (typedStatusResponse.output?.result ||\r\n          typedStatusResponse.output?.image_url)\r\n      ) {\r\n        return (\r\n          typedStatusResponse.output.result ||\r\n          typedStatusResponse.output.image_url\r\n        );\r\n      }\r\n\r\n      if (typedStatusResponse.status === 'FAILED') {\r\n        throw new Error(\r\n          `Image generation failed: ${typedStatusResponse.error || 'Unknown error'}`\r\n        );\r\n      }\r\n\r\n      // Wait before next poll\r\n      await new Promise((resolve) => setTimeout(resolve, pollInterval));\r\n    }\r\n\r\n    throw new Error(\r\n      `Image generation timed out after ${maxAttempts} attempts (${(maxAttempts * pollInterval) / 1000}s)`\r\n    );\r\n  }\r\n\r\n  private buildInputPayload(\r\n    prompt: string,\r\n    runpodSize: string,\r\n    seed?: number,\r\n    runpodOptions?: Record<string, unknown>\r\n  ): Record<string, unknown> {\r\n    // Check if this is a Flux model that uses different parameters\r\n    const isFluxModel =\r\n      this.modelId.includes('flux') ||\r\n      this.modelId.includes('black-forest-labs');\r\n\r\n    if (isFluxModel) {\r\n      // Check if this is Flux Kontext (uses different parameters)\r\n      const isKontext = this.modelId.includes('kontext');\r\n\r\n      if (isKontext) {\r\n        // Flux Kontext uses size format and has image input\r\n        return {\r\n          prompt,\r\n          negative_prompt: runpodOptions?.negative_prompt ?? '',\r\n          seed: seed ?? -1,\r\n          num_inference_steps: 28,\r\n          guidance: 2,\r\n          size: runpodSize,\r\n          output_format: 'png',\r\n          enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,\r\n          ...runpodOptions, // This will include the 'image' parameter if provided\r\n        };\r\n      } else {\r\n        // Regular Flux models use width/height\r\n        const [width, height] = runpodSize.split('*').map(Number);\r\n\r\n        return {\r\n          prompt,\r\n          negative_prompt: runpodOptions?.negative_prompt ?? '',\r\n          seed: seed ?? -1,\r\n          num_inference_steps: this.modelId.includes('schnell') ? 4 : 28,\r\n          guidance: this.modelId.includes('schnell') ? 7 : 2,\r\n          width,\r\n          height,\r\n          image_format: 'png',\r\n          ...runpodOptions,\r\n        };\r\n      }\r\n    }\r\n\r\n    // Default format for Qwen and other models\r\n    return {\r\n      prompt,\r\n      negative_prompt: runpodOptions?.negative_prompt ?? '',\r\n      size: runpodSize,\r\n      seed: seed ?? -1,\r\n      enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,\r\n      ...runpodOptions,\r\n    };\r\n  }\r\n}\r\n\r\n// Runpod image API response schema (handles both sync and async responses)\r\nconst runpodImageResponseSchema = z.object({\r\n  id: z.string(),\r\n  status: z.enum(['COMPLETED', 'IN_QUEUE', 'IN_PROGRESS', 'FAILED']),\r\n  delayTime: z.number().optional(),\r\n  executionTime: z.number().optional(),\r\n  output: z\r\n    .object({\r\n      cost: z.number().optional(),\r\n      result: z.string().optional(), // URL to the generated image (Qwen format)\r\n      image_url: z.string().optional(), // URL to the generated image (Flux format)\r\n    })\r\n    .optional(), // Optional for IN_QUEUE/IN_PROGRESS responses\r\n});\r\n\r\n// Schema for polling status endpoint\r\nconst runpodImageStatusSchema = z.object({\r\n  id: z.string(),\r\n  status: z.enum(['COMPLETED', 'IN_QUEUE', 'IN_PROGRESS', 'FAILED']),\r\n  output: z\r\n    .object({\r\n      cost: z.number().optional(),\r\n      result: z.string().optional(),\r\n      image_url: z.string().optional(),\r\n    })\r\n    .optional(),\r\n  error: z.string().optional(), // Error message if FAILED\r\n});\r\n\r\n// Runpod image API error schema\r\nconst runpodImageErrorSchema = z.object({\r\n  error: z.string(),\r\n  message: z.string().optional(),\r\n});\r\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,+BAGO;AACP,IAAAA,yBAIO;;;ACRP,4BAQO;AACP,sBAAqC;AACrC,iBAAkB;AAclB,IAAM,0BAAkD;AAAA,EACtD,OAAO;AAAA;AAAA,EACP,OAAO;AAAA;AAAA,EACP,OAAO;AAAA;AACT;AAGA,IAAM,kBAAkB,oBAAI,IAAI;AAAA;AAAA,EAE9B;AAAA;AAAA,EACA;AAAA;AAAA,EACA;AAAA;AAAA;AAAA,EAEA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAEM,IAAM,mBAAN,MAA+C;AAAA,EAQpD,YACW,SACD,QACR;AAFS;AACD;AATV,SAAS,uBAAuB;AAChC,SAAS,mBAAmB;AAAA,EASzB;AAAA,EAPH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAOA,MAAM,WAAW;AAAA,IACf;AAAA,IACA,IAAI;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AACA,UAAM,WAA2C,CAAC;AAGlD,QAAI;AAEJ,QAAI,MAAM;AAER,YAAM,sBAAsB,KAAK,QAAQ,KAAK,GAAG;AAGjD,UAAI,CAAC,gBAAgB,IAAI,mBAAmB,GAAG;AAC7C,cAAM,IAAI,qCAAqB;AAAA,UAC7B,UAAU;AAAA,UACV,SAAS,QAAQ,IAAI,iDAAiD,MAAM;AAAA,YAC1E;AAAA,UACF,EACG,IAAI,CAAC,MAAM,EAAE,QAAQ,KAAK,GAAG,CAAC,EAC9B,KAAK,IAAI,CAAC;AAAA,QACf,CAAC;AAAA,MACH;AAEA,mBAAa;AAAA,IACf,WAAW,aAAa;AAEtB,UAAI,CAAC,wBAAwB,WAAW,GAAG;AACzC,cAAM,IAAI,qCAAqB;AAAA,UAC7B,UAAU;AAAA,UACV,SAAS,gBAAgB,WAAW,yDAAyD,OAAO,KAAK,uBAAuB,EAAE,KAAK,IAAI,CAAC;AAAA,QAC9I,CAAC;AAAA,MACH;AAGA,mBAAa,wBAAwB,WAAW;AAAA,IAClD,OAAO;AAEL,mBAAa;AAAA,IACf;AAGA,QAAI,IAAI,GAAG;AACT,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SACE;AAAA,MACJ,CAAC;AAAA,IACH;AAEA,UAAM,cAAc,KAAK,OAAO,WAAW,cAAc,KAAK,oBAAI,KAAK;AAGvE,UAAM,eAAe,KAAK;AAAA,MACxB;AAAA,MACA;AAAA,MACA;AAAA,MACA,gBAAgB;AAAA,IAClB;AAEA,UAAM,EAAE,OAAO,UAAU,gBAAgB,IAAI,UAAM,qCAAc;AAAA,MAC/D,KAAK,GAAG,KAAK,OAAO,OAAO;AAAA,MAC3B,aAAS,sCAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;AAAA,MACtD,MAAM;AAAA,QACJ,OAAO;AAAA,MACT;AAAA,MACA,2BAAuB,sDAA+B;AAAA,QACpD,aAAa;AAAA,QACb,gBAAgB,CAAC,SAAc,KAAK,SAAS;AAAA,MAC/C,CAAC;AAAA,MACD,+BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,MACA;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAGD,UAAM,gBAAgB;AACtB,QACE,cAAc,WAAW,gBACxB,cAAc,QAAQ,UAAU,cAAc,QAAQ,YACvD;AAGA,YAAM,WACJ,cAAc,OAAO,UAAU,cAAc,OAAO;AACtD,YAAM,YAAY,MAAM,KAAK,cAAc,UAAU,WAAW;AAEhE,aAAO;AAAA,QACL,QAAQ,CAAC,SAAS;AAAA,QAClB;AAAA,QACA,UAAU;AAAA,UACR,WAAW;AAAA,UACX,SAAS,KAAK;AAAA,UACd,SAAS;AAAA,QACX;AAAA,QACA,kBAAkB;AAAA,UAChB,QAAQ;AAAA,YACN,QAAQ;AAAA,cACN;AAAA,gBACE,KAAK;AAAA,gBACL,MAAM,cAAc,QAAQ;AAAA,cAC9B;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,WACE,cAAc,WAAW,cACzB,cAAc,WAAW,eACzB;AAEA,YAAM,cAAc;AAAA,QAClB,aAAa,gBAAgB,QAAQ;AAAA,QACrC,oBAAoB,gBAAgB,QAChC;AAAA,MACN;AACA,YAAM,WAAW,MAAM,KAAK;AAAA,QAC1B,cAAc;AAAA,QACd;AAAA,QACA;AAAA,MACF;AACA,YAAM,YAAY,MAAM,KAAK,cAAc,UAAU,WAAW;AAEhE,aAAO;AAAA,QACL,QAAQ,CAAC,SAAS;AAAA,QAClB;AAAA,QACA,UAAU;AAAA,UACR,WAAW;AAAA,UACX,SAAS,KAAK;AAAA,UACd,SAAS;AAAA,QACX;AAAA,QACA,kBAAkB;AAAA,UAChB,QAAQ;AAAA,YACN,QAAQ;AAAA,cACN;AAAA,gBACE,KAAK;AAAA,gBACL,OAAO,cAAc;AAAA,cACvB;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,OAAO;AACL,YAAM,IAAI,MAAM,+BAA+B,cAAc,MAAM,EAAE;AAAA,IACvE;AAAA,EACF;AAAA,EAEA,MAAc,cACZ,UACA,aACqB;AACrB,UAAM,EAAE,OAAO,UAAU,IAAI,UAAM,kCAAW;AAAA,MAC5C,KAAK;AAAA,MACL,+BAA2B,mDAA4B;AAAA,MACvD,2BAAuB,sDAA+B;AAAA,QACpD,aAAa;AAAA,QACb,gBAAgB,CAAC,SAAc,KAAK,SAAS;AAAA,MAC/C,CAAC;AAAA,MACD;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,MAAc,kBACZ,OACA,aACA,aACiB;AACjB,UAAM,cAAc,aAAa,eAAe;AAChD,UAAM,eAAe,aAAa,sBAAsB;AAExD,aAAS,UAAU,GAAG,UAAU,aAAa,WAAW;AACtD,UAAI,aAAa,SAAS;AACxB,cAAM,IAAI,MAAM,8BAA8B;AAAA,MAChD;AAEA,YAAM,EAAE,OAAO,eAAe,IAAI,UAAM,kCAAW;AAAA,QACjD,KAAK,GAAG,KAAK,OAAO,OAAO,WAAW,KAAK;AAAA,QAC3C,SAAS,KAAK,OAAO,QAAQ;AAAA,QAC7B,+BAA2B;AAAA,UACzB;AAAA,QACF;AAAA,QACA,2BAAuB,sDAA+B;AAAA,UACpD,aAAa;AAAA,UACb,gBAAgB,CAAC,SACf,KAAK,SAAS;AAAA,QAClB,CAAC;AAAA,QACD;AAAA,QACA,OAAO,KAAK,OAAO;AAAA,MACrB,CAAC;AAED,YAAM,sBAAsB;AAC5B,UACE,oBAAoB,WAAW,gBAC9B,oBAAoB,QAAQ,UAC3B,oBAAoB,QAAQ,YAC9B;AACA,eACE,oBAAoB,OAAO,UAC3B,oBAAoB,OAAO;AAAA,MAE/B;AAEA,UAAI,oBAAoB,WAAW,UAAU;AAC3C,cAAM,IAAI;AAAA,UACR,4BAA4B,oBAAoB,SAAS,eAAe;AAAA,QAC1E;AAAA,MACF;AAGA,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,YAAY,CAAC;AAAA,IAClE;AAEA,UAAM,IAAI;AAAA,MACR,oCAAoC,WAAW,cAAe,cAAc,eAAgB,GAAI;AAAA,IAClG;AAAA,EACF;AAAA,EAEQ,kBACN,QACA,YACA,MACA,eACyB;AAEzB,UAAM,cACJ,KAAK,QAAQ,SAAS,MAAM,KAC5B,KAAK,QAAQ,SAAS,mBAAmB;AAE3C,QAAI,aAAa;AAEf,YAAM,YAAY,KAAK,QAAQ,SAAS,SAAS;AAEjD,UAAI,WAAW;AAEb,eAAO;AAAA,UACL;AAAA,UACA,iBAAiB,eAAe,mBAAmB;AAAA,UACnD,MAAM,QAAQ;AAAA,UACd,qBAAqB;AAAA,UACrB,UAAU;AAAA,UACV,MAAM;AAAA,UACN,eAAe;AAAA,UACf,uBAAuB,eAAe,yBAAyB;AAAA,UAC/D,GAAG;AAAA;AAAA,QACL;AAAA,MACF,OAAO;AAEL,cAAM,CAAC,OAAO,MAAM,IAAI,WAAW,MAAM,GAAG,EAAE,IAAI,MAAM;AAExD,eAAO;AAAA,UACL;AAAA,UACA,iBAAiB,eAAe,mBAAmB;AAAA,UACnD,MAAM,QAAQ;AAAA,UACd,qBAAqB,KAAK,QAAQ,SAAS,SAAS,IAAI,IAAI;AAAA,UAC5D,UAAU,KAAK,QAAQ,SAAS,SAAS,IAAI,IAAI;AAAA,UACjD;AAAA,UACA;AAAA,UACA,cAAc;AAAA,UACd,GAAG;AAAA,QACL;AAAA,MACF;AAAA,IACF;AAGA,WAAO;AAAA,MACL;AAAA,MACA,iBAAiB,eAAe,mBAAmB;AAAA,MACnD,MAAM;AAAA,MACN,MAAM,QAAQ;AAAA,MACd,uBAAuB,eAAe,yBAAyB;AAAA,MAC/D,GAAG;AAAA,IACL;AAAA,EACF;AACF;AAGA,IAAM,4BAA4B,aAAE,OAAO;AAAA,EACzC,IAAI,aAAE,OAAO;AAAA,EACb,QAAQ,aAAE,KAAK,CAAC,aAAa,YAAY,eAAe,QAAQ,CAAC;AAAA,EACjE,WAAW,aAAE,OAAO,EAAE,SAAS;AAAA,EAC/B,eAAe,aAAE,OAAO,EAAE,SAAS;AAAA,EACnC,QAAQ,aACL,OAAO;AAAA,IACN,MAAM,aAAE,OAAO,EAAE,SAAS;AAAA,IAC1B,QAAQ,aAAE,OAAO,EAAE,SAAS;AAAA;AAAA,IAC5B,WAAW,aAAE,OAAO,EAAE,SAAS;AAAA;AAAA,EACjC,CAAC,EACA,SAAS;AAAA;AACd,CAAC;AAGD,IAAM,0BAA0B,aAAE,OAAO;AAAA,EACvC,IAAI,aAAE,OAAO;AAAA,EACb,QAAQ,aAAE,KAAK,CAAC,aAAa,YAAY,eAAe,QAAQ,CAAC;AAAA,EACjE,QAAQ,aACL,OAAO;AAAA,IACN,MAAM,aAAE,OAAO,EAAE,SAAS;AAAA,IAC1B,QAAQ,aAAE,OAAO,EAAE,SAAS;AAAA,IAC5B,WAAW,aAAE,OAAO,EAAE,SAAS;AAAA,EACjC,CAAC,EACA,SAAS;AAAA,EACZ,OAAO,aAAE,OAAO,EAAE,SAAS;AAAA;AAC7B,CAAC;AAGD,IAAM,yBAAyB,aAAE,OAAO;AAAA,EACtC,OAAO,aAAE,OAAO;AAAA,EAChB,SAAS,aAAE,OAAO,EAAE,SAAS;AAC/B,CAAC;;;ADnUD,IAAM,2BAAmD;AAAA,EACvD,wCACE;AAAA,EACF,sBAAsB;AACxB;AAGA,IAAM,iCAAyD;AAAA,EAC7D,mBAAmB;AAAA,EACnB,0BAA0B;AAAA,EAC1B,wCACE;AAAA,EACF,oCACE;AAAA,EACF,gCACE;AACJ;AAGA,IAAM,0BAAkD;AAAA,EACtD,wCACE;AAAA,EACF,sBAAsB;AACxB;AAEO,SAAS,aACd,UAAkC,CAAC,GACnB;AAChB,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,mCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AASA,QAAM,iBAAiB,CACrB,SACA,cACsB;AACtB,UAAM,UAAU,yBAAyB,OAAO;AAChD,QAAI,CAAC,SAAS;AACZ,YAAM,IAAI;AAAA,QACR,6BAA6B,OAAO,uBAAuB,OAAO;AAAA,UAChE;AAAA,QACF,EAAE,KAAK,IAAI,CAAC;AAAA,MACd;AAAA,IACF;AAEA,WAAO;AAAA,MACL,UAAU,UAAU,SAAS;AAAA,MAC7B,KAAK,CAAC,EAAE,KAAK,MAAM,OAAG,6CAAqB,OAAO,CAAC,GAAG,IAAI;AAAA,MAC1D,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,IACjB;AAAA,EACF;AAEA,QAAM,kBAAkB,CAAC,YAA+B;AACtD,UAAM,kBAAkB,wBAAwB,OAAO,KAAK;AAC5D,WAAO,IAAI;AAAA,MACT;AAAA,MACA,eAAe,SAAS,MAAM;AAAA,IAChC;AAAA,EACF;AAEA,QAAM,wBAAwB,CAAC,YAAqC;AAClE,UAAM,kBAAkB,wBAAwB,OAAO,KAAK;AAC5D,WAAO,IAAI;AAAA,MACT;AAAA,MACA,eAAe,SAAS,YAAY;AAAA,IACtC;AAAA,EACF;AAEA,QAAM,mBAAmB,CAAC,YAAgC;AACxD,UAAM,UAAU,+BAA+B,OAAO;AACtD,QAAI,CAAC,SAAS;AACZ,YAAM,IAAI;AAAA,QACR,mCAAmC,OAAO,uBAAuB,OAAO;AAAA,UACtE;AAAA,QACF,EAAE,KAAK,IAAI,CAAC;AAAA,MACd;AAAA,IACF;AAEA,WAAO,IAAI,iBAAiB,SAAS;AAAA,MACnC,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,IACjB,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CAAC,YAA+B,gBAAgB,OAAO;AAExE,WAAS,kBAAkB;AAC3B,WAAS,gBAAgB;AACzB,WAAS,YAAY;AACrB,WAAS,aAAa;AAEtB,SAAO;AACT;AAEO,IAAM,SAAS,aAAa;","names":["import_provider_utils"]}

package/dist/index.mjs

@@ -7,10 +7,306 @@ import {
   loadApiKey,
   withoutTrailingSlash
 } from "@ai-sdk/provider-utils";
+
+// src/runpod-image-model.ts
+import {
+  combineHeaders,
+  createJsonResponseHandler,
+  createJsonErrorResponseHandler,
+  createBinaryResponseHandler,
+  postJsonToApi,
+  getFromApi
+} from "@ai-sdk/provider-utils";
+import { InvalidArgumentError } from "@ai-sdk/provider";
+import { z } from "zod";
+var SUPPORTED_ASPECT_RATIOS = {
+  "1:1": "1328*1328",
+  // ✅ Native support
+  "4:3": "1472*1140",
+  // ✅ Native support
+  "3:4": "1140*1472"
+  // ✅ Native support
+};
+var SUPPORTED_SIZES = /* @__PURE__ */ new Set([
+  // Native aspect ratio sizes
+  "1328*1328",
+  // 1:1
+  "1472*1140",
+  // 4:3
+  "1140*1472",
+  // 3:4
+  // Additional validated sizes
+  "512*512",
+  "768*768",
+  "1024*1024",
+  "512*768",
+  "768*512",
+  "1024*768",
+  "768*1024"
+]);
+var RunpodImageModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v2";
+    this.maxImagesPerCall = 1;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate({
+    prompt,
+    n = 1,
+    size,
+    aspectRatio,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    const warnings = [];
+    let runpodSize;
+    if (size) {
+      const runpodSizeCandidate = size.replace("x", "*");
+      if (!SUPPORTED_SIZES.has(runpodSizeCandidate)) {
+        throw new InvalidArgumentError({
+          argument: "size",
+          message: `Size ${size} is not supported by Runpod. Supported sizes: ${Array.from(
+            SUPPORTED_SIZES
+          ).map((s) => s.replace("*", "x")).join(", ")}`
+        });
+      }
+      runpodSize = runpodSizeCandidate;
+    } else if (aspectRatio) {
+      if (!SUPPORTED_ASPECT_RATIOS[aspectRatio]) {
+        throw new InvalidArgumentError({
+          argument: "aspectRatio",
+          message: `Aspect ratio ${aspectRatio} is not supported by Runpod. Supported aspect ratios: ${Object.keys(SUPPORTED_ASPECT_RATIOS).join(", ")}`
+        });
+      }
+      runpodSize = SUPPORTED_ASPECT_RATIOS[aspectRatio];
+    } else {
+      runpodSize = "1328*1328";
+    }
+    if (n > 1) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "n",
+        details: "Runpod image models only support generating 1 image at a time. Using n=1."
+      });
+    }
+    const currentDate = this.config._internal?.currentDate?.() ?? /* @__PURE__ */ new Date();
+    const inputPayload = this.buildInputPayload(
+      prompt,
+      runpodSize,
+      seed,
+      providerOptions.runpod
+    );
+    const { value: response, responseHeaders } = await postJsonToApi({
+      url: `${this.config.baseURL}/runsync`,
+      headers: combineHeaders(this.config.headers(), headers),
+      body: {
+        input: inputPayload
+      },
+      failedResponseHandler: createJsonErrorResponseHandler({
+        errorSchema: runpodImageErrorSchema,
+        errorToMessage: (data) => data.error ?? "Unknown error"
+      }),
+      successfulResponseHandler: createJsonResponseHandler(
+        runpodImageResponseSchema
+      ),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    const typedResponse = response;
+    if (typedResponse.status === "COMPLETED" && (typedResponse.output?.result || typedResponse.output?.image_url)) {
+      const imageUrl = typedResponse.output.result || typedResponse.output.image_url;
+      const imageData = await this.downloadImage(imageUrl, abortSignal);
+      return {
+        images: [imageData],
+        warnings,
+        response: {
+          timestamp: currentDate,
+          modelId: this.modelId,
+          headers: responseHeaders
+        },
+        providerMetadata: {
+          runpod: {
+            images: [
+              {
+                url: imageUrl,
+                cost: typedResponse.output?.cost
+              }
+            ]
+          }
+        }
+      };
+    } else if (typedResponse.status === "IN_QUEUE" || typedResponse.status === "IN_PROGRESS") {
+      const pollOptions = {
+        maxAttempts: providerOptions.runpod?.maxPollAttempts,
+        pollIntervalMillis: providerOptions.runpod?.pollIntervalMillis
+      };
+      const imageUrl = await this.pollForCompletion(
+        typedResponse.id,
+        abortSignal,
+        pollOptions
+      );
+      const imageData = await this.downloadImage(imageUrl, abortSignal);
+      return {
+        images: [imageData],
+        warnings,
+        response: {
+          timestamp: currentDate,
+          modelId: this.modelId,
+          headers: responseHeaders
+        },
+        providerMetadata: {
+          runpod: {
+            images: [
+              {
+                url: imageUrl,
+                jobId: typedResponse.id
+              }
+            ]
+          }
+        }
+      };
+    } else {
+      throw new Error(`Unexpected response status: ${typedResponse.status}`);
+    }
+  }
+  async downloadImage(imageUrl, abortSignal) {
+    const { value: imageData } = await getFromApi({
+      url: imageUrl,
+      successfulResponseHandler: createBinaryResponseHandler(),
+      failedResponseHandler: createJsonErrorResponseHandler({
+        errorSchema: runpodImageErrorSchema,
+        errorToMessage: (data) => data.error ?? "Failed to download image"
+      }),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    return imageData;
+  }
+  async pollForCompletion(jobId, abortSignal, pollOptions) {
+    const maxAttempts = pollOptions?.maxAttempts ?? 60;
+    const pollInterval = pollOptions?.pollIntervalMillis ?? 5e3;
+    for (let attempt = 0; attempt < maxAttempts; attempt++) {
+      if (abortSignal?.aborted) {
+        throw new Error("Image generation was aborted");
+      }
+      const { value: statusResponse } = await getFromApi({
+        url: `${this.config.baseURL}/status/${jobId}`,
+        headers: this.config.headers(),
+        successfulResponseHandler: createJsonResponseHandler(
+          runpodImageStatusSchema
+        ),
+        failedResponseHandler: createJsonErrorResponseHandler({
+          errorSchema: runpodImageErrorSchema,
+          errorToMessage: (data) => data.error ?? "Failed to check job status"
+        }),
+        abortSignal,
+        fetch: this.config.fetch
+      });
+      const typedStatusResponse = statusResponse;
+      if (typedStatusResponse.status === "COMPLETED" && (typedStatusResponse.output?.result || typedStatusResponse.output?.image_url)) {
+        return typedStatusResponse.output.result || typedStatusResponse.output.image_url;
+      }
+      if (typedStatusResponse.status === "FAILED") {
+        throw new Error(
+          `Image generation failed: ${typedStatusResponse.error || "Unknown error"}`
+        );
+      }
+      await new Promise((resolve) => setTimeout(resolve, pollInterval));
+    }
+    throw new Error(
+      `Image generation timed out after ${maxAttempts} attempts (${maxAttempts * pollInterval / 1e3}s)`
+    );
+  }
+  buildInputPayload(prompt, runpodSize, seed, runpodOptions) {
+    const isFluxModel = this.modelId.includes("flux") || this.modelId.includes("black-forest-labs");
+    if (isFluxModel) {
+      const isKontext = this.modelId.includes("kontext");
+      if (isKontext) {
+        return {
+          prompt,
+          negative_prompt: runpodOptions?.negative_prompt ?? "",
+          seed: seed ?? -1,
+          num_inference_steps: 28,
+          guidance: 2,
+          size: runpodSize,
+          output_format: "png",
+          enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,
+          ...runpodOptions
+          // This will include the 'image' parameter if provided
+        };
+      } else {
+        const [width, height] = runpodSize.split("*").map(Number);
+        return {
+          prompt,
+          negative_prompt: runpodOptions?.negative_prompt ?? "",
+          seed: seed ?? -1,
+          num_inference_steps: this.modelId.includes("schnell") ? 4 : 28,
+          guidance: this.modelId.includes("schnell") ? 7 : 2,
+          width,
+          height,
+          image_format: "png",
+          ...runpodOptions
+        };
+      }
+    }
+    return {
+      prompt,
+      negative_prompt: runpodOptions?.negative_prompt ?? "",
+      size: runpodSize,
+      seed: seed ?? -1,
+      enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,
+      ...runpodOptions
+    };
+  }
+};
+var runpodImageResponseSchema = z.object({
+  id: z.string(),
+  status: z.enum(["COMPLETED", "IN_QUEUE", "IN_PROGRESS", "FAILED"]),
+  delayTime: z.number().optional(),
+  executionTime: z.number().optional(),
+  output: z.object({
+    cost: z.number().optional(),
+    result: z.string().optional(),
+    // URL to the generated image (Qwen format)
+    image_url: z.string().optional()
+    // URL to the generated image (Flux format)
+  }).optional()
+  // Optional for IN_QUEUE/IN_PROGRESS responses
+});
+var runpodImageStatusSchema = z.object({
+  id: z.string(),
+  status: z.enum(["COMPLETED", "IN_QUEUE", "IN_PROGRESS", "FAILED"]),
+  output: z.object({
+    cost: z.number().optional(),
+    result: z.string().optional(),
+    image_url: z.string().optional()
+  }).optional(),
+  error: z.string().optional()
+  // Error message if FAILED
+});
+var runpodImageErrorSchema = z.object({
+  error: z.string(),
+  message: z.string().optional()
+});
+
+// src/runpod-provider.ts
 var MODEL_ID_TO_ENDPOINT_URL = {
   "deep-cogito/deep-cogito-v2-llama-70b": "https://api.runpod.ai/v2/deep-cogito-v2-llama-70b/openai/v1",
   "qwen/qwen3-32b-awq": "https://api.runpod.ai/v2/qwen3-32b-awq/openai/v1"
 };
+var IMAGE_MODEL_ID_TO_ENDPOINT_URL = {
+  "qwen/qwen-image": "https://api.runpod.ai/v2/qwen-image-t2i",
+  "bytedance/seedream-3.0": "https://api.runpod.ai/v2/seedream-3-0-t2i",
+  "black-forest-labs/flux-1-kontext-dev": "https://api.runpod.ai/v2/black-forest-labs-flux-1-kontext-dev",
+  "black-forest-labs/flux-1-schnell": "https://api.runpod.ai/v2/black-forest-labs-flux-1-schnell",
+  "black-forest-labs/flux-1-dev": "https://api.runpod.ai/v2/black-forest-labs-flux-1-dev"
+};
 var MODEL_ID_TO_OPENAI_NAME = {
   "deep-cogito/deep-cogito-v2-llama-70b": "deepcogito/cogito-v2-preview-llama-70B",
   "qwen/qwen3-32b-awq": "Qwen/Qwen3-32B-AWQ"
@@ -54,10 +350,27 @@ function createRunPod(options = {}) {
       getModelConfig(modelId, "completion")
     );
   };
+  const createImageModel = (modelId) => {
+    const baseURL = IMAGE_MODEL_ID_TO_ENDPOINT_URL[modelId];
+    if (!baseURL) {
+      throw new Error(
+        `Unsupported Runpod image model: ${modelId}. Supported models: ${Object.keys(
+          IMAGE_MODEL_ID_TO_ENDPOINT_URL
+        ).join(", ")}`
+      );
+    }
+    return new RunpodImageModel(modelId, {
+      provider: "runpod.image",
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    });
+  };
   const provider = (modelId) => createChatModel(modelId);
   provider.completionModel = createCompletionModel;
   provider.languageModel = createChatModel;
   provider.chatModel = createChatModel;
+  provider.imageModel = createImageModel;
   return provider;
 }
 var runpod = createRunPod();

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/runpod-provider.ts"],"sourcesContent":["import { LanguageModelV2 } from '@ai-sdk/provider';\nimport {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n} from '@ai-sdk/openai-compatible';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { RunPodChatModelId } from './runpod-chat-options';\nimport { RunPodCompletionModelId } from './runpod-completion-options';\n\nexport interface RunPodProviderSettings {\n  /**\nRunPod API key.\n*/\n  apiKey?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n*/\n  fetch?: FetchFunction;\n}\n\nexport interface RunPodProvider {\n  /**\nCreates a model for text generation.\n*/\n  (modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a chat model for text generation.\n*/\n  chatModel(modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a chat model for text generation.\n*/\n  languageModel(modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a completion model for text generation.\n*/\n  completionModel(modelId: RunPodCompletionModelId): LanguageModelV2;\n}\n\n// Mapping of RunPod model IDs to their endpoint URLs\nconst MODEL_ID_TO_ENDPOINT_URL: Record<string, string> = {\n  'deep-cogito/deep-cogito-v2-llama-70b':\n    'https://api.runpod.ai/v2/deep-cogito-v2-llama-70b/openai/v1',\n  'qwen/qwen3-32b-awq': 'https://api.runpod.ai/v2/qwen3-32b-awq/openai/v1',\n};\n\n// Mapping of RunPod model IDs to their OpenAI model names\nconst MODEL_ID_TO_OPENAI_NAME: Record<string, string> = {\n  'deep-cogito/deep-cogito-v2-llama-70b':\n    'deepcogito/cogito-v2-preview-llama-70B',\n  'qwen/qwen3-32b-awq': 'Qwen/Qwen3-32B-AWQ',\n};\n\nexport function createRunPod(\n  options: RunPodProviderSettings = {}\n): RunPodProvider {\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'RUNPOD_API_KEY',\n      description: 'RunPod',\n    })}`,\n    ...options.headers,\n  });\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const getModelConfig = (\n    modelId: string,\n    modelType: string\n  ): CommonModelConfig => {\n    const baseURL = MODEL_ID_TO_ENDPOINT_URL[modelId];\n    if (!baseURL) {\n      throw new Error(\n        `Unsupported RunPod model: ${modelId}. Supported models: ${Object.keys(\n          MODEL_ID_TO_ENDPOINT_URL\n        ).join(', ')}`\n      );\n    }\n\n    return {\n      provider: `runpod.${modelType}`,\n      url: ({ path }) => `${withoutTrailingSlash(baseURL)}${path}`,\n      headers: getHeaders,\n      fetch: options.fetch,\n    };\n  };\n\n  const createChatModel = (modelId: RunPodChatModelId) => {\n    const openaiModelName = MODEL_ID_TO_OPENAI_NAME[modelId] || modelId;\n    return new OpenAICompatibleChatLanguageModel(\n      openaiModelName,\n      getModelConfig(modelId, 'chat')\n    );\n  };\n\n  const createCompletionModel = (modelId: RunPodCompletionModelId) => {\n    const openaiModelName = MODEL_ID_TO_OPENAI_NAME[modelId] || modelId;\n    return new OpenAICompatibleCompletionLanguageModel(\n      openaiModelName,\n      getModelConfig(modelId, 'completion')\n    );\n  };\n\n  const provider = (modelId: RunPodChatModelId) => createChatModel(modelId);\n\n  provider.completionModel = createCompletionModel;\n  provider.languageModel = createChatModel;\n  provider.chatModel = createChatModel;\n\n  return provider;\n}\n\nexport const runpod = createRunPod();\n"],"mappings":";AACA;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AA2CP,IAAM,2BAAmD;AAAA,EACvD,wCACE;AAAA,EACF,sBAAsB;AACxB;AAGA,IAAM,0BAAkD;AAAA,EACtD,wCACE;AAAA,EACF,sBAAsB;AACxB;AAEO,SAAS,aACd,UAAkC,CAAC,GACnB;AAChB,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AASA,QAAM,iBAAiB,CACrB,SACA,cACsB;AACtB,UAAM,UAAU,yBAAyB,OAAO;AAChD,QAAI,CAAC,SAAS;AACZ,YAAM,IAAI;AAAA,QACR,6BAA6B,OAAO,uBAAuB,OAAO;AAAA,UAChE;AAAA,QACF,EAAE,KAAK,IAAI,CAAC;AAAA,MACd;AAAA,IACF;AAEA,WAAO;AAAA,MACL,UAAU,UAAU,SAAS;AAAA,MAC7B,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,qBAAqB,OAAO,CAAC,GAAG,IAAI;AAAA,MAC1D,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,IACjB;AAAA,EACF;AAEA,QAAM,kBAAkB,CAAC,YAA+B;AACtD,UAAM,kBAAkB,wBAAwB,OAAO,KAAK;AAC5D,WAAO,IAAI;AAAA,MACT;AAAA,MACA,eAAe,SAAS,MAAM;AAAA,IAChC;AAAA,EACF;AAEA,QAAM,wBAAwB,CAAC,YAAqC;AAClE,UAAM,kBAAkB,wBAAwB,OAAO,KAAK;AAC5D,WAAO,IAAI;AAAA,MACT;AAAA,MACA,eAAe,SAAS,YAAY;AAAA,IACtC;AAAA,EACF;AAEA,QAAM,WAAW,CAAC,YAA+B,gBAAgB,OAAO;AAExE,WAAS,kBAAkB;AAC3B,WAAS,gBAAgB;AACzB,WAAS,YAAY;AAErB,SAAO;AACT;AAEO,IAAM,SAAS,aAAa;","names":[]}
+{"version":3,"sources":["../src/runpod-provider.ts","../src/runpod-image-model.ts"],"sourcesContent":["import { LanguageModelV2, ImageModelV2 } from '@ai-sdk/provider';\nimport {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n} from '@ai-sdk/openai-compatible';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { RunPodChatModelId } from './runpod-chat-options';\nimport { RunPodCompletionModelId } from './runpod-completion-options';\nimport { RunpodImageModelId } from './runpod-image-options';\nimport { RunpodImageModel } from './runpod-image-model';\n\nexport interface RunPodProviderSettings {\n  /**\nRunPod API key.\n*/\n  apiKey?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n*/\n  fetch?: FetchFunction;\n}\n\nexport interface RunPodProvider {\n  /**\nCreates a model for text generation.\n*/\n  (modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a chat model for text generation.\n*/\n  chatModel(modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a chat model for text generation.\n*/\n  languageModel(modelId: RunPodChatModelId): LanguageModelV2;\n\n  /**\nCreates a completion model for text generation.\n*/\n  completionModel(modelId: RunPodCompletionModelId): LanguageModelV2;\n\n  /**\nCreates an image model for image generation.\n*/\n  imageModel(modelId: RunpodImageModelId): ImageModelV2;\n}\n\n// Mapping of RunPod model IDs to their endpoint URLs\nconst MODEL_ID_TO_ENDPOINT_URL: Record<string, string> = {\n  'deep-cogito/deep-cogito-v2-llama-70b':\n    'https://api.runpod.ai/v2/deep-cogito-v2-llama-70b/openai/v1',\n  'qwen/qwen3-32b-awq': 'https://api.runpod.ai/v2/qwen3-32b-awq/openai/v1',\n};\n\n// Mapping of RunPod image model IDs to their endpoint URLs\nconst IMAGE_MODEL_ID_TO_ENDPOINT_URL: Record<string, string> = {\n  'qwen/qwen-image': 'https://api.runpod.ai/v2/qwen-image-t2i',\n  'bytedance/seedream-3.0': 'https://api.runpod.ai/v2/seedream-3-0-t2i',\n  'black-forest-labs/flux-1-kontext-dev':\n    'https://api.runpod.ai/v2/black-forest-labs-flux-1-kontext-dev',\n  'black-forest-labs/flux-1-schnell':\n    'https://api.runpod.ai/v2/black-forest-labs-flux-1-schnell',\n  'black-forest-labs/flux-1-dev':\n    'https://api.runpod.ai/v2/black-forest-labs-flux-1-dev',\n};\n\n// Mapping of RunPod model IDs to their OpenAI model names\nconst MODEL_ID_TO_OPENAI_NAME: Record<string, string> = {\n  'deep-cogito/deep-cogito-v2-llama-70b':\n    'deepcogito/cogito-v2-preview-llama-70B',\n  'qwen/qwen3-32b-awq': 'Qwen/Qwen3-32B-AWQ',\n};\n\nexport function createRunPod(\n  options: RunPodProviderSettings = {}\n): RunPodProvider {\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'RUNPOD_API_KEY',\n      description: 'RunPod',\n    })}`,\n    ...options.headers,\n  });\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const getModelConfig = (\n    modelId: string,\n    modelType: string\n  ): CommonModelConfig => {\n    const baseURL = MODEL_ID_TO_ENDPOINT_URL[modelId];\n    if (!baseURL) {\n      throw new Error(\n        `Unsupported RunPod model: ${modelId}. Supported models: ${Object.keys(\n          MODEL_ID_TO_ENDPOINT_URL\n        ).join(', ')}`\n      );\n    }\n\n    return {\n      provider: `runpod.${modelType}`,\n      url: ({ path }) => `${withoutTrailingSlash(baseURL)}${path}`,\n      headers: getHeaders,\n      fetch: options.fetch,\n    };\n  };\n\n  const createChatModel = (modelId: RunPodChatModelId) => {\n    const openaiModelName = MODEL_ID_TO_OPENAI_NAME[modelId] || modelId;\n    return new OpenAICompatibleChatLanguageModel(\n      openaiModelName,\n      getModelConfig(modelId, 'chat')\n    );\n  };\n\n  const createCompletionModel = (modelId: RunPodCompletionModelId) => {\n    const openaiModelName = MODEL_ID_TO_OPENAI_NAME[modelId] || modelId;\n    return new OpenAICompatibleCompletionLanguageModel(\n      openaiModelName,\n      getModelConfig(modelId, 'completion')\n    );\n  };\n\n  const createImageModel = (modelId: RunpodImageModelId) => {\n    const baseURL = IMAGE_MODEL_ID_TO_ENDPOINT_URL[modelId];\n    if (!baseURL) {\n      throw new Error(\n        `Unsupported Runpod image model: ${modelId}. Supported models: ${Object.keys(\n          IMAGE_MODEL_ID_TO_ENDPOINT_URL\n        ).join(', ')}`\n      );\n    }\n\n    return new RunpodImageModel(modelId, {\n      provider: 'runpod.image',\n      baseURL,\n      headers: getHeaders,\n      fetch: options.fetch,\n    });\n  };\n\n  const provider = (modelId: RunPodChatModelId) => createChatModel(modelId);\n\n  provider.completionModel = createCompletionModel;\n  provider.languageModel = createChatModel;\n  provider.chatModel = createChatModel;\n  provider.imageModel = createImageModel;\n\n  return provider;\n}\n\nexport const runpod = createRunPod();\n","import { ImageModelV2, ImageModelV2CallWarning } from '@ai-sdk/provider';\r\nimport {\r\n  combineHeaders,\r\n  createJsonResponseHandler,\r\n  createJsonErrorResponseHandler,\r\n  createBinaryResponseHandler,\r\n  FetchFunction,\r\n  postJsonToApi,\r\n  getFromApi,\r\n} from '@ai-sdk/provider-utils';\r\nimport { InvalidArgumentError } from '@ai-sdk/provider';\r\nimport { z } from 'zod';\r\nimport { RunpodImageModelId } from './runpod-image-options';\r\n\r\ninterface RunpodImageModelConfig {\r\n  provider: string;\r\n  baseURL: string;\r\n  headers: () => Record<string, string>;\r\n  fetch?: FetchFunction;\r\n  _internal?: {\r\n    currentDate?: () => Date;\r\n  };\r\n}\r\n\r\n// Runpod supported aspect ratios (only validated working sizes)\r\nconst SUPPORTED_ASPECT_RATIOS: Record<string, string> = {\r\n  '1:1': '1328*1328', // ✅ Native support\r\n  '4:3': '1472*1140', // ✅ Native support\r\n  '3:4': '1140*1472', // ✅ Native support\r\n};\r\n\r\n// Runpod supported sizes (validated working sizes)\r\nconst SUPPORTED_SIZES = new Set([\r\n  // Native aspect ratio sizes\r\n  '1328*1328', // 1:1\r\n  '1472*1140', // 4:3\r\n  '1140*1472', // 3:4\r\n  // Additional validated sizes\r\n  '512*512',\r\n  '768*768',\r\n  '1024*1024',\r\n  '512*768',\r\n  '768*512',\r\n  '1024*768',\r\n  '768*1024',\r\n]);\r\n\r\nexport class RunpodImageModel implements ImageModelV2 {\r\n  readonly specificationVersion = 'v2';\r\n  readonly maxImagesPerCall = 1;\r\n\r\n  get provider(): string {\r\n    return this.config.provider;\r\n  }\r\n\r\n  constructor(\r\n    readonly modelId: RunpodImageModelId,\r\n    private config: RunpodImageModelConfig\r\n  ) {}\r\n\r\n  async doGenerate({\r\n    prompt,\r\n    n = 1,\r\n    size,\r\n    aspectRatio,\r\n    seed,\r\n    providerOptions,\r\n    headers,\r\n    abortSignal,\r\n  }: Parameters<ImageModelV2['doGenerate']>[0]): Promise<\r\n    Awaited<ReturnType<ImageModelV2['doGenerate']>>\r\n  > {\r\n    const warnings: Array<ImageModelV2CallWarning> = [];\r\n\r\n    // Determine the size to use\r\n    let runpodSize: string;\r\n\r\n    if (size) {\r\n      // Convert AI SDK format \"1328x1328\" to Runpod format \"1328*1328\"\r\n      const runpodSizeCandidate = size.replace('x', '*');\r\n\r\n      // Validate size is supported\r\n      if (!SUPPORTED_SIZES.has(runpodSizeCandidate)) {\r\n        throw new InvalidArgumentError({\r\n          argument: 'size',\r\n          message: `Size ${size} is not supported by Runpod. Supported sizes: ${Array.from(\r\n            SUPPORTED_SIZES\r\n          )\r\n            .map((s) => s.replace('*', 'x'))\r\n            .join(', ')}`,\r\n        });\r\n      }\r\n\r\n      runpodSize = runpodSizeCandidate;\r\n    } else if (aspectRatio) {\r\n      // Validate aspect ratio is supported\r\n      if (!SUPPORTED_ASPECT_RATIOS[aspectRatio]) {\r\n        throw new InvalidArgumentError({\r\n          argument: 'aspectRatio',\r\n          message: `Aspect ratio ${aspectRatio} is not supported by Runpod. Supported aspect ratios: ${Object.keys(SUPPORTED_ASPECT_RATIOS).join(', ')}`,\r\n        });\r\n      }\r\n\r\n      // Use supported aspect ratio mapping\r\n      runpodSize = SUPPORTED_ASPECT_RATIOS[aspectRatio];\r\n    } else {\r\n      // Default to square format\r\n      runpodSize = '1328*1328';\r\n    }\r\n\r\n    // Handle multiple images warning\r\n    if (n > 1) {\r\n      warnings.push({\r\n        type: 'unsupported-setting',\r\n        setting: 'n',\r\n        details:\r\n          'Runpod image models only support generating 1 image at a time. Using n=1.',\r\n      });\r\n    }\r\n\r\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\r\n\r\n    // Runpod uses a different request format - /runsync endpoint with input wrapper\r\n    const inputPayload = this.buildInputPayload(\r\n      prompt,\r\n      runpodSize,\r\n      seed,\r\n      providerOptions.runpod\r\n    );\r\n\r\n    const { value: response, responseHeaders } = await postJsonToApi({\r\n      url: `${this.config.baseURL}/runsync`,\r\n      headers: combineHeaders(this.config.headers(), headers),\r\n      body: {\r\n        input: inputPayload,\r\n      },\r\n      failedResponseHandler: createJsonErrorResponseHandler({\r\n        errorSchema: runpodImageErrorSchema as any,\r\n        errorToMessage: (data: any) => data.error ?? 'Unknown error',\r\n      }),\r\n      successfulResponseHandler: createJsonResponseHandler(\r\n        runpodImageResponseSchema as any\r\n      ),\r\n      abortSignal,\r\n      fetch: this.config.fetch,\r\n    });\r\n\r\n    // Handle both sync and async responses from Runpod\r\n    const typedResponse = response as any;\r\n    if (\r\n      typedResponse.status === 'COMPLETED' &&\r\n      (typedResponse.output?.result || typedResponse.output?.image_url)\r\n    ) {\r\n      // Sync response - image is ready\r\n      // Different models use different response formats: result vs image_url\r\n      const imageUrl =\r\n        typedResponse.output.result || typedResponse.output.image_url;\r\n      const imageData = await this.downloadImage(imageUrl, abortSignal);\r\n\r\n      return {\r\n        images: [imageData],\r\n        warnings,\r\n        response: {\r\n          timestamp: currentDate,\r\n          modelId: this.modelId,\r\n          headers: responseHeaders,\r\n        },\r\n        providerMetadata: {\r\n          runpod: {\r\n            images: [\r\n              {\r\n                url: imageUrl,\r\n                cost: typedResponse.output?.cost,\r\n              },\r\n            ],\r\n          },\r\n        },\r\n      };\r\n    } else if (\r\n      typedResponse.status === 'IN_QUEUE' ||\r\n      typedResponse.status === 'IN_PROGRESS'\r\n    ) {\r\n      // Async response - need to poll for completion\r\n      const pollOptions = {\r\n        maxAttempts: providerOptions.runpod?.maxPollAttempts as number,\r\n        pollIntervalMillis: providerOptions.runpod\r\n          ?.pollIntervalMillis as number,\r\n      };\r\n      const imageUrl = await this.pollForCompletion(\r\n        typedResponse.id,\r\n        abortSignal,\r\n        pollOptions\r\n      );\r\n      const imageData = await this.downloadImage(imageUrl, abortSignal);\r\n\r\n      return {\r\n        images: [imageData],\r\n        warnings,\r\n        response: {\r\n          timestamp: currentDate,\r\n          modelId: this.modelId,\r\n          headers: responseHeaders,\r\n        },\r\n        providerMetadata: {\r\n          runpod: {\r\n            images: [\r\n              {\r\n                url: imageUrl,\r\n                jobId: typedResponse.id,\r\n              },\r\n            ],\r\n          },\r\n        },\r\n      };\r\n    } else {\r\n      throw new Error(`Unexpected response status: ${typedResponse.status}`);\r\n    }\r\n  }\r\n\r\n  private async downloadImage(\r\n    imageUrl: string,\r\n    abortSignal?: AbortSignal\r\n  ): Promise<Uint8Array> {\r\n    const { value: imageData } = await getFromApi({\r\n      url: imageUrl,\r\n      successfulResponseHandler: createBinaryResponseHandler(),\r\n      failedResponseHandler: createJsonErrorResponseHandler({\r\n        errorSchema: runpodImageErrorSchema as any,\r\n        errorToMessage: (data: any) => data.error ?? 'Failed to download image',\r\n      }),\r\n      abortSignal,\r\n      fetch: this.config.fetch,\r\n    });\r\n    return imageData;\r\n  }\r\n\r\n  private async pollForCompletion(\r\n    jobId: string,\r\n    abortSignal?: AbortSignal,\r\n    pollOptions?: { maxAttempts?: number; pollIntervalMillis?: number }\r\n  ): Promise<string> {\r\n    const maxAttempts = pollOptions?.maxAttempts ?? 60; // 5 minutes with 5-second intervals\r\n    const pollInterval = pollOptions?.pollIntervalMillis ?? 5000; // 5 seconds\r\n\r\n    for (let attempt = 0; attempt < maxAttempts; attempt++) {\r\n      if (abortSignal?.aborted) {\r\n        throw new Error('Image generation was aborted');\r\n      }\r\n\r\n      const { value: statusResponse } = await getFromApi({\r\n        url: `${this.config.baseURL}/status/${jobId}`,\r\n        headers: this.config.headers(),\r\n        successfulResponseHandler: createJsonResponseHandler(\r\n          runpodImageStatusSchema as any\r\n        ),\r\n        failedResponseHandler: createJsonErrorResponseHandler({\r\n          errorSchema: runpodImageErrorSchema as any,\r\n          errorToMessage: (data: any) =>\r\n            data.error ?? 'Failed to check job status',\r\n        }),\r\n        abortSignal,\r\n        fetch: this.config.fetch,\r\n      });\r\n\r\n      const typedStatusResponse = statusResponse as any;\r\n      if (\r\n        typedStatusResponse.status === 'COMPLETED' &&\r\n        (typedStatusResponse.output?.result ||\r\n          typedStatusResponse.output?.image_url)\r\n      ) {\r\n        return (\r\n          typedStatusResponse.output.result ||\r\n          typedStatusResponse.output.image_url\r\n        );\r\n      }\r\n\r\n      if (typedStatusResponse.status === 'FAILED') {\r\n        throw new Error(\r\n          `Image generation failed: ${typedStatusResponse.error || 'Unknown error'}`\r\n        );\r\n      }\r\n\r\n      // Wait before next poll\r\n      await new Promise((resolve) => setTimeout(resolve, pollInterval));\r\n    }\r\n\r\n    throw new Error(\r\n      `Image generation timed out after ${maxAttempts} attempts (${(maxAttempts * pollInterval) / 1000}s)`\r\n    );\r\n  }\r\n\r\n  private buildInputPayload(\r\n    prompt: string,\r\n    runpodSize: string,\r\n    seed?: number,\r\n    runpodOptions?: Record<string, unknown>\r\n  ): Record<string, unknown> {\r\n    // Check if this is a Flux model that uses different parameters\r\n    const isFluxModel =\r\n      this.modelId.includes('flux') ||\r\n      this.modelId.includes('black-forest-labs');\r\n\r\n    if (isFluxModel) {\r\n      // Check if this is Flux Kontext (uses different parameters)\r\n      const isKontext = this.modelId.includes('kontext');\r\n\r\n      if (isKontext) {\r\n        // Flux Kontext uses size format and has image input\r\n        return {\r\n          prompt,\r\n          negative_prompt: runpodOptions?.negative_prompt ?? '',\r\n          seed: seed ?? -1,\r\n          num_inference_steps: 28,\r\n          guidance: 2,\r\n          size: runpodSize,\r\n          output_format: 'png',\r\n          enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,\r\n          ...runpodOptions, // This will include the 'image' parameter if provided\r\n        };\r\n      } else {\r\n        // Regular Flux models use width/height\r\n        const [width, height] = runpodSize.split('*').map(Number);\r\n\r\n        return {\r\n          prompt,\r\n          negative_prompt: runpodOptions?.negative_prompt ?? '',\r\n          seed: seed ?? -1,\r\n          num_inference_steps: this.modelId.includes('schnell') ? 4 : 28,\r\n          guidance: this.modelId.includes('schnell') ? 7 : 2,\r\n          width,\r\n          height,\r\n          image_format: 'png',\r\n          ...runpodOptions,\r\n        };\r\n      }\r\n    }\r\n\r\n    // Default format for Qwen and other models\r\n    return {\r\n      prompt,\r\n      negative_prompt: runpodOptions?.negative_prompt ?? '',\r\n      size: runpodSize,\r\n      seed: seed ?? -1,\r\n      enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,\r\n      ...runpodOptions,\r\n    };\r\n  }\r\n}\r\n\r\n// Runpod image API response schema (handles both sync and async responses)\r\nconst runpodImageResponseSchema = z.object({\r\n  id: z.string(),\r\n  status: z.enum(['COMPLETED', 'IN_QUEUE', 'IN_PROGRESS', 'FAILED']),\r\n  delayTime: z.number().optional(),\r\n  executionTime: z.number().optional(),\r\n  output: z\r\n    .object({\r\n      cost: z.number().optional(),\r\n      result: z.string().optional(), // URL to the generated image (Qwen format)\r\n      image_url: z.string().optional(), // URL to the generated image (Flux format)\r\n    })\r\n    .optional(), // Optional for IN_QUEUE/IN_PROGRESS responses\r\n});\r\n\r\n// Schema for polling status endpoint\r\nconst runpodImageStatusSchema = z.object({\r\n  id: z.string(),\r\n  status: z.enum(['COMPLETED', 'IN_QUEUE', 'IN_PROGRESS', 'FAILED']),\r\n  output: z\r\n    .object({\r\n      cost: z.number().optional(),\r\n      result: z.string().optional(),\r\n      image_url: z.string().optional(),\r\n    })\r\n    .optional(),\r\n  error: z.string().optional(), // Error message if FAILED\r\n});\r\n\r\n// Runpod image API error schema\r\nconst runpodImageErrorSchema = z.object({\r\n  error: z.string(),\r\n  message: z.string().optional(),\r\n});\r\n"],"mappings":";AACA;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,OACK;;;ACRP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAEA;AAAA,EACA;AAAA,OACK;AACP,SAAS,4BAA4B;AACrC,SAAS,SAAS;AAclB,IAAM,0BAAkD;AAAA,EACtD,OAAO;AAAA;AAAA,EACP,OAAO;AAAA;AAAA,EACP,OAAO;AAAA;AACT;AAGA,IAAM,kBAAkB,oBAAI,IAAI;AAAA;AAAA,EAE9B;AAAA;AAAA,EACA;AAAA;AAAA,EACA;AAAA;AAAA;AAAA,EAEA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAEM,IAAM,mBAAN,MAA+C;AAAA,EAQpD,YACW,SACD,QACR;AAFS;AACD;AATV,SAAS,uBAAuB;AAChC,SAAS,mBAAmB;AAAA,EASzB;AAAA,EAPH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAOA,MAAM,WAAW;AAAA,IACf;AAAA,IACA,IAAI;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AACA,UAAM,WAA2C,CAAC;AAGlD,QAAI;AAEJ,QAAI,MAAM;AAER,YAAM,sBAAsB,KAAK,QAAQ,KAAK,GAAG;AAGjD,UAAI,CAAC,gBAAgB,IAAI,mBAAmB,GAAG;AAC7C,cAAM,IAAI,qBAAqB;AAAA,UAC7B,UAAU;AAAA,UACV,SAAS,QAAQ,IAAI,iDAAiD,MAAM;AAAA,YAC1E;AAAA,UACF,EACG,IAAI,CAAC,MAAM,EAAE,QAAQ,KAAK,GAAG,CAAC,EAC9B,KAAK,IAAI,CAAC;AAAA,QACf,CAAC;AAAA,MACH;AAEA,mBAAa;AAAA,IACf,WAAW,aAAa;AAEtB,UAAI,CAAC,wBAAwB,WAAW,GAAG;AACzC,cAAM,IAAI,qBAAqB;AAAA,UAC7B,UAAU;AAAA,UACV,SAAS,gBAAgB,WAAW,yDAAyD,OAAO,KAAK,uBAAuB,EAAE,KAAK,IAAI,CAAC;AAAA,QAC9I,CAAC;AAAA,MACH;AAGA,mBAAa,wBAAwB,WAAW;AAAA,IAClD,OAAO;AAEL,mBAAa;AAAA,IACf;AAGA,QAAI,IAAI,GAAG;AACT,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SACE;AAAA,MACJ,CAAC;AAAA,IACH;AAEA,UAAM,cAAc,KAAK,OAAO,WAAW,cAAc,KAAK,oBAAI,KAAK;AAGvE,UAAM,eAAe,KAAK;AAAA,MACxB;AAAA,MACA;AAAA,MACA;AAAA,MACA,gBAAgB;AAAA,IAClB;AAEA,UAAM,EAAE,OAAO,UAAU,gBAAgB,IAAI,MAAM,cAAc;AAAA,MAC/D,KAAK,GAAG,KAAK,OAAO,OAAO;AAAA,MAC3B,SAAS,eAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;AAAA,MACtD,MAAM;AAAA,QACJ,OAAO;AAAA,MACT;AAAA,MACA,uBAAuB,+BAA+B;AAAA,QACpD,aAAa;AAAA,QACb,gBAAgB,CAAC,SAAc,KAAK,SAAS;AAAA,MAC/C,CAAC;AAAA,MACD,2BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,MACA;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAGD,UAAM,gBAAgB;AACtB,QACE,cAAc,WAAW,gBACxB,cAAc,QAAQ,UAAU,cAAc,QAAQ,YACvD;AAGA,YAAM,WACJ,cAAc,OAAO,UAAU,cAAc,OAAO;AACtD,YAAM,YAAY,MAAM,KAAK,cAAc,UAAU,WAAW;AAEhE,aAAO;AAAA,QACL,QAAQ,CAAC,SAAS;AAAA,QAClB;AAAA,QACA,UAAU;AAAA,UACR,WAAW;AAAA,UACX,SAAS,KAAK;AAAA,UACd,SAAS;AAAA,QACX;AAAA,QACA,kBAAkB;AAAA,UAChB,QAAQ;AAAA,YACN,QAAQ;AAAA,cACN;AAAA,gBACE,KAAK;AAAA,gBACL,MAAM,cAAc,QAAQ;AAAA,cAC9B;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,WACE,cAAc,WAAW,cACzB,cAAc,WAAW,eACzB;AAEA,YAAM,cAAc;AAAA,QAClB,aAAa,gBAAgB,QAAQ;AAAA,QACrC,oBAAoB,gBAAgB,QAChC;AAAA,MACN;AACA,YAAM,WAAW,MAAM,KAAK;AAAA,QAC1B,cAAc;AAAA,QACd;AAAA,QACA;AAAA,MACF;AACA,YAAM,YAAY,MAAM,KAAK,cAAc,UAAU,WAAW;AAEhE,aAAO;AAAA,QACL,QAAQ,CAAC,SAAS;AAAA,QAClB;AAAA,QACA,UAAU;AAAA,UACR,WAAW;AAAA,UACX,SAAS,KAAK;AAAA,UACd,SAAS;AAAA,QACX;AAAA,QACA,kBAAkB;AAAA,UAChB,QAAQ;AAAA,YACN,QAAQ;AAAA,cACN;AAAA,gBACE,KAAK;AAAA,gBACL,OAAO,cAAc;AAAA,cACvB;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,OAAO;AACL,YAAM,IAAI,MAAM,+BAA+B,cAAc,MAAM,EAAE;AAAA,IACvE;AAAA,EACF;AAAA,EAEA,MAAc,cACZ,UACA,aACqB;AACrB,UAAM,EAAE,OAAO,UAAU,IAAI,MAAM,WAAW;AAAA,MAC5C,KAAK;AAAA,MACL,2BAA2B,4BAA4B;AAAA,MACvD,uBAAuB,+BAA+B;AAAA,QACpD,aAAa;AAAA,QACb,gBAAgB,CAAC,SAAc,KAAK,SAAS;AAAA,MAC/C,CAAC;AAAA,MACD;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,MAAc,kBACZ,OACA,aACA,aACiB;AACjB,UAAM,cAAc,aAAa,eAAe;AAChD,UAAM,eAAe,aAAa,sBAAsB;AAExD,aAAS,UAAU,GAAG,UAAU,aAAa,WAAW;AACtD,UAAI,aAAa,SAAS;AACxB,cAAM,IAAI,MAAM,8BAA8B;AAAA,MAChD;AAEA,YAAM,EAAE,OAAO,eAAe,IAAI,MAAM,WAAW;AAAA,QACjD,KAAK,GAAG,KAAK,OAAO,OAAO,WAAW,KAAK;AAAA,QAC3C,SAAS,KAAK,OAAO,QAAQ;AAAA,QAC7B,2BAA2B;AAAA,UACzB;AAAA,QACF;AAAA,QACA,uBAAuB,+BAA+B;AAAA,UACpD,aAAa;AAAA,UACb,gBAAgB,CAAC,SACf,KAAK,SAAS;AAAA,QAClB,CAAC;AAAA,QACD;AAAA,QACA,OAAO,KAAK,OAAO;AAAA,MACrB,CAAC;AAED,YAAM,sBAAsB;AAC5B,UACE,oBAAoB,WAAW,gBAC9B,oBAAoB,QAAQ,UAC3B,oBAAoB,QAAQ,YAC9B;AACA,eACE,oBAAoB,OAAO,UAC3B,oBAAoB,OAAO;AAAA,MAE/B;AAEA,UAAI,oBAAoB,WAAW,UAAU;AAC3C,cAAM,IAAI;AAAA,UACR,4BAA4B,oBAAoB,SAAS,eAAe;AAAA,QAC1E;AAAA,MACF;AAGA,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,YAAY,CAAC;AAAA,IAClE;AAEA,UAAM,IAAI;AAAA,MACR,oCAAoC,WAAW,cAAe,cAAc,eAAgB,GAAI;AAAA,IAClG;AAAA,EACF;AAAA,EAEQ,kBACN,QACA,YACA,MACA,eACyB;AAEzB,UAAM,cACJ,KAAK,QAAQ,SAAS,MAAM,KAC5B,KAAK,QAAQ,SAAS,mBAAmB;AAE3C,QAAI,aAAa;AAEf,YAAM,YAAY,KAAK,QAAQ,SAAS,SAAS;AAEjD,UAAI,WAAW;AAEb,eAAO;AAAA,UACL;AAAA,UACA,iBAAiB,eAAe,mBAAmB;AAAA,UACnD,MAAM,QAAQ;AAAA,UACd,qBAAqB;AAAA,UACrB,UAAU;AAAA,UACV,MAAM;AAAA,UACN,eAAe;AAAA,UACf,uBAAuB,eAAe,yBAAyB;AAAA,UAC/D,GAAG;AAAA;AAAA,QACL;AAAA,MACF,OAAO;AAEL,cAAM,CAAC,OAAO,MAAM,IAAI,WAAW,MAAM,GAAG,EAAE,IAAI,MAAM;AAExD,eAAO;AAAA,UACL;AAAA,UACA,iBAAiB,eAAe,mBAAmB;AAAA,UACnD,MAAM,QAAQ;AAAA,UACd,qBAAqB,KAAK,QAAQ,SAAS,SAAS,IAAI,IAAI;AAAA,UAC5D,UAAU,KAAK,QAAQ,SAAS,SAAS,IAAI,IAAI;AAAA,UACjD;AAAA,UACA;AAAA,UACA,cAAc;AAAA,UACd,GAAG;AAAA,QACL;AAAA,MACF;AAAA,IACF;AAGA,WAAO;AAAA,MACL;AAAA,MACA,iBAAiB,eAAe,mBAAmB;AAAA,MACnD,MAAM;AAAA,MACN,MAAM,QAAQ;AAAA,MACd,uBAAuB,eAAe,yBAAyB;AAAA,MAC/D,GAAG;AAAA,IACL;AAAA,EACF;AACF;AAGA,IAAM,4BAA4B,EAAE,OAAO;AAAA,EACzC,IAAI,EAAE,OAAO;AAAA,EACb,QAAQ,EAAE,KAAK,CAAC,aAAa,YAAY,eAAe,QAAQ,CAAC;AAAA,EACjE,WAAW,EAAE,OAAO,EAAE,SAAS;AAAA,EAC/B,eAAe,EAAE,OAAO,EAAE,SAAS;AAAA,EACnC,QAAQ,EACL,OAAO;AAAA,IACN,MAAM,EAAE,OAAO,EAAE,SAAS;AAAA,IAC1B,QAAQ,EAAE,OAAO,EAAE,SAAS;AAAA;AAAA,IAC5B,WAAW,EAAE,OAAO,EAAE,SAAS;AAAA;AAAA,EACjC,CAAC,EACA,SAAS;AAAA;AACd,CAAC;AAGD,IAAM,0BAA0B,EAAE,OAAO;AAAA,EACvC,IAAI,EAAE,OAAO;AAAA,EACb,QAAQ,EAAE,KAAK,CAAC,aAAa,YAAY,eAAe,QAAQ,CAAC;AAAA,EACjE,QAAQ,EACL,OAAO;AAAA,IACN,MAAM,EAAE,OAAO,EAAE,SAAS;AAAA,IAC1B,QAAQ,EAAE,OAAO,EAAE,SAAS;AAAA,IAC5B,WAAW,EAAE,OAAO,EAAE,SAAS;AAAA,EACjC,CAAC,EACA,SAAS;AAAA,EACZ,OAAO,EAAE,OAAO,EAAE,SAAS;AAAA;AAC7B,CAAC;AAGD,IAAM,yBAAyB,EAAE,OAAO;AAAA,EACtC,OAAO,EAAE,OAAO;AAAA,EAChB,SAAS,EAAE,OAAO,EAAE,SAAS;AAC/B,CAAC;;;ADnUD,IAAM,2BAAmD;AAAA,EACvD,wCACE;AAAA,EACF,sBAAsB;AACxB;AAGA,IAAM,iCAAyD;AAAA,EAC7D,mBAAmB;AAAA,EACnB,0BAA0B;AAAA,EAC1B,wCACE;AAAA,EACF,oCACE;AAAA,EACF,gCACE;AACJ;AAGA,IAAM,0BAAkD;AAAA,EACtD,wCACE;AAAA,EACF,sBAAsB;AACxB;AAEO,SAAS,aACd,UAAkC,CAAC,GACnB;AAChB,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AASA,QAAM,iBAAiB,CACrB,SACA,cACsB;AACtB,UAAM,UAAU,yBAAyB,OAAO;AAChD,QAAI,CAAC,SAAS;AACZ,YAAM,IAAI;AAAA,QACR,6BAA6B,OAAO,uBAAuB,OAAO;AAAA,UAChE;AAAA,QACF,EAAE,KAAK,IAAI,CAAC;AAAA,MACd;AAAA,IACF;AAEA,WAAO;AAAA,MACL,UAAU,UAAU,SAAS;AAAA,MAC7B,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,qBAAqB,OAAO,CAAC,GAAG,IAAI;AAAA,MAC1D,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,IACjB;AAAA,EACF;AAEA,QAAM,kBAAkB,CAAC,YAA+B;AACtD,UAAM,kBAAkB,wBAAwB,OAAO,KAAK;AAC5D,WAAO,IAAI;AAAA,MACT;AAAA,MACA,eAAe,SAAS,MAAM;AAAA,IAChC;AAAA,EACF;AAEA,QAAM,wBAAwB,CAAC,YAAqC;AAClE,UAAM,kBAAkB,wBAAwB,OAAO,KAAK;AAC5D,WAAO,IAAI;AAAA,MACT;AAAA,MACA,eAAe,SAAS,YAAY;AAAA,IACtC;AAAA,EACF;AAEA,QAAM,mBAAmB,CAAC,YAAgC;AACxD,UAAM,UAAU,+BAA+B,OAAO;AACtD,QAAI,CAAC,SAAS;AACZ,YAAM,IAAI;AAAA,QACR,mCAAmC,OAAO,uBAAuB,OAAO;AAAA,UACtE;AAAA,QACF,EAAE,KAAK,IAAI,CAAC;AAAA,MACd;AAAA,IACF;AAEA,WAAO,IAAI,iBAAiB,SAAS;AAAA,MACnC,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,IACjB,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CAAC,YAA+B,gBAAgB,OAAO;AAExE,WAAS,kBAAkB;AAC3B,WAAS,gBAAgB;AACzB,WAAS,YAAY;AACrB,WAAS,aAAa;AAEtB,SAAO;AACT;AAEO,IAAM,SAAS,aAAa;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@runpod/ai-sdk-provider",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",