@shop-os/modelkit 0.2.0 → 0.2.2

package/dist/index.cjs

@@ -64,6 +64,7 @@ __export(index_exports, {
   keysFromEnv: () => keysFromEnv,
   languageModel: () => languageModel,
   liveCatalog: () => liveCatalog,
+  liveState: () => liveState,
   mergeTextProviderOptions: () => mergeTextProviderOptions,
   nearestAspectRatio: () => nearestAspectRatio,
   openaiImageAdapter: () => openaiImageAdapter,
@@ -211,7 +212,7 @@ var GEMINI_RATIOS = [
   "5:4",
   "adaptive"
 ];
-function geminiImage(id, label, aliases, sizes, display, exposure) {
+function geminiImage(id, label, aliases, sizes, display, exposure, references) {
   const model = {
     id,
     aliases,
@@ -224,7 +225,8 @@ function geminiImage(id, label, aliases, sizes, display, exposure) {
     constraints: {
       aspectRatios: [...GEMINI_RATIOS],
       sizes,
-      supports: ["safetySettings", "temperature"]
+      supports: ["safetySettings", "temperature"],
+      ...references ? { references } : {}
     },
     defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { ...IMAGE_USAGE, slug: "gemini" },
@@ -244,7 +246,8 @@ var IMAGE_MODELS = [
       description: "Create stunning, production-quality visuals with exceptional realism, detail, and typography. Best for campaigns, e-commerce, product photography, and hero creatives.",
       logo: "https://d31vvfpk2ppdyq.cloudfront.net/assets/image/192485d2-a76a-4657-9c06-930b2c70badd_compressed.png"
     },
-    "catalog"
+    "catalog",
+    { images: 14 }
   ),
   geminiImage(
     "gemini-3.1-flash-image-preview",
@@ -257,7 +260,8 @@ var IMAGE_MODELS = [
       description: "Generate high-quality images at speed. Perfect for brainstorming, moodboards, concept testing, creative exploration, and large batch workflows.",
       logo: "https://d31vvfpk2ppdyq.cloudfront.net/assets/image/192485d2-a76a-4657-9c06-930b2c70badd_compressed.png"
     },
-    "catalog"
+    "catalog",
+    { images: 14 }
   ),
   geminiImage(
     "gemini-3.1-flash-lite-image",
@@ -269,14 +273,15 @@ var IMAGE_MODELS = [
       description: "The fastest and most affordable image model. Ideal for rapid drafts, bulk exploration, and iterative concept testing at 1K resolution.",
       logo: "https://d31vvfpk2ppdyq.cloudfront.net/assets/image/192485d2-a76a-4657-9c06-930b2c70badd_compressed.png"
     },
-    "catalog"
+    "catalog",
+    { images: 14 }
   ),
   // trigger-workflows validates 1K only; pricing seeds 2K/4K rows so allow all three.
   geminiImage(
     "gemini-2.5-flash-image",
     "Gemini 2.5 Flash Image",
     [],
-    ["1K", "2K", "4K"],
+    ["1K"],
     {
       provider: "google",
       description: "Nano Banana (legacy Gemini 2.5 Flash Image) native image generation and editing model for high-volume, low-latency creative workflows."
@@ -307,7 +312,7 @@ var IMAGE_MODELS = [
       sizes: ["1K", "2K", "4K"],
       qualityOptions: ["auto", "low", "medium", "high"],
       references: { images: 16 },
-      supports: ["quality", "count"]
+      supports: ["quality", "count", "outputFormat"]
     },
     defaults: { size: "1K", aspectRatio: "adaptive", quality: "auto" },
     usage: { ...IMAGE_USAGE, slug: "gpt" },
@@ -327,7 +332,7 @@ var IMAGE_MODELS = [
     adapter: "openai-image",
     providerModelId: "gpt-image-2.5-sunburst",
     lifecycle: "active",
-    exposure: "internal",
+    exposure: "catalog",
     constraints: {
       exactSizes: {
         "1:1": "1024x1024",
@@ -340,9 +345,9 @@ var IMAGE_MODELS = [
       },
       aspectRatios: ["1:1", "3:2", "2:3", "16:9", "9:16", "4:5", "5:4", "adaptive"],
       sizes: ["1K", "2K", "4K"],
-      qualityOptions: ["auto", "low", "medium", "high"],
+      qualityOptions: ["auto", "low", "medium", "high", "xhigh", "max"],
       references: { images: 16 },
-      supports: ["quality", "count"]
+      supports: ["quality", "count", "outputFormat"]
     },
     defaults: { size: "1K", aspectRatio: "adaptive", quality: "auto" },
     usage: { ...IMAGE_USAGE },
@@ -361,7 +366,7 @@ var IMAGE_MODELS = [
     adapter: "openai-image",
     providerModelId: "gpt-image-2.5-flare",
     lifecycle: "active",
-    exposure: "internal",
+    exposure: "catalog",
     constraints: {
       exactSizes: {
         "1:1": "1024x1024",
@@ -374,9 +379,9 @@ var IMAGE_MODELS = [
       },
       aspectRatios: ["1:1", "3:2", "2:3", "16:9", "9:16", "4:5", "5:4", "adaptive"],
       sizes: ["1K", "2K", "4K"],
-      qualityOptions: ["auto", "low", "medium", "high"],
+      qualityOptions: ["auto", "low", "medium", "high", "xhigh", "max"],
       references: { images: 16 },
-      supports: ["quality", "count"]
+      supports: ["quality", "count", "outputFormat"]
     },
     defaults: { size: "1K", aspectRatio: "adaptive", quality: "auto" },
     usage: { ...IMAGE_USAGE },
@@ -406,8 +411,6 @@ var IMAGE_MODELS = [
         "2:3",
         "3:4",
         "9:16",
-        "4:5",
-        "5:4",
         "9:21",
         "adaptive"
       ],
@@ -439,7 +442,7 @@ var IMAGE_MODELS = [
       references: { images: 10 },
       // Ark's Images API has no output-format, quality or count field; only
       // the seed is forwarded, so anything else must warn, not vanish.
-      supports: ["seed"]
+      supports: ["seed", "outputFormat"]
     },
     defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { ...IMAGE_USAGE },
@@ -475,7 +478,7 @@ var IMAGE_MODELS = [
         "21:9",
         "9:21"
       ],
-      sizes: ["1K", "2K", "4K"],
+      sizes: ["2K", "4K"],
       references: { images: 10 },
       supports: ["seed"]
     },
@@ -527,7 +530,7 @@ var IMAGE_MODELS = [
       aspectRatios: ["1:1", "2:3", "3:2", "4:3", "3:4", "4:5", "5:4", "9:16", "16:9", "21:9"],
       sizes: ["1K", "2K", "4K"],
       requires: ["references"],
-      supports: ["seed"]
+      supports: []
     },
     defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { ...IMAGE_USAGE },
@@ -552,7 +555,7 @@ var IMAGE_MODELS = [
       aspectRatios: ["1:1", "2:3", "3:4", "4:3", "9:16", "16:9", "21:9"],
       sizes: ["1K"],
       requires: ["references"],
-      supports: ["seed"]
+      supports: []
     },
     defaults: { size: "1K", aspectRatio: "1:1" },
     usage: { ...IMAGE_USAGE },
@@ -604,7 +607,6 @@ var IMAGE_MODELS = [
       aspectRatios: ["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3"],
       sizes: [],
       references: { images: 1 },
-      requires: ["references"],
       supports: []
     },
     defaults: { aspectRatio: "1:1" },
@@ -1323,7 +1325,7 @@ var VIDEO_MODELS = [
       resolutions: ["720p", "1080p", "4k"],
       durations: [8],
       references: { images: 3 },
-      supports: ["lastFrame"]
+      supports: ["lastFrame", "seed"]
     },
     defaults: { durationSeconds: 8, resolution: "720p", aspectRatio: "16:9" },
     usage: { ...VIDEO_USAGE, slug: "veo" },
@@ -1351,9 +1353,9 @@ var VIDEO_MODELS = [
     constraints: {
       aspectRatios: [...OMNI_RATIOS],
       resolutions: ["720p", "1080p", "4k"],
-      durations: [4, 8],
+      durations: [4, 6, 8],
       references: { images: 3 },
-      supports: ["lastFrame"]
+      supports: ["lastFrame", "seed"]
     },
     defaults: { durationSeconds: 8, resolution: "720p", aspectRatio: "16:9" },
     usage: { ...VIDEO_USAGE },
@@ -1380,9 +1382,8 @@ var VIDEO_MODELS = [
     constraints: {
       aspectRatios: [...OMNI_RATIOS],
       resolutions: ["720p", "1080p"],
-      durations: [4, 8],
-      references: { images: 3 },
-      supports: ["lastFrame"]
+      durations: [4, 6, 8],
+      supports: ["lastFrame", "seed"]
     },
     defaults: { durationSeconds: 8, resolution: "720p", aspectRatio: "16:9" },
     usage: { ...VIDEO_USAGE },
@@ -1414,7 +1415,7 @@ var VIDEO_MODELS = [
       // reference clip keeps its own framing.
       aspectRatios: [...OMNI_RATIOS],
       durations: [3, 4, 5, 6, 7, 8, 9, 10],
-      references: { images: 6, videos: 1 },
+      references: { images: 6, videos: 3 },
       supports: []
     },
     defaults: {},
@@ -1440,7 +1441,7 @@ var VIDEO_MODELS = [
       aspectRatios: [...OMNI_RATIOS],
       resolutions: ["360p", "720p", "1080p", "4k"],
       durations: [3, 4, 5, 6, 7, 8, 9, 10],
-      references: { images: 6, videos: 1 },
+      references: { images: 6, videos: 3 },
       supports: []
     },
     defaults: {},
@@ -1469,7 +1470,7 @@ var VIDEO_MODELS = [
       durations: { min: 4, max: 15 },
       durationPresets: [4, 6, 8, 10, 12, 15],
       references: { images: 9, videos: 3, audios: 3, audioOnly: false },
-      ratioIgnoredWith: ["firstFrame"],
+      ratioIgnoredWith: ["firstFrame", "sourceVideo"],
       supports: ["generateAudio", "seed", "lastFrame", "watermark"]
     },
     defaults: {
@@ -1501,7 +1502,7 @@ var VIDEO_MODELS = [
       resolutions: ["480p", "720p"],
       durations: { min: 4, max: 15 },
       references: { images: 9, videos: 3, audios: 3, audioOnly: false },
-      ratioIgnoredWith: ["firstFrame"],
+      ratioIgnoredWith: ["firstFrame", "sourceVideo"],
       supports: ["generateAudio", "seed", "lastFrame", "watermark"]
     },
     defaults: {
@@ -1534,7 +1535,7 @@ var VIDEO_MODELS = [
       resolutions: ["480p", "720p", "1080p"],
       durations: { min: 4, max: 30 },
       references: { images: 30, videos: 10, audios: 10 },
-      ratioIgnoredWith: ["firstFrame"],
+      ratioIgnoredWith: ["firstFrame", "sourceVideo"],
       supports: ["generateAudio", "seed", "lastFrame", "watermark"]
     },
     defaults: {
@@ -1653,18 +1654,7 @@ var VIDEO_MODELS = [
     lifecycle: "active",
     exposure: "catalog",
     constraints: {
-      aspectRatios: [
-        "1:1",
-        "16:9",
-        "9:16",
-        "4:3",
-        "3:4",
-        "3:2",
-        "2:3",
-        "21:9",
-        "4:5",
-        "adaptive"
-      ],
+      aspectRatios: [...KLING_RATIOS],
       resolutions: ["720p", "1080p"],
       durations: { min: 3, max: 15 },
       references: { images: 7 },
@@ -2126,13 +2116,18 @@ var CatalogSchema = import_zod.z.array(CatalogModelSchemaBase).superRefine((rows
 });
 
 // src/catalog/live.ts
-var liveCatalog = new Catalog(MODELS);
+var KEY = /* @__PURE__ */ Symbol.for("@shop-os/modelkit/live-catalog");
+var store = globalThis;
+var liveState = store[KEY] ??= {
+  catalog: new Catalog(MODELS),
+  memo: void 0,
+  timer: void 0,
+  revision: void 0,
+  origin: "snapshot"
+};
+var liveCatalog = liveState.catalog;
 
 // src/catalog/load.ts
-var memo;
-var timer;
-var revision;
-var origin = "snapshot";
 function rowCount() {
   return liveCatalog.list({ lifecycle: ["active", "deprecated", "retired"] }).length;
 }
@@ -2162,7 +2157,7 @@ function logPrimed(logger, result) {
 async function fetchRemote(source, logger) {
   const fetchImpl = source.fetch ?? globalThis.fetch;
   const headers2 = authHeaders(source.auth);
-  if (revision !== void 0) headers2["If-None-Match"] = `"${revision}"`;
+  if (liveState.revision !== void 0) headers2["If-None-Match"] = `"${liveState.revision}"`;
   const res = await fetchImpl(`${source.baseUrl}/models/catalog`, {
     headers: headers2,
     signal: AbortSignal.timeout(source.timeoutMs ?? 1e4)
@@ -2186,16 +2181,16 @@ async function prime(source, logger) {
   try {
     const fetched = await fetchRemote(source, logger);
     if ("kept" in fetched) {
-      const result2 = { origin, rowCount: rowCount() };
-      if (revision !== void 0) result2.revision = revision;
+      const result2 = { origin: liveState.origin, rowCount: rowCount() };
+      if (liveState.revision !== void 0) result2.revision = liveState.revision;
       logPrimed(logger, result2);
       return result2;
     }
     liveCatalog.replace(fetched.rows);
-    origin = "remote";
-    revision = fetched.revision;
+    liveState.origin = "remote";
+    liveState.revision = fetched.revision;
     await writeCache(source, logger, fetched.rows, fetched.revision);
-    const result = { origin, rowCount: rowCount() };
+    const result = { origin: liveState.origin, rowCount: rowCount() };
     if (fetched.revision !== void 0) result.revision = fetched.revision;
     logPrimed(logger, result);
     return result;
@@ -2207,19 +2202,23 @@ async function prime(source, logger) {
       if (!cached) throw new Error("no cached catalog");
       const { rows, revision: rev } = parseEnvelope(JSON.parse(cached));
       liveCatalog.replace(rows);
-      origin = "last-good";
-      revision = rev;
-      const result = { origin, rowCount: rowCount(), error };
+      liveState.origin = "last-good";
+      liveState.revision = rev;
+      const result = {
+        origin: liveState.origin,
+        rowCount: rowCount(),
+        error
+      };
       if (rev !== void 0) result.revision = rev;
       logPrimed(logger, result);
       return result;
     } catch (cacheErr) {
       logger.warn("[modelkit] catalog cache failed, using snapshot", cacheErr);
       liveCatalog.replace([...MODELS]);
-      origin = "snapshot";
-      revision = void 0;
+      liveState.origin = "snapshot";
+      liveState.revision = void 0;
       const result = {
-        origin,
+        origin: liveState.origin,
         rowCount: rowCount(),
         error
       };
@@ -2233,35 +2232,36 @@ async function refresh(source, logger) {
     const fetched = await fetchRemote(source, logger);
     if ("kept" in fetched) return;
     liveCatalog.replace(fetched.rows);
-    origin = "remote";
-    revision = fetched.revision;
+    liveState.origin = "remote";
+    liveState.revision = fetched.revision;
     await writeCache(source, logger, fetched.rows, fetched.revision);
   } catch (err) {
     logger.warn("[modelkit] catalog refresh failed, keeping current rows", err);
   }
 }
 function primeCatalog(source) {
+  const memo = liveState.memo;
   if (memo) return memo;
   const logger = source.logger ?? console;
-  memo = prime(source, logger).then((result) => {
+  liveState.memo = prime(source, logger).then((result) => {
     const ttlMs = source.ttlMs ?? 3e5;
-    if (ttlMs > 0 && timer === void 0) {
-      timer = setInterval(() => void refresh(source, logger), ttlMs);
-      const maybeUnref = timer;
+    if (ttlMs > 0 && liveState.timer === void 0) {
+      liveState.timer = setInterval(() => void refresh(source, logger), ttlMs);
+      const maybeUnref = liveState.timer;
       maybeUnref.unref?.();
     }
     return result;
   });
-  return memo;
+  return liveState.memo;
 }
 function resetLiveCatalog() {
-  if (timer !== void 0) {
-    clearInterval(timer);
-    timer = void 0;
+  if (liveState.timer !== void 0) {
+    clearInterval(liveState.timer);
+    liveState.timer = void 0;
   }
-  memo = void 0;
-  revision = void 0;
-  origin = "snapshot";
+  liveState.memo = void 0;
+  liveState.revision = void 0;
+  liveState.origin = "snapshot";
   liveCatalog.replace([...MODELS]);
 }
 
@@ -2805,12 +2805,12 @@ function sleep(ms, signal) {
       rejectAborted();
       return;
     }
-    let timer2;
+    let timer;
     const onAbort = () => {
-      if (timer2 !== void 0) clearTimeout(timer2);
+      if (timer !== void 0) clearTimeout(timer);
       rejectAborted();
     };
-    timer2 = setTimeout(() => {
+    timer = setTimeout(() => {
       signal?.removeEventListener("abort", onAbort);
       resolve();
     }, ms);
@@ -3896,6 +3896,18 @@ async function generateSeedreamImage(req, ctx) {
     watermark: false
   };
   if (req.seed !== void 0) body.seed = req.seed;
+  if (req.outputFormat !== void 0) {
+    if (req.outputFormat === "webp") {
+      ctx.warnings.push({
+        code: "option_ignored",
+        field: "outputFormat",
+        from: "webp",
+        to: "png",
+        message: `${modelId} accepts png or jpeg output; using png`
+      });
+    }
+    body.output_format = req.outputFormat === "jpeg" ? "jpeg" : "png";
+  }
   const firstRef = refs[0];
   if (refs.length === 1 && firstRef !== void 0) body.image = firstRef;
   else if (refs.length > 1) body.image = refs;
@@ -5039,6 +5051,7 @@ async function generateVeoVideo(req, ctx, createGenAIClient) {
   const config = { durationSeconds: duration, personGeneration };
   if (req.aspectRatio !== void 0) config.aspectRatio = req.aspectRatio;
   if (req.resolution !== void 0) config.resolution = req.resolution;
+  if (req.seed !== void 0) config.seed = req.seed;
   if (lastFrame !== void 0) config.lastFrame = lastFrame;
   if (referenceImages !== void 0) config.referenceImages = referenceImages;
   const requestBody = {
@@ -6182,6 +6195,19 @@ var openaiImageAdapter = {
     const quality = req.quality ?? (isEdit ? "medium" : "auto");
     const sdkQuality = quality;
     const count = req.count !== void 0 && req.count > 1 ? { n: req.count } : {};
+    let outputFormat = req.outputFormat;
+    if (outputFormat === "webp") {
+      ctx.warnings.push({
+        code: "option_ignored",
+        field: "outputFormat",
+        from: "webp",
+        to: "png",
+        message: `${ctx.model.id} returns PNG for output_format webp; using png`
+      });
+      outputFormat = "png";
+    }
+    const format = outputFormat !== void 0 ? { output_format: outputFormat } : {};
+    const mime = outputFormat === "jpeg" ? "image/jpeg" : "image/png";
     const retryOptions = {};
     if (ctx.signal !== void 0) retryOptions.signal = ctx.signal;
     const response = await withRetry(async () => {
@@ -6194,7 +6220,8 @@ var openaiImageAdapter = {
                 prompt: req.prompt,
                 size,
                 quality: sdkQuality,
-                ...count
+                ...count,
+                ...format
               },
               { signal: ctx.signal }
             );
@@ -6204,20 +6231,21 @@ var openaiImageAdapter = {
               model: ctx.model.providerModelId,
               prompt: req.prompt,
               quality: sdkQuality,
-              ...count
+              ...count,
+              ...format
             },
             { signal: ctx.signal }
           );
         }
         const files = await Promise.all(
           refs.slice(0, MAX_EDIT_REFERENCES).map(async (ref, i) => {
-            const { bytes, mime } = await mediaRefToBytes(ref, ctx.fetch, {
+            const { bytes, mime: mime2 } = await mediaRefToBytes(ref, ctx.fetch, {
               signal: ctx.signal,
               provider: "openai-image",
               model: ctx.model.id
             });
-            const ext = mime.includes("jpeg") ? "jpg" : mime.includes("webp") ? "webp" : "png";
-            return new File([bytes], `ref-${i}.${ext}`, { type: mime });
+            const ext = mime2.includes("jpeg") ? "jpg" : mime2.includes("webp") ? "webp" : "png";
+            return new File([bytes], `ref-${i}.${ext}`, { type: mime2 });
           })
         );
         const single = files.length === 1 ? files[0] : void 0;
@@ -6230,7 +6258,8 @@ var openaiImageAdapter = {
               prompt: req.prompt,
               size,
               quality: sdkQuality,
-              ...count
+              ...count,
+              ...format
             },
             { signal: ctx.signal }
           );
@@ -6241,7 +6270,8 @@ var openaiImageAdapter = {
             image: imageInput,
             prompt: req.prompt,
             quality: sdkQuality,
-            ...count
+            ...count,
+            ...format
           },
           { signal: ctx.signal }
         );
@@ -6250,7 +6280,7 @@ var openaiImageAdapter = {
       }
     }, retryOptions);
     const images = (response.data ?? []).flatMap(
-      (item) => item.b64_json ? [{ bytes: base64ToBytes(item.b64_json), mime: "image/png" }] : []
+      (item) => item.b64_json ? [{ bytes: base64ToBytes(item.b64_json), mime }] : []
     );
     if (images.length === 0)
       throw classifyHttpError(
@@ -7399,6 +7429,7 @@ var describe = (modelId) => createDefaultModelKit().describe(modelId);
   keysFromEnv,
   languageModel,
   liveCatalog,
+  liveState,
   mergeTextProviderOptions,
   nearestAspectRatio,
   openaiImageAdapter,