@runpod/ai-sdk-provider 1.2.0 → 1.4.0

package/dist/index.js

@@ -27,7 +27,7 @@ module.exports = __toCommonJS(index_exports);
 
 // src/runpod-provider.ts
 var import_openai_compatible = require("@ai-sdk/openai-compatible");
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
+var import_provider_utils6 = require("@ai-sdk/provider-utils");
 
 // src/runpod-image-model.ts
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
@@ -69,6 +69,10 @@ var runpodTranscriptionFailedResponseHandler = (0, import_provider_utils.createJ
   errorSchema: runpodImageErrorSchema,
   errorToMessage: extractErrorMessage
 });
+var runpodVideoFailedResponseHandler = (0, import_provider_utils.createJsonErrorResponseHandler)({
+  errorSchema: runpodImageErrorSchema,
+  errorToMessage: extractErrorMessage
+});
 
 // src/runpod-image-model.ts
 var SUPPORTED_ASPECT_RATIOS = {
@@ -101,6 +105,50 @@ var SUPPORTED_SIZES = /* @__PURE__ */ new Set([
   "1024*768",
   "768*1024"
 ]);
+var Z_IMAGE_TURBO_SUPPORTED_SIZES = /* @__PURE__ */ new Set([
+  "1328*1328",
+  // 1:1
+  "1472*1140",
+  // 4:3
+  "1140*1472",
+  // 3:4
+  "512*512",
+  "768*768",
+  "1024*1024",
+  "1280*1280",
+  "1536*1536",
+  "512*768",
+  "768*512",
+  "1024*768",
+  "768*1024",
+  "768*432",
+  "1024*576",
+  "1280*720",
+  "1536*864",
+  "432*768",
+  "576*1024",
+  "720*1280",
+  "864*1536"
+]);
+var Z_IMAGE_TURBO_ASPECT_RATIOS = {
+  "1:1": "1328*1328",
+  "4:3": "1472*1140",
+  "3:4": "1140*1472",
+  "3:2": "768*512",
+  "2:3": "512*768",
+  "16:9": "1280*720",
+  "9:16": "720*1280"
+};
+var MODEL_SUPPORTED_SIZES = {
+  "tongyi-mai/z-image-turbo": Z_IMAGE_TURBO_SUPPORTED_SIZES,
+  "z-image-turbo": Z_IMAGE_TURBO_SUPPORTED_SIZES
+  // alias, not advertised
+};
+var MODEL_SUPPORTED_ASPECT_RATIOS = {
+  "tongyi-mai/z-image-turbo": Z_IMAGE_TURBO_ASPECT_RATIOS,
+  "z-image-turbo": Z_IMAGE_TURBO_ASPECT_RATIOS
+  // alias, not advertised
+};
 var WAN_ASPECT_RATIOS = {
   "1:1": "1280*1280",
   // 1,638,400 pixels
@@ -184,9 +232,10 @@ var RunpodImageModel = class {
     }
     const isPrunaModel = this.modelId.includes("pruna") || this.modelId.includes("p-image");
     const isNanoBananaProModel = this.modelId.includes("nano-banana-pro");
+    const isNanoBanana2Model = this.modelId.includes("nano-banana-2");
     const isWanModel = this.modelId.includes("wan-2");
     let runpodSize;
-    if (isPrunaModel || isNanoBananaProModel) {
+    if (isPrunaModel || isNanoBananaProModel || isNanoBanana2Model) {
       runpodSize = aspectRatio || "1:1";
     } else if (isWanModel) {
       if (size) {
@@ -206,23 +255,27 @@ var RunpodImageModel = class {
       }
     } else if (size) {
       const runpodSizeCandidate = size.replace("x", "*");
-      if (!SUPPORTED_SIZES.has(runpodSizeCandidate)) {
+      const supportedSizes = MODEL_SUPPORTED_SIZES[this.modelId] ?? SUPPORTED_SIZES;
+      if (!supportedSizes.has(runpodSizeCandidate)) {
         throw new import_provider.InvalidArgumentError({
           argument: "size",
           message: `Size ${size} is not supported by Runpod. Supported sizes: ${Array.from(
-            SUPPORTED_SIZES
+            supportedSizes
           ).map((s) => s.replace("*", "x")).join(", ")}`
         });
       }
       runpodSize = runpodSizeCandidate;
     } else if (aspectRatio) {
-      if (!SUPPORTED_ASPECT_RATIOS[aspectRatio]) {
+      const supportedAspectRatios = MODEL_SUPPORTED_ASPECT_RATIOS[this.modelId] ?? SUPPORTED_ASPECT_RATIOS;
+      if (!supportedAspectRatios[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(", ")}`
+          message: `Aspect ratio ${aspectRatio} is not supported by Runpod. Supported aspect ratios: ${Object.keys(
+            supportedAspectRatios
+          ).join(", ")}`
         });
       }
-      runpodSize = SUPPORTED_ASPECT_RATIOS[aspectRatio];
+      runpodSize = supportedAspectRatios[aspectRatio];
     } else {
       runpodSize = "1328*1328";
     }
@@ -477,6 +530,22 @@ var RunpodImageModel = class {
         return t2iPayload;
       }
     }
+    const isNanoBanana2Model = this.modelId.includes("nano-banana-2");
+    if (isNanoBanana2Model) {
+      const nanoBanana2Payload = {
+        prompt,
+        aspect_ratio: runpodOptions?.aspect_ratio ?? aspectRatio ?? "1:1",
+        resolution: runpodOptions?.resolution ?? "1k",
+        output_format: runpodOptions?.output_format ?? "jpeg",
+        enable_safety_checker: runpodOptions?.enable_safety_checker ?? true
+      };
+      if (standardizedImages && standardizedImages.length > 0) {
+        nanoBanana2Payload.images = standardizedImages;
+      } else if (runpodOptions?.images) {
+        nanoBanana2Payload.images = runpodOptions.images;
+      }
+      return nanoBanana2Payload;
+    }
     const isNanaBananaProModel = this.modelId.includes("nano-banana-pro");
     if (isNanaBananaProModel) {
       const nanoBananaPayload = {
@@ -525,6 +594,25 @@ var RunpodImageModel = class {
       }
       return qwenEdit2511Payload;
     }
+    const isZImageTurbo = this.modelId === "tongyi-mai/z-image-turbo" || this.modelId === "z-image-turbo";
+    if (isZImageTurbo) {
+      const zImageTurboPayload = {
+        prompt,
+        size: runpodSize,
+        seed: seed ?? -1,
+        output_format: runpodOptions?.output_format ?? "png",
+        enable_safety_checker: runpodOptions?.enable_safety_checker ?? true,
+        ...runpodOptions
+      };
+      if (standardizedImages && standardizedImages.length > 0) {
+        if (standardizedImages.length === 1) {
+          zImageTurboPayload.image = standardizedImages[0];
+        } else {
+          zImageTurboPayload.images = standardizedImages;
+        }
+      }
+      return zImageTurboPayload;
+    }
     const isWanModel = this.modelId.includes("wan-2");
     if (isWanModel) {
       return {
@@ -938,6 +1026,210 @@ var RunpodTranscriptionModel = class {
   }
 };
 
+// src/runpod-video-model.ts
+var import_provider_utils5 = require("@ai-sdk/provider-utils");
+var import_zod4 = require("zod");
+function isRecord3(value) {
+  return typeof value === "object" && value !== null;
+}
+var runpodVideoJobResponseSchema = import_zod4.z.object({
+  id: import_zod4.z.string(),
+  status: import_zod4.z.string().optional()
+});
+var runpodVideoStatusSchema = import_zod4.z.object({
+  id: import_zod4.z.string().optional(),
+  status: import_zod4.z.string(),
+  output: import_zod4.z.unknown().optional(),
+  error: import_zod4.z.string().optional()
+});
+var RunpodVideoModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v3";
+    this.maxVideosPerCall = 1;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  getRunpodRunUrl() {
+    const baseURL = (0, import_provider_utils5.withoutTrailingSlash)(this.config.baseURL) ?? this.config.baseURL;
+    if (baseURL.endsWith("/run") || baseURL.endsWith("/runsync")) {
+      return baseURL;
+    }
+    return `${baseURL}/run`;
+  }
+  async doGenerate(options) {
+    const currentDate = this.config._internal?.currentDate?.() ?? /* @__PURE__ */ new Date();
+    const warnings = [];
+    if (options.n > 1) {
+      warnings.push({
+        type: "unsupported",
+        feature: "n > 1",
+        details: "Runpod video models only support generating 1 video per call. Only 1 video will be generated."
+      });
+    }
+    const { providerOptions, abortSignal, headers } = options;
+    const runpodOptions = this.extractRunpodOptions(providerOptions);
+    const input = this.buildInputPayload(options, runpodOptions);
+    const requestBody = { input };
+    const url = this.getRunpodRunUrl();
+    const effectiveBaseURL = (0, import_provider_utils5.withoutTrailingSlash)(this.config.baseURL) ?? this.config.baseURL;
+    const { value: response, responseHeaders } = await (0, import_provider_utils5.postJsonToApi)({
+      url,
+      headers: {
+        ...this.config.headers(),
+        ...headers
+      },
+      body: requestBody,
+      failedResponseHandler: runpodVideoFailedResponseHandler,
+      successfulResponseHandler: (0, import_provider_utils5.createJsonResponseHandler)(
+        runpodVideoJobResponseSchema
+      ),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    const typedResponse = response;
+    const jobId = typedResponse.id;
+    if (!jobId) {
+      throw new Error("Runpod video response did not include a job id.");
+    }
+    const pollOptions = {
+      maxAttempts: runpodOptions.maxPollAttempts ?? 120,
+      pollIntervalMillis: runpodOptions.pollIntervalMillis ?? 5e3
+    };
+    const result = await this.pollForCompletion(
+      jobId,
+      abortSignal,
+      pollOptions,
+      effectiveBaseURL
+    );
+    const videoUrl = this.extractVideoUrl(result.output);
+    const providerMetadata = {
+      runpod: {
+        jobId
+      }
+    };
+    return {
+      videos: [{ type: "url", url: videoUrl, mediaType: "video/mp4" }],
+      warnings,
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders
+      },
+      providerMetadata
+    };
+  }
+  buildInputPayload(options, runpodOptions) {
+    const apiOptions = Object.fromEntries(
+      Object.entries(runpodOptions).filter(
+        ([key]) => key !== "maxPollAttempts" && key !== "pollIntervalMillis"
+      )
+    );
+    const input = {
+      ...apiOptions
+    };
+    if (options.prompt) {
+      input.prompt = options.prompt;
+    }
+    if (options.duration !== void 0) {
+      input.duration = options.duration;
+    }
+    if (options.fps !== void 0) {
+      input.fps = options.fps;
+    }
+    if (options.seed !== void 0) {
+      input.seed = options.seed;
+    }
+    if (options.resolution) {
+      input.size = options.resolution.replace("x", "*");
+    } else if (options.aspectRatio) {
+      input.aspect_ratio = options.aspectRatio;
+    }
+    if (options.image) {
+      input.image = this.convertFileToRunpodFormat(options.image);
+    }
+    return input;
+  }
+  convertFileToRunpodFormat(file) {
+    if (file.type === "url") {
+      return file.url;
+    }
+    const mediaType = file.mediaType;
+    const data = file.data;
+    if (typeof data === "string") {
+      return `data:${mediaType};base64,${data}`;
+    }
+    const base64 = this.uint8ArrayToBase64(data);
+    return `data:${mediaType};base64,${base64}`;
+  }
+  uint8ArrayToBase64(data) {
+    if (typeof Buffer !== "undefined") {
+      return Buffer.from(data).toString("base64");
+    }
+    let binary = "";
+    for (let i = 0; i < data.length; i++) {
+      binary += String.fromCharCode(data[i]);
+    }
+    return btoa(binary);
+  }
+  extractRunpodOptions(providerOptions) {
+    if (!providerOptions) return {};
+    const runpod2 = providerOptions.runpod;
+    if (isRecord3(runpod2)) {
+      return runpod2;
+    }
+    return {};
+  }
+  extractVideoUrl(output) {
+    if (isRecord3(output)) {
+      if (typeof output.video_url === "string") return output.video_url;
+      if (typeof output.result === "string") return output.result;
+      if (typeof output.url === "string") return output.url;
+    }
+    if (typeof output === "string") {
+      return output;
+    }
+    throw new Error(
+      `Runpod video generation completed but no video URL was found in the output: ${JSON.stringify(output)}`
+    );
+  }
+  async pollForCompletion(jobId, abortSignal, pollOptions, effectiveBaseURL) {
+    const maxAttempts = pollOptions?.maxAttempts ?? 120;
+    const pollInterval = pollOptions?.pollIntervalMillis ?? 5e3;
+    const baseURL = effectiveBaseURL ?? this.config.baseURL;
+    for (let attempt = 0; attempt < maxAttempts; attempt++) {
+      if (abortSignal?.aborted) {
+        throw new Error("Video generation was aborted");
+      }
+      const { value: statusResponse } = await (0, import_provider_utils5.getFromApi)({
+        url: `${baseURL}/status/${jobId}`,
+        headers: this.config.headers(),
+        successfulResponseHandler: (0, import_provider_utils5.createJsonResponseHandler)(
+          runpodVideoStatusSchema
+        ),
+        failedResponseHandler: runpodVideoFailedResponseHandler,
+        abortSignal,
+        fetch: this.config.fetch
+      });
+      const typedStatusResponse = statusResponse;
+      if (typedStatusResponse.status === "COMPLETED") {
+        return typedStatusResponse;
+      }
+      if (typedStatusResponse.status === "FAILED") {
+        throw new Error(
+          `Video generation failed: ${typedStatusResponse.error || "Unknown error"}`
+        );
+      }
+      await new Promise((resolve) => setTimeout(resolve, pollInterval));
+    }
+    throw new Error(
+      `Video generation timed out after ${maxAttempts} attempts (${maxAttempts * pollInterval / 1e3}s)`
+    );
+  }
+};
+
 // src/runpod-provider.ts
 var MODEL_ID_TO_ENDPOINT_URL = {
   "qwen/qwen3-32b-awq": "https://api.runpod.ai/v2/qwen3-32b-awq/openai/v1",
@@ -961,10 +1253,16 @@ var IMAGE_MODEL_ID_TO_ENDPOINT_URL = {
   "black-forest-labs/flux-1-dev": "https://api.runpod.ai/v2/black-forest-labs-flux-1-dev",
   // Alibaba Wan 2.6 (t2i)
   "alibaba/wan-2.6": "https://api.runpod.ai/v2/wan-2-6-t2i",
+  // Tongyi Z-Image Turbo (t2i)
+  "tongyi-mai/z-image-turbo": "https://api.runpod.ai/v2/z-image-turbo",
+  "z-image-turbo": "https://api.runpod.ai/v2/z-image-turbo",
+  // alias, not advertised
   // Nano Banana (edit only)
   "google/nano-banana-edit": "https://api.runpod.ai/v2/nano-banana-edit",
   "nano-banana-edit": "https://api.runpod.ai/v2/nano-banana-edit",
   // backwards compatibility
+  // Nano Banana 2 (edit only)
+  "google/nano-banana-2-edit": "https://api.runpod.ai/v2/google-nano-banana-2-edit",
   // Nano Banana Pro (edit only)
   "google/nano-banana-pro-edit": "https://api.runpod.ai/v2/nano-banana-pro-edit",
   // Pruna (t2i and edit)
@@ -977,6 +1275,23 @@ var SPEECH_MODEL_ID_TO_ENDPOINT_URL = {
 var TRANSCRIPTION_MODEL_ID_TO_ENDPOINT_URL = {
   "pruna/whisper-v3-large": "https://api.runpod.ai/v2/whisper-v3-large"
 };
+var VIDEO_MODEL_ID_TO_ENDPOINT_URL = {
+  "pruna/p-video": "https://api.runpod.ai/v2/p-video",
+  "vidu/q3-t2v": "https://api.runpod.ai/v2/vidu-q3-t2v",
+  "vidu/q3-i2v": "https://api.runpod.ai/v2/vidu-q3-i2v",
+  "kwaivgi/kling-v2.6-std-motion-control": "https://api.runpod.ai/v2/kling-v2-6-std-motion-control",
+  "kwaivgi/kling-video-o1-r2v": "https://api.runpod.ai/v2/kling-video-o1-r2v",
+  "kwaivgi/kling-v2.1-i2v-pro": "https://api.runpod.ai/v2/kling-v2-1-i2v-pro",
+  "alibaba/wan-2.6-t2v": "https://api.runpod.ai/v2/wan-2-6-t2v",
+  "alibaba/wan-2.6-i2v": "https://api.runpod.ai/v2/wan-2-6-i2v",
+  "alibaba/wan-2.5": "https://api.runpod.ai/v2/wan-2-5",
+  "alibaba/wan-2.2-t2v-720-lora": "https://api.runpod.ai/v2/wan-2-2-t2v-720-lora",
+  "alibaba/wan-2.2-i2v-720": "https://api.runpod.ai/v2/wan-2-2-i2v-720",
+  "alibaba/wan-2.1-i2v-720": "https://api.runpod.ai/v2/wan-2-1-i2v-720",
+  "bytedance/seedance-v1.5-pro-i2v": "https://api.runpod.ai/v2/seedance-v1-5-pro-i2v",
+  "openai/sora-2-pro-i2v": "https://api.runpod.ai/v2/sora-2-pro-i2v",
+  "openai/sora-2-i2v": "https://api.runpod.ai/v2/sora-2-i2v"
+};
 var MODEL_ID_TO_OPENAI_NAME = {
   "qwen/qwen3-32b-awq": "Qwen/Qwen3-32B-AWQ",
   "deepcogito/cogito-671b-v2.1-fp8": "deepcogito/cogito-671b-v2.1-FP8",
@@ -1005,7 +1320,7 @@ function parseRunpodConsoleEndpointId(modelIdOrUrl) {
 }
 function createRunpod(options = {}) {
   const getHeaders = () => ({
-    Authorization: `Bearer ${(0, import_provider_utils5.loadApiKey)({
+    Authorization: `Bearer ${(0, import_provider_utils6.loadApiKey)({
       apiKey: options.apiKey,
       environmentVariableName: "RUNPOD_API_KEY",
       description: "Runpod"
@@ -1035,7 +1350,7 @@ function createRunpod(options = {}) {
     }
     return {
       provider: `runpod.${modelType}`,
-      url: ({ path }) => `${(0, import_provider_utils5.withoutTrailingSlash)(baseURL)}${path}`,
+      url: ({ path }) => `${(0, import_provider_utils6.withoutTrailingSlash)(baseURL)}${path}`,
       headers: getHeaders,
       fetch: runpodFetch
     };
@@ -1055,13 +1370,11 @@ function createRunpod(options = {}) {
     });
   };
   const createImageModel = (modelId) => {
-    let baseURL;
-    if (options.baseURL) {
-      baseURL = options.baseURL;
-    } else {
-      baseURL = IMAGE_MODEL_ID_TO_ENDPOINT_URL[modelId] || deriveEndpointURL(modelId);
-    }
-    return new RunpodImageModel(modelId, {
+    const endpointIdFromConsole = parseRunpodConsoleEndpointId(modelId);
+    const normalizedModelId = endpointIdFromConsole ?? modelId;
+    const mappedBaseURL = IMAGE_MODEL_ID_TO_ENDPOINT_URL[normalizedModelId];
+    const baseURL = options.baseURL ?? mappedBaseURL ?? (normalizedModelId.startsWith("http") ? normalizedModelId : `https://api.runpod.ai/v2/${normalizedModelId}`);
+    return new RunpodImageModel(normalizedModelId, {
       provider: "runpod.image",
       baseURL,
       headers: getHeaders,
@@ -1092,6 +1405,18 @@ function createRunpod(options = {}) {
       fetch: runpodFetch
     });
   };
+  const createVideoModel = (modelId) => {
+    const endpointIdFromConsole = parseRunpodConsoleEndpointId(modelId);
+    const normalizedModelId = endpointIdFromConsole ?? modelId;
+    const mappedBaseURL = VIDEO_MODEL_ID_TO_ENDPOINT_URL[normalizedModelId];
+    const baseURL = mappedBaseURL ?? (normalizedModelId.startsWith("http") ? normalizedModelId : `https://api.runpod.ai/v2/${normalizedModelId}`);
+    return new RunpodVideoModel(normalizedModelId, {
+      provider: "runpod.video",
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    });
+  };
   const provider = (modelId) => createChatModel(modelId);
   provider.completionModel = createCompletionModel;
   provider.languageModel = createChatModel;
@@ -1102,6 +1427,8 @@ function createRunpod(options = {}) {
   provider.speech = createSpeechModel;
   provider.transcriptionModel = createTranscriptionModel;
   provider.transcription = createTranscriptionModel;
+  provider.videoModel = createVideoModel;
+  provider.video = createVideoModel;
   return provider;
 }
 var runpod = createRunpod();