@shop-os/modelkit 0.2.23 → 0.2.25

package/dist/index.cjs

@@ -81,6 +81,7 @@ __export(index_exports, {
   liveCatalog: () => liveCatalog,
   liveState: () => liveState,
   mergeTextProviderOptions: () => mergeTextProviderOptions,
+  missingTierKeys: () => missingTierKeys,
   nearestAspectRatio: () => nearestAspectRatio,
   normalizeExposure: () => normalizeExposure,
   openaiImageAdapter: () => openaiImageAdapter,
@@ -506,6 +507,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"],
@@ -3695,6 +3722,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) => {
@@ -3951,6 +3979,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"
@@ -8002,10 +8096,22 @@ function hasAll(keys, names) {
 function hasKey(keys, name) {
   return hasAll(keys, [name]);
 }
+function missingTierKeys(config, tier, names) {
+  return names.filter((name) => !hasAll(config.keys[tier], [name]));
+}
 function missingEnterpriseKeys(config, names) {
-  return names.filter((name) => !hasAll(config.keys.enterprise, [name]));
+  return missingTierKeys(config, "enterprise", names);
 }
 function resolveKeys(config, names, tier = "pro") {
+  if (tier === "sloosh") {
+    if (names.length === 0) return config.keys.sloosh ?? {};
+    if (hasAll(config.keys.sloosh, names)) return config.keys.sloosh;
+    throw new ModelKitError({
+      code: "auth",
+      message: `Missing Sloosh API key for ${names.join(", ")}`,
+      retryable: false
+    });
+  }
   if (tier === "enterprise" && hasAll(config.keys.enterprise, names))
     return config.keys.enterprise;
   if (names.length > 0 && !hasAll(config.keys.pro, names)) {
@@ -8194,6 +8300,7 @@ function resolveAnthropicTransport(model, config, tier, warn) {
   const preferred = preferredAnthropicTransport(model, config.routing ?? {});
   const usable = (transport) => {
     const name = transport === "bedrock" ? "aws" : "anthropic";
+    if (tier === "sloosh") return hasKey(config.keys.sloosh, name);
     return hasKey(config.keys.pro, name) || tier === "enterprise" && hasKey(config.keys.enterprise, name);
   };
   if (usable(preferred)) return preferred;
@@ -8775,11 +8882,19 @@ function createModelKit(config, adapters) {
 }
 
 // src/default.ts
+function withSharedAws(keys) {
+  const { pro, sloosh } = keys;
+  if (!sloosh || sloosh.aws || !pro.aws) return keys;
+  return { ...keys, sloosh: { ...sloosh, aws: pro.aws } };
+}
 function createDefaultModelKit(config = {}) {
-  const keys = config.keys ?? {
-    pro: keysFromEnv(process.env),
-    enterprise: keysFromEnv(process.env, "ENTERPRISE_")
-  };
+  const keys = withSharedAws(
+    config.keys ?? {
+      pro: keysFromEnv(process.env),
+      enterprise: keysFromEnv(process.env, "ENTERPRISE_"),
+      sloosh: keysFromEnv(process.env, "SLOOSH_")
+    }
+  );
   return createModelKit({ ...config, keys }, DEFAULT_ADAPTERS);
 }
 var generateText2 = (options, call) => createDefaultModelKit().generateText(options, call);
@@ -8844,6 +8959,7 @@ var describe = (modelId) => createDefaultModelKit().describe(modelId);
   liveCatalog,
   liveState,
   mergeTextProviderOptions,
+  missingTierKeys,
   nearestAspectRatio,
   normalizeExposure,
   openaiImageAdapter,