@shop-os/modelkit 0.3.0 → 0.3.2

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. */
@@ -689,6 +689,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 +828,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<{

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. */
@@ -689,6 +689,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 +828,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<{

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",
@@ -2253,6 +2276,10 @@ var ALL_KEY_SET = /* @__PURE__ */ new Set([
   ...IMAGE_INPUT_SPEC_KEYS,
   ...AUDIO_INPUT_SPEC_KEYS
 ]);
+var TYPES_BY_ADAPTER = /* @__PURE__ */ new Map([
+  ["replicate", ["video", "image", "audio"]],
+  ["fal", ["image"]]
+]);
 var ALLOWED_KEYS_BY_TYPE = /* @__PURE__ */ new Map([
   ["video", VIDEO_KEY_SET],
   ["image", IMAGE_KEY_SET],
@@ -2291,7 +2318,8 @@ var StaticSchema = z2.record(z2.string(), ProviderScalarSchema);
 var InputSpecSchema = z2.strictObject({
   fields: FieldsSchema,
   values: ValuesSchema.optional(),
-  static: StaticSchema.optional()
+  static: StaticSchema.optional(),
+  editEndpoint: z2.string().min(1).optional()
 }).superRefine((spec, ctx) => {
   const mappedNames = new Set(Object.values(spec.fields));
   for (const key of Object.keys(spec.static ?? {})) {
@@ -2316,18 +2344,21 @@ function validateInputSpec(raw, row) {
     };
   }
   const spec = parsed.data;
-  if (row.adapter !== "replicate") {
+  const supportedTypes = TYPES_BY_ADAPTER.get(row.adapter);
+  if (supportedTypes === void 0) {
     return {
       ok: false,
-      issues: [{ path: [], message: "inputSpec is only supported on replicate rows" }]
+      issues: [
+        { path: [], message: "inputSpec is only supported on replicate and fal rows" }
+      ]
     };
   }
   const issues = [];
   const allowed = ALLOWED_KEYS_BY_TYPE.get(row.type);
-  if (allowed === void 0) {
+  if (allowed === void 0 || !supportedTypes.includes(row.type)) {
     issues.push({
       path: [],
-      message: `inputSpec is not supported for "${row.type}" rows`
+      message: row.adapter === "fal" ? `inputSpec on fal rows is only supported for image rows, not "${row.type}"` : `inputSpec is not supported for "${row.type}" rows`
     });
   } else {
     for (const key of Object.keys(spec.fields)) {
@@ -2347,6 +2378,19 @@ function validateInputSpec(raw, row) {
       }
     }
   }
+  if (spec.editEndpoint !== void 0) {
+    if (row.adapter !== "fal") {
+      issues.push({
+        path: ["editEndpoint"],
+        message: "inputSpec.editEndpoint is only supported on fal rows"
+      });
+    } else if (!("references" in spec.fields)) {
+      issues.push({
+        path: ["editEndpoint"],
+        message: 'inputSpec.editEndpoint needs a "references" entry in inputSpec.fields'
+      });
+    }
+  }
   const seenNames = /* @__PURE__ */ new Map();
   for (const [key, name] of Object.entries(spec.fields)) {
     const first = seenNames.get(name);
@@ -2422,6 +2466,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"]);