@shop-os/modelkit 0.2.23 → 0.2.24

package/dist/index.js

@@ -396,6 +396,32 @@ var IMAGE_MODELS = [
       logo: "https://d31vvfpk2ppdyq.cloudfront.net/assets/characters/1787909065794-k9a7zz.png"
     }
   },
+  {
+    // References switch to the edit endpoint: the first is the image to edit,
+    // the rest (up to 4) guide it.
+    id: "ideogram-4.5",
+    aliases: [],
+    label: "Ideogram 4.5",
+    type: "image",
+    adapter: "fal",
+    providerModelId: "ideogram/v4.5",
+    lifecycle: "active",
+    exposure: ["cowork", "space_builder"],
+    constraints: {
+      aspectRatios: ["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3", "5:4", "4:5"],
+      sizes: ["1K", "2K"],
+      qualityOptions: ["auto", "low", "medium", "high"],
+      references: { images: 5 },
+      supports: ["quality", "count", "seed"]
+    },
+    defaults: { size: "2K", aspectRatio: "1:1", quality: "medium", count: 1 },
+    usage: { ...IMAGE_USAGE, slug: "ideogram" },
+    pricingKind: "image_size",
+    display: {
+      provider: "ideogram",
+      description: "Ideogram's precise edit model: repeated edits keep the untouched pixels and colours intact, at the source's native resolution. Also generates from text."
+    }
+  },
   {
     id: "seedream-5-0-pro",
     aliases: ["seedream-4-5"],
@@ -3585,6 +3611,7 @@ function createFalAdapter(deps = defaultDeps) {
     id: "fal",
     generateImage: (req, ctx) => {
       throwIfAborted(ctx.signal);
+      if (isIdeogram(ctx.model.providerModelId)) return generateIdeogramImage(req, ctx, deps);
       return generateMuseImage(req, ctx, deps);
     },
     generateVideo: (req, ctx) => {
@@ -3841,6 +3868,72 @@ async function generateMuseImage(req, ctx, deps) {
     raw: payload
   };
 }
+var IDEOGRAM_MODEL_IDS = /* @__PURE__ */ new Set(["ideogram/v4.5"]);
+function isIdeogram(providerModelId) {
+  return IDEOGRAM_MODEL_IDS.has(providerModelId.toLowerCase());
+}
+var IDEOGRAM_SIZES = {
+  "1:1": { "1K": "1024x1024", "2K": "2048x2048" },
+  "16:9": { "1K": "1280x720", "2K": "2560x1440" },
+  "9:16": { "1K": "720x1280", "2K": "1440x2560" },
+  "4:3": { "1K": "1152x864", "2K": "2304x1728" },
+  "3:4": { "1K": "864x1152", "2K": "1728x2304" },
+  "3:2": { "1K": "1248x832", "2K": "2496x1664" },
+  "2:3": { "1K": "832x1248", "2K": "1664x2496" },
+  "5:4": { "1K": "1120x896", "2K": "2240x1792" },
+  "4:5": { "1K": "896x1120", "2K": "1792x2240" }
+};
+function parseExactSize(value) {
+  const match = /^(\d+)x(\d+)$/.exec(value ?? "");
+  return match ? { width: Number(match[1]), height: Number(match[2]) } : void 0;
+}
+function ideogramImageSize(req) {
+  const exact = parseExactSize(req.exactSize);
+  if (exact) return exact;
+  if (req.aspectRatio === void 0 || req.aspectRatio === "adaptive") return void 0;
+  return parseExactSize(IDEOGRAM_SIZES[req.aspectRatio]?.[req.size === "1K" ? "1K" : "2K"]);
+}
+async function generateIdeogramImage(req, ctx, deps) {
+  const [source, ...references] = (req.references ?? []).map(refToUrl);
+  const input = { prompt: req.prompt, num_images: req.count ?? 1 };
+  if (req.quality !== void 0 && req.quality !== "auto") input.quality = req.quality;
+  if (req.seed !== void 0) input.seed = req.seed;
+  const id = ctx.model.providerModelId;
+  let endpoint = id;
+  if (source === void 0) {
+    input.enable_prompt_expansion = false;
+    const size = ideogramImageSize(req);
+    if (size) input.image_size = size;
+  } else {
+    endpoint = `${id}/edit`;
+    input.image_url = source;
+    input.edit_precision = "high";
+    if (references.length > 0) input.reference_image_urls = references;
+  }
+  const payload = await runJob(ctx, deps, endpoint, input);
+  const parsed = museImageResultSchema.safeParse(payload);
+  const images = parsed.success ? parsed.data.images ?? [] : [];
+  const files = await Promise.all(
+    images.flatMap((img) => {
+      if (img.url === void 0) return [];
+      return [decodeImage(ctx, img.url, img.content_type ?? void 0)];
+    })
+  );
+  if (files.length === 0) {
+    throw new ModelKitError({
+      code: "provider_rejected",
+      message: "fal Ideogram returned no image",
+      provider: "fal",
+      model: ctx.model.id,
+      retryable: false
+    });
+  }
+  return {
+    images: files.map((f) => ({ bytes: f.bytes, mime: f.mime })),
+    usage: imageUsage(ctx, req, files.length),
+    raw: payload
+  };
+}
 var H3_MAX_MODEL_IDS = /* @__PURE__ */ new Set([
   "minimax/h3-max",
   "minimax/h3-max/reference-to-video"