climage 0.6.0 → 0.6.2

package/dist/index.js

@@ -417,6 +417,11 @@ function isKlingV3Model(model) {
 function isViduModel(model) {
   return model.includes("/vidu/");
 }
+function usesImageUrlsArray(model) {
+  if (model.includes("/hunyuan-image/") && model.includes("/edit")) return true;
+  if (model.includes("/kling-image/")) return true;
+  return false;
+}
 function selectVideoModel(req) {
   if (req.model) return req.model;
   if (req.startFrame && req.endFrame) {
@@ -504,8 +509,13 @@ function buildImageInput(req) {
   if (imageSize) input.image_size = imageSize;
   if (req.n) input.num_images = req.n;
   if (req.inputImages?.[0]) {
-    input.image_url = req.inputImages[0];
-    input.strength = 0.75;
+    const model = req.model ?? "";
+    if (usesImageUrlsArray(model)) {
+      input.image_urls = req.inputImages.slice(0, 3);
+    } else {
+      input.image_url = req.inputImages[0];
+      input.strength = 0.75;
+    }
   }
   return input;
 }
@@ -1284,11 +1294,22 @@ async function generateVercelImage(req, apiKey) {
   const gw = makeGateway(apiKey);
   const model = req.model ?? DEFAULT_IMAGE_MODEL2;
   log5("Starting image generation, model:", model);
+  const hasInputImages = req.inputImages && req.inputImages.length > 0;
+  let prompt;
+  if (hasInputImages) {
+    const images2 = req.inputImages.map(
+      (img) => img.startsWith("data:") ? dataUriToUint8Array(img) : Uint8Array.from(Buffer.from(img))
+    );
+    prompt = { images: images2, text: req.prompt };
+    log5("Using image editing mode with", images2.length, "input image(s)");
+  } else {
+    prompt = req.prompt;
+  }
   log5("Calling generateImage...");
   const startTime = Date.now();
   const result = await sdkGenerateImage({
     model: gw.image(model),
-    prompt: req.prompt,
+    prompt,
     ...req.aspectRatio ? { aspectRatio: req.aspectRatio } : {},
     n: req.n
   });
@@ -1312,7 +1333,7 @@ var vercelCapabilities = {
   supportsCustomAspectRatio: true,
   supportsVideoInterpolation: false,
   videoDurationRange: [1, 15],
-  supportsImageEditing: false
+  supportsImageEditing: true
 };
 var vercelProvider = {
   id: "vercel",