@shop-os/modelkit 0.3.1 → 0.3.3

package/dist/catalog.d.cts

@@ -8,7 +8,7 @@ import { z } from 'zod';
 
 /** The AI SDK video model spelling: a catalog id or a provider video model instance. */
 type VideoModel = Parameters<typeof experimental_generateVideo>[0]["model"];
-type AspectRatio = "1:1" | "16:9" | "9:16" | "4:3" | "3:4" | "3:2" | "2:3" | "4:5" | "5:4" | "21:9" | "9:21" | "97:30" | "adaptive";
+type AspectRatio = "1:1" | "16:9" | "9:16" | "4:3" | "3:4" | "3:2" | "2:3" | "4:5" | "5:4" | "21:9" | "9:21" | "97:30" | "1:4" | "4:1" | "1:8" | "8:1" | "adaptive";
 type ImageSize = "512" | "1K" | "2K" | "4K";
 type VideoResolution = "360p" | "480p" | "720p" | "768p" | "1080p" | "2k" | "4k";
 /** "xhigh" and "max" exist only on gpt-image-2.5; gpt-image-2 rejects them at the provider. */
@@ -393,6 +393,8 @@ interface CatalogCapabilities {
     videoEdit?: boolean;
     /** Video rows: first/last frames and reference images are mutually exclusive inputs. */
     framesExcludeReferences?: boolean;
+    /** Video rows: a source video is never sent with first/last frames or reference images. */
+    sourceVideoExcludesFrames?: boolean;
     /** Video rows: a last frame is accepted only together with a first frame. */
     lastFrameRequiresFirstFrame?: boolean;
     /** Text rows. `reasoning`: the model streams reasoning the UI should surface. */
@@ -689,6 +691,10 @@ declare const CatalogModelSchemaBase: z.ZodObject<{
             "21:9": "21:9";
             "9:21": "9:21";
             "97:30": "97:30";
+            "1:4": "1:4";
+            "4:1": "4:1";
+            "1:8": "1:8";
+            "8:1": "8:1";
             adaptive: "adaptive";
         }>>>;
         sizes: z.ZodOptional<z.ZodArray<z.ZodEnum<{
@@ -824,6 +830,10 @@ declare const CatalogModelSchemaBase: z.ZodObject<{
             "21:9": "21:9";
             "9:21": "9:21";
             "97:30": "97:30";
+            "1:4": "1:4";
+            "4:1": "4:1";
+            "1:8": "1:8";
+            "8:1": "8:1";
             adaptive: "adaptive";
         }>>;
         size: z.ZodOptional<z.ZodEnum<{
@@ -986,6 +996,7 @@ declare const CatalogModelSchemaBase: z.ZodObject<{
         acceptsSourceVideo: z.ZodOptional<z.ZodBoolean>;
         videoEdit: z.ZodOptional<z.ZodBoolean>;
         framesExcludeReferences: z.ZodOptional<z.ZodBoolean>;
+        sourceVideoExcludesFrames: z.ZodOptional<z.ZodBoolean>;
         lastFrameRequiresFirstFrame: z.ZodOptional<z.ZodBoolean>;
         reasoning: z.ZodOptional<z.ZodBoolean>;
         vision: z.ZodOptional<z.ZodBoolean>;

package/dist/catalog.d.ts

@@ -8,7 +8,7 @@ import { z } from 'zod';
 
 /** The AI SDK video model spelling: a catalog id or a provider video model instance. */
 type VideoModel = Parameters<typeof experimental_generateVideo>[0]["model"];
-type AspectRatio = "1:1" | "16:9" | "9:16" | "4:3" | "3:4" | "3:2" | "2:3" | "4:5" | "5:4" | "21:9" | "9:21" | "97:30" | "adaptive";
+type AspectRatio = "1:1" | "16:9" | "9:16" | "4:3" | "3:4" | "3:2" | "2:3" | "4:5" | "5:4" | "21:9" | "9:21" | "97:30" | "1:4" | "4:1" | "1:8" | "8:1" | "adaptive";
 type ImageSize = "512" | "1K" | "2K" | "4K";
 type VideoResolution = "360p" | "480p" | "720p" | "768p" | "1080p" | "2k" | "4k";
 /** "xhigh" and "max" exist only on gpt-image-2.5; gpt-image-2 rejects them at the provider. */
@@ -393,6 +393,8 @@ interface CatalogCapabilities {
     videoEdit?: boolean;
     /** Video rows: first/last frames and reference images are mutually exclusive inputs. */
     framesExcludeReferences?: boolean;
+    /** Video rows: a source video is never sent with first/last frames or reference images. */
+    sourceVideoExcludesFrames?: boolean;
     /** Video rows: a last frame is accepted only together with a first frame. */
     lastFrameRequiresFirstFrame?: boolean;
     /** Text rows. `reasoning`: the model streams reasoning the UI should surface. */
@@ -689,6 +691,10 @@ declare const CatalogModelSchemaBase: z.ZodObject<{
             "21:9": "21:9";
             "9:21": "9:21";
             "97:30": "97:30";
+            "1:4": "1:4";
+            "4:1": "4:1";
+            "1:8": "1:8";
+            "8:1": "8:1";
             adaptive: "adaptive";
         }>>>;
         sizes: z.ZodOptional<z.ZodArray<z.ZodEnum<{
@@ -824,6 +830,10 @@ declare const CatalogModelSchemaBase: z.ZodObject<{
             "21:9": "21:9";
             "9:21": "9:21";
             "97:30": "97:30";
+            "1:4": "1:4";
+            "4:1": "4:1";
+            "1:8": "1:8";
+            "8:1": "8:1";
             adaptive: "adaptive";
         }>>;
         size: z.ZodOptional<z.ZodEnum<{
@@ -986,6 +996,7 @@ declare const CatalogModelSchemaBase: z.ZodObject<{
         acceptsSourceVideo: z.ZodOptional<z.ZodBoolean>;
         videoEdit: z.ZodOptional<z.ZodBoolean>;
         framesExcludeReferences: z.ZodOptional<z.ZodBoolean>;
+        sourceVideoExcludesFrames: z.ZodOptional<z.ZodBoolean>;
         lastFrameRequiresFirstFrame: z.ZodOptional<z.ZodBoolean>;
         reasoning: z.ZodOptional<z.ZodBoolean>;
         vision: z.ZodOptional<z.ZodBoolean>;

package/dist/catalog.js

@@ -158,13 +158,21 @@ var GEMINI_RATIOS = [
   "5:4",
   "adaptive"
 ];
+var NANO_BANANA_2_1_RATIOS = [
+  ...GEMINI_RATIOS.filter((r) => r !== "adaptive"),
+  "1:4",
+  "4:1",
+  "1:8",
+  "8:1",
+  "adaptive"
+];
 var GEMINI_PROMPT_MAX_CHARS = 5e4;
 var GPT_IMAGE_PROMPT = { referenceSyntax: "numbered", maxChars: 32e3 };
-function geminiImage(id, label, aliases, sizes, display, exposure, references, promptMaxChars) {
+function geminiImage(id, label, aliases, sizes, display, exposure, references, promptMaxChars, aspectRatios = GEMINI_RATIOS) {
   const prompt = { referenceSyntax: "caption" };
   if (promptMaxChars !== void 0) prompt.maxChars = promptMaxChars;
   const constraints = {
-    aspectRatios: [...GEMINI_RATIOS],
+    aspectRatios: [...aspectRatios],
     sizes,
     supports: ["safetySettings", "temperature"],
     prompt
@@ -188,6 +196,21 @@ function geminiImage(id, label, aliases, sizes, display, exposure, references, p
   return model;
 }
 var IMAGE_MODELS = [
+  geminiImage(
+    "gemini-nano-banana-2.1",
+    "Nano Banana 2.1",
+    ["nb-2.1"],
+    ["1K", "2K", "4K"],
+    {
+      provider: "google",
+      description: "Latest high-efficiency Nano Banana. Better realism, prompt adherence, text rendering and character consistency at Flash speed. Great for campaigns, infographics, and iterative edits.",
+      logo: "https://d31vvfpk2ppdyq.cloudfront.net/assets/image/192485d2-a76a-4657-9c06-930b2c70badd_compressed.png"
+    },
+    ["cowork", "space_builder"],
+    { images: 14 },
+    GEMINI_PROMPT_MAX_CHARS,
+    NANO_BANANA_2_1_RATIOS
+  ),
   geminiImage(
     "gemini-3-pro-image-preview",
     "Nano Banana Pro",
@@ -2216,7 +2239,8 @@ var VIDEO_INPUT_SPEC_KEYS = [
   "generateAudio",
   "cfgScale",
   "promptExpansion",
-  "referenceImages"
+  "referenceImages",
+  "sourceVideo"
 ];
 var IMAGE_INPUT_SPEC_KEYS = [
   "prompt",
@@ -2235,6 +2259,12 @@ var AUDIO_INPUT_SPEC_KEYS = [
   "sourceVideo",
   "instrumental"
 ];
+var ARRAY_INPUT_SPEC_KEYS = [
+  "firstFrame",
+  "referenceImages",
+  "lastFrame",
+  "references"
+];
 var INPUT_SPEC_SUPPORTS_KEYS = [
   "lastFrame",
   "seed",
@@ -2248,6 +2278,7 @@ var INPUT_SPEC_SUPPORTS_KEYS = [
 var VIDEO_KEY_SET = new Set(VIDEO_INPUT_SPEC_KEYS);
 var IMAGE_KEY_SET = new Set(IMAGE_INPUT_SPEC_KEYS);
 var AUDIO_KEY_SET = new Set(AUDIO_INPUT_SPEC_KEYS);
+var ARRAY_KEY_SET = new Set(ARRAY_INPUT_SPEC_KEYS);
 var ALL_KEY_SET = /* @__PURE__ */ new Set([
   ...VIDEO_INPUT_SPEC_KEYS,
   ...IMAGE_INPUT_SPEC_KEYS,
@@ -2292,13 +2323,30 @@ var ValuesSchema = z2.record(z2.string(), z2.record(z2.string(), ProviderScalarS
   }
 });
 var StaticSchema = z2.record(z2.string(), ProviderScalarSchema);
+var ArraysSchema = z2.record(z2.string().min(1), z2.array(z2.string()).min(1)).superRefine((arrays, ctx) => {
+  for (const [name, keys] of Object.entries(arrays)) {
+    for (const key of keys) {
+      if (!ARRAY_KEY_SET.has(key)) {
+        ctx.addIssue({
+          code: "custom",
+          path: [name],
+          message: `inputSpec.arrays "${name}" has unknown image key "${key}"`
+        });
+      }
+    }
+  }
+});
 var InputSpecSchema = z2.strictObject({
   fields: FieldsSchema,
   values: ValuesSchema.optional(),
   static: StaticSchema.optional(),
+  arrays: ArraysSchema.optional(),
   editEndpoint: z2.string().min(1).optional()
 }).superRefine((spec, ctx) => {
-  const mappedNames = new Set(Object.values(spec.fields));
+  const mappedNames = /* @__PURE__ */ new Set([
+    ...Object.values(spec.fields),
+    ...Object.keys(spec.arrays ?? {})
+  ]);
   for (const key of Object.keys(spec.static ?? {})) {
     if (mappedNames.has(key)) {
       ctx.addIssue({
@@ -2387,14 +2435,54 @@ function validateInputSpec(raw, row) {
       });
     }
   }
+  const arrayOwner = /* @__PURE__ */ new Map();
+  for (const [name, keys] of Object.entries(spec.arrays ?? {})) {
+    const fieldKey = seenNames.get(name);
+    if (fieldKey !== void 0) {
+      issues.push({
+        path: ["arrays", name],
+        message: `inputSpec.arrays "${name}" is also the provider field of inputSpec.fields "${fieldKey}"`
+      });
+    }
+    for (const key of keys) {
+      if (allowed !== void 0 && !allowed.has(key)) {
+        issues.push({
+          path: ["arrays", name],
+          message: `inputSpec.arrays "${key}" is not supported for ${row.type} rows`
+        });
+      }
+      if (key in spec.fields) {
+        issues.push({
+          path: ["arrays", name],
+          message: `inputSpec.arrays "${key}" is already mapped in inputSpec.fields`
+        });
+      }
+      const owner = arrayOwner.get(key);
+      if (owner === void 0) arrayOwner.set(key, name);
+      else {
+        issues.push({
+          path: ["arrays", name],
+          message: owner === name ? `inputSpec.arrays "${name}" lists "${key}" twice` : `inputSpec.arrays "${owner}" and "${name}" both take "${key}"`
+        });
+      }
+    }
+  }
   const supports = row.constraints.supports ?? [];
   for (const key of INPUT_SPEC_SUPPORTS_KEYS) {
-    if (key in spec.fields && !supports.includes(key)) {
+    if (supports.includes(key)) continue;
+    if (key in spec.fields) {
       issues.push({
         path: ["fields", key],
         message: `inputSpec.fields "${key}" needs "${key}" in constraints.supports`
       });
     }
+    const arrayName = arrayOwner.get(key);
+    if (arrayName !== void 0) {
+      issues.push({
+        path: ["arrays", arrayName],
+        message: `inputSpec.arrays "${key}" needs "${key}" in constraints.supports`
+      });
+    }
   }
   if (issues.length > 0) return { ok: false, issues };
   return { ok: true, spec };
@@ -2443,6 +2531,10 @@ var AspectRatioSchema = z3.enum([
   "21:9",
   "9:21",
   "97:30",
+  "1:4",
+  "4:1",
+  "1:8",
+  "8:1",
   "adaptive"
 ]);
 var ImageSizeSchema = z3.enum(["512", "1K", "2K", "4K"]);
@@ -2612,6 +2704,7 @@ var CatalogCapabilitiesSchema = z3.looseObject({
   acceptsSourceVideo: z3.boolean().optional(),
   videoEdit: z3.boolean().optional(),
   framesExcludeReferences: z3.boolean().optional(),
+  sourceVideoExcludesFrames: z3.boolean().optional(),
   lastFrameRequiresFirstFrame: z3.boolean().optional(),
   reasoning: z3.boolean().optional(),
   vision: z3.boolean().optional(),
@@ -3309,6 +3402,26 @@ function resolveVideoOptions(model, request, onUnsupported = "error") {
     });
     delete merged.sourceVideo;
   }
+  const imageInputs = ["firstFrame", "lastFrame", "referenceImages"].filter(
+    (field) => field === "referenceImages" ? compact(merged.referenceImages).length > 0 : merged[field] !== void 0
+  );
+  if (merged.sourceVideo !== void 0 && imageInputs.length > 0 && model.capabilities?.sourceVideoExcludesFrames) {
+    if (onUnsupported === "error") {
+      throw new ModelKitError({
+        code: "unsupported_option",
+        message: `${model.id} takes a source video or images, not both`,
+        model: model.id
+      });
+    }
+    for (const field of imageInputs) {
+      warnings.push({
+        code: "option_ignored",
+        field,
+        message: `${model.id} takes a source video or images, not both; ${field} dropped`
+      });
+      delete merged[field];
+    }
+  }
   if (merged.lastFrame !== void 0 && merged.firstFrame === void 0 && model.capabilities?.lastFrameRequiresFirstFrame) {
     if (onUnsupported === "error") {
       throw new ModelKitError({