@shop-os/modelkit 0.3.3 → 0.3.4

package/dist/index.d.cts

@@ -152,6 +152,8 @@ interface VideoToolRequest {
     source: VideoToolSource;
     /** Upscale rows. */
     upscaleFactor?: number;
+    /** Upscale rows whose `constraints.tool.prompt` is set; ignored by every other row. */
+    prompt?: string;
     /** Interpolate rows. */
     targetFps?: number;
     /** HDR rows; the row's first `outputFormats` entry when unset. */
@@ -274,6 +276,8 @@ interface VideoToolConstraints {
     };
     /** Upscale: the output's long edge may not pass this, in pixels. */
     maxOutputLongEdge?: number;
+    /** Upscale: the row takes an optional prompt that guides added detail. */
+    prompt?: boolean;
     /** Interpolate: allowed target frame rate range. */
     fps?: {
         min: number;
@@ -821,6 +825,7 @@ declare const CatalogModelSchemaBase: z.ZodObject<{
                 max: z.ZodNumber;
             }, z.core.$loose>>;
             maxOutputLongEdge: z.ZodOptional<z.ZodNumber>;
+            prompt: z.ZodOptional<z.ZodBoolean>;
             fps: z.ZodOptional<z.ZodObject<{
                 min: z.ZodNumber;
                 max: z.ZodNumber;
@@ -1414,7 +1419,11 @@ declare function planVideoTool(row: CatalogModel, req: Omit<VideoToolRequest, "m
 
 /**
  * Topaz Labs video tools on fal. `providerOptions.fal.model` names the Topaz model the
- * endpoint runs; the adapter sends it as `model`. Exposed on cowork only (Creative Studio).
+ * endpoint runs; the adapter sends it as `model`. Precision and Starlight are also listed in
+ * Space Builder's video upscale node.
+ *
+ * FLUX video upscale on fal: one endpoint, two rows, because fal prices its `creativity`
+ * modes differently and pricing is keyed by model. `providerOptions.fal.input` fixes the mode.
  */
 declare const VIDEO_TOOL_MODELS: CatalogModel[];
 

package/dist/index.d.ts

@@ -152,6 +152,8 @@ interface VideoToolRequest {
     source: VideoToolSource;
     /** Upscale rows. */
     upscaleFactor?: number;
+    /** Upscale rows whose `constraints.tool.prompt` is set; ignored by every other row. */
+    prompt?: string;
     /** Interpolate rows. */
     targetFps?: number;
     /** HDR rows; the row's first `outputFormats` entry when unset. */
@@ -274,6 +276,8 @@ interface VideoToolConstraints {
     };
     /** Upscale: the output's long edge may not pass this, in pixels. */
     maxOutputLongEdge?: number;
+    /** Upscale: the row takes an optional prompt that guides added detail. */
+    prompt?: boolean;
     /** Interpolate: allowed target frame rate range. */
     fps?: {
         min: number;
@@ -821,6 +825,7 @@ declare const CatalogModelSchemaBase: z.ZodObject<{
                 max: z.ZodNumber;
             }, z.core.$loose>>;
             maxOutputLongEdge: z.ZodOptional<z.ZodNumber>;
+            prompt: z.ZodOptional<z.ZodBoolean>;
             fps: z.ZodOptional<z.ZodObject<{
                 min: z.ZodNumber;
                 max: z.ZodNumber;
@@ -1414,7 +1419,11 @@ declare function planVideoTool(row: CatalogModel, req: Omit<VideoToolRequest, "m
 
 /**
  * Topaz Labs video tools on fal. `providerOptions.fal.model` names the Topaz model the
- * endpoint runs; the adapter sends it as `model`. Exposed on cowork only (Creative Studio).
+ * endpoint runs; the adapter sends it as `model`. Precision and Starlight are also listed in
+ * Space Builder's video upscale node.
+ *
+ * FLUX video upscale on fal: one endpoint, two rows, because fal prices its `creativity`
+ * modes differently and pricing is keyed by model. `providerOptions.fal.input` fixes the mode.
  */
 declare const VIDEO_TOOL_MODELS: CatalogModel[];
 

package/dist/index.js

@@ -2013,8 +2013,16 @@ var VIDEO_MODELS = [
 
 // src/catalog/video-tools.ts
 var TOPAZ_LOGO = "https://cdn.jsdelivr.net/npm/@lobehub/icons-static-png@1.97.0/light/topazlabs.png";
+var FLUX_LOGO = "https://cdn.jsdelivr.net/npm/@lobehub/icons-static-png@1.97.0/light/flux.png";
 var TOPAZ_SOURCE_SECONDS = { max: 300 };
-var TOPAZ_USAGE = { unit: "seconds" };
+var SECONDS_USAGE = { unit: "seconds" };
+var FLUX_SOURCE_SECONDS = { max: 20 };
+var FLUX_UPSCALE_TOOL = {
+  kind: "upscale",
+  upscaleFactor: { min: 1.5, max: 3 },
+  maxOutputLongEdge: 3840,
+  prompt: true
+};
 var VIDEO_TOOL_MODELS = [
   {
     id: "topaz-precision",
@@ -2024,7 +2032,7 @@ var VIDEO_TOOL_MODELS = [
     adapter: "fal",
     providerModelId: "topaz/upscale/video/precision",
     lifecycle: "active",
-    exposure: ["cowork"],
+    exposure: ["cowork", "space_builder"],
     constraints: {
       supports: [],
       sourceVideoSeconds: TOPAZ_SOURCE_SECONDS,
@@ -2032,7 +2040,7 @@ var VIDEO_TOOL_MODELS = [
     },
     defaults: {},
     providerOptions: { fal: { model: "Proteus" } },
-    usage: TOPAZ_USAGE,
+    usage: SECONDS_USAGE,
     pricingKind: "video_per_second",
     display: {
       provider: "topaz",
@@ -2049,7 +2057,7 @@ var VIDEO_TOOL_MODELS = [
     adapter: "fal",
     providerModelId: "topaz/upscale/video/generative",
     lifecycle: "active",
-    exposure: ["cowork"],
+    exposure: ["cowork", "space_builder"],
     constraints: {
       supports: [],
       sourceVideoSeconds: TOPAZ_SOURCE_SECONDS,
@@ -2057,7 +2065,7 @@ var VIDEO_TOOL_MODELS = [
     },
     defaults: {},
     providerOptions: { fal: { model: "Starlight Precise 2.6" } },
-    usage: TOPAZ_USAGE,
+    usage: SECONDS_USAGE,
     pricingKind: "video_per_second",
     display: {
       provider: "topaz",
@@ -2082,7 +2090,7 @@ var VIDEO_TOOL_MODELS = [
     },
     defaults: {},
     providerOptions: { fal: { model: "Apollo" } },
-    usage: TOPAZ_USAGE,
+    usage: SECONDS_USAGE,
     pricingKind: "video_per_second",
     display: {
       provider: "topaz",
@@ -2105,13 +2113,63 @@ var VIDEO_TOOL_MODELS = [
       tool: { kind: "hdr", outputFormats: ["h265", "prores"] }
     },
     defaults: {},
-    usage: TOPAZ_USAGE,
+    usage: SECONDS_USAGE,
     pricingKind: "video_per_second",
     display: {
       provider: "topaz",
       logo: TOPAZ_LOGO,
       description: "Topaz Hyperion 2.5 SDR to HDR: a 10-bit HDR master in H.265 (HDR10) or ProRes 422 HQ, at the source's size and frame rate."
     }
+  },
+  {
+    id: "flux-video-upscale-precise",
+    aliases: [],
+    label: "FLUX Upscale Precise",
+    type: "video-tool",
+    adapter: "fal",
+    providerModelId: "blackforestlabs/flux-video-upscale",
+    lifecycle: "active",
+    exposure: ["space_builder"],
+    constraints: {
+      supports: [],
+      sourceVideoSeconds: FLUX_SOURCE_SECONDS,
+      tool: FLUX_UPSCALE_TOOL
+    },
+    defaults: {},
+    providerOptions: { fal: { input: { creativity: 0 } } },
+    usage: SECONDS_USAGE,
+    pricingKind: "video_per_second",
+    display: {
+      provider: "black-forest-labs",
+      logo: FLUX_LOGO,
+      description: "FLUX 3 super-resolution, source-faithful: 1.5x to 3x, up to 4K, for clips up to 20 seconds.",
+      sortOrder: 30
+    }
+  },
+  {
+    id: "flux-video-upscale-creative",
+    aliases: [],
+    label: "FLUX Upscale Creative",
+    type: "video-tool",
+    adapter: "fal",
+    providerModelId: "blackforestlabs/flux-video-upscale",
+    lifecycle: "active",
+    exposure: ["space_builder"],
+    constraints: {
+      supports: [],
+      sourceVideoSeconds: FLUX_SOURCE_SECONDS,
+      tool: FLUX_UPSCALE_TOOL
+    },
+    defaults: {},
+    providerOptions: { fal: { input: { creativity: 1 } } },
+    usage: SECONDS_USAGE,
+    pricingKind: "video_per_second",
+    display: {
+      provider: "black-forest-labs",
+      logo: FLUX_LOGO,
+      description: "FLUX 3 super-resolution that adds fine detail, optionally guided by a prompt: 1.5x to 3x, up to 4K, for clips up to 20 seconds. Costs more than Precise.",
+      sortOrder: 40
+    }
   }
 ];
 
@@ -2612,6 +2670,7 @@ var VideoToolConstraintsSchema = z3.looseObject({
   kind: z3.enum(["upscale", "interpolate", "hdr"]),
   upscaleFactor: RangeSchema.optional(),
   maxOutputLongEdge: z3.number().positive().optional(),
+  prompt: z3.boolean().optional(),
   fps: RangeSchema.optional(),
   outputFormats: z3.array(z3.enum(["h265", "prores"])).min(1).optional()
 });
@@ -3907,6 +3966,7 @@ var ToolSchema = z10.object({
   kind: z10.enum(["upscale", "interpolate", "hdr"]),
   upscaleFactor: RangeSchema2.optional(),
   maxOutputLongEdge: z10.number().positive().optional(),
+  prompt: z10.boolean().optional(),
   fps: RangeSchema2.optional(),
   outputFormats: z10.array(z10.enum(["h265", "prores"])).min(1).optional()
 });
@@ -3914,11 +3974,12 @@ function videoToolConstraints(row) {
   if (row.type !== "video-tool") return void 0;
   const parsed = ToolSchema.safeParse(row.constraints.tool);
   if (!parsed.success) return void 0;
-  const { kind, upscaleFactor, maxOutputLongEdge, fps, outputFormats } = parsed.data;
+  const { kind, upscaleFactor, maxOutputLongEdge, prompt, fps, outputFormats } = parsed.data;
   return {
     kind,
     ...upscaleFactor !== void 0 && { upscaleFactor },
     ...maxOutputLongEdge !== void 0 && { maxOutputLongEdge },
+    ...prompt !== void 0 && { prompt },
     ...fps !== void 0 && { fps },
     ...outputFormats !== void 0 && { outputFormats }
   };
@@ -4074,14 +4135,21 @@ async function readJson(ctx, res) {
 function providerModelName(ctx) {
   return z11.string().safeParse(ctx.model.providerOptions?.["fal"]?.["model"]).data;
 }
+function staticInput(ctx) {
+  return JsonObjectSchema.safeParse(ctx.model.providerOptions?.["fal"]?.["input"]).data ?? {};
+}
 function toolInput(ctx, req, output) {
   const tool = videoToolConstraints(ctx.model);
   const model = providerModelName(ctx);
-  const input = { video_url: req.source.url };
+  const input = { ...staticInput(ctx), video_url: req.source.url };
   if (tool?.kind === "upscale") {
-    if (model !== void 0) input["model"] = model;
     input["upscale_factor"] = req.upscaleFactor ?? 1;
-    input["H264_output"] = true;
+    if (model !== void 0) {
+      input["model"] = model;
+      input["H264_output"] = true;
+    }
+    const prompt = req.prompt?.trim();
+    if (tool.prompt === true && prompt) input["prompt"] = prompt;
   } else if (tool?.kind === "interpolate") {
     if (model !== void 0) input["model"] = model;
     input["target_fps"] = req.targetFps ?? 60;