@pixeldojo/mcp 0.5.0 → 0.6.0

package/dist/index.cjs

@@ -260,7 +260,7 @@ var init_generate = __esm({
     InputSchema = zod.z.object({
       prompt: zod.z.string().min(1).max(8e3).describe("Prompt describing what to generate."),
       model: zod.z.string().optional().describe(
-        'Override the model. Any apiId from /api/v1/models (e.g. "flux", "google-nano-banana", "veo-3.1", "kling-video"). Omit to use the default.'
+        'Override the model. Any apiId from /api/v1/models (e.g. "flux", "google-nano-banana", "veo-3.1-fast", "kling-video-v3-standard-text"). Omit to use the default.'
       ),
       preset: zod.z.string().optional().describe(
         'Preset slug to seed the params (e.g. "cinematic-portrait", "product-floating-clean", "magazine-cover-text"). Browse the catalog at https://pixeldojo.ai/presets. Explicit fields override preset values.'
@@ -279,7 +279,7 @@ var init_generate = __esm({
         prompt: { type: "string", description: "Prompt describing what to generate." },
         model: {
           type: "string",
-          description: 'Override the model. Any apiId from /api/v1/models (e.g. "flux", "google-nano-banana", "veo-3.1", "kling-video"). Omit to use the default.'
+          description: 'Override the model. Any apiId from /api/v1/models (e.g. "flux", "google-nano-banana", "veo-3.1-fast", "kling-video-v3-standard-text"). Omit to use the default.'
         },
         preset: {
           type: "string",
@@ -473,17 +473,22 @@ async function run3(input, ctx) {
     };
   }
   const apiId = requested;
-  if ((apiId === "flux-edit" || apiId === "gpt-image-2-edit") && !input.reference_image_url) {
+  if (REQUIRES_REFERENCE.has(apiId) && !input.reference_image_url) {
     return {
       kind: "text",
-      text: `pixeldojo:character with model="${apiId}" requires reference_image_url (the source image of the character to edit). For generation from a prompt alone, use model="ideogram-character".`
+      text: `pixeldojo:character with model="${apiId}" requires reference_image_url (the source image of the character to edit). For generation from a prompt alone, use model="${DEFAULT_MODEL3}".`
     };
   }
   const body = { prompt: input.prompt };
-  if (input.aspect_ratio) body.aspect_ratio = input.aspect_ratio;
   if (input.reference_image_url) {
-    body.image_url = input.reference_image_url;
-    body.reference_image_url = input.reference_image_url;
+    if (apiId === "flux-edit") {
+      body.input_image = input.reference_image_url;
+    } else {
+      body.image_urls = [input.reference_image_url];
+    }
+  }
+  if (input.aspect_ratio && apiId === "flux-edit" && FLUX_EDIT_ASPECT_RATIOS.has(input.aspect_ratio)) {
+    body.aspect_ratio = input.aspect_ratio;
   }
   const submitted = await ctx.client.run(apiId, body);
   const result = await pollJobUntilDoneOrHandoff(
@@ -500,20 +505,35 @@ async function run3(input, ctx) {
     (urls) => urls.length === 0 ? `Job ${submitted.jobId} completed but returned no assets.` : `Character shot ready. Asset: ${urls[0]}`
   );
 }
-var ALLOWED, DEFAULT_MODEL3, InputSchema3, InputJsonSchema3, characterSkill;
+var ALLOWED, DEFAULT_MODEL3, REQUIRES_REFERENCE, FLUX_EDIT_ASPECT_RATIOS, InputSchema3, InputJsonSchema3, characterSkill;
 var init_character = __esm({
   "src/skills/character.ts"() {
     init_poll();
     init_types();
-    ALLOWED = ["ideogram-character", "flux-edit", "gpt-image-2-edit"];
-    DEFAULT_MODEL3 = "ideogram-character";
+    ALLOWED = ["seedream-5", "seedream-4", "flux-edit", "gpt-image-2-edit"];
+    DEFAULT_MODEL3 = "seedream-5";
+    REQUIRES_REFERENCE = /* @__PURE__ */ new Set(["flux-edit", "gpt-image-2-edit"]);
+    FLUX_EDIT_ASPECT_RATIOS = /* @__PURE__ */ new Set([
+      "1:1",
+      "16:9",
+      "9:16",
+      "21:9",
+      "9:21",
+      "3:2",
+      "2:3",
+      "4:5",
+      "5:4",
+      "3:4",
+      "4:3",
+      "match_input_image"
+    ]);
     InputSchema3 = zod.z.object({
       prompt: zod.z.string().min(1).max(8e3),
       model: zod.z.string().optional().describe(`One of: ${ALLOWED.join(", ")}. Defaults to ${DEFAULT_MODEL3}.`),
       reference_image_url: zod.z.string().url().optional().describe(
-        "Reference image of the character. Required by flux-edit and gpt-image-2-edit; optional for ideogram-character."
+        "Reference image of the character. Required by flux-edit and gpt-image-2-edit; optional for seedream-5/seedream-4."
       ),
-      aspect_ratio: zod.z.string().optional()
+      aspect_ratio: zod.z.string().optional().describe('Only honored for model="flux-edit".')
     });
     InputJsonSchema3 = {
       type: "object",
@@ -531,7 +551,7 @@ var init_character = __esm({
     };
     characterSkill = {
       name: "pixeldojo:character",
-      description: "Generate or edit consistent characters across shots. Pass a reference_image_url for edit-style models.",
+      description: "Generate or edit consistent characters across shots. Pass a reference_image_url for identity locking (seedream-5 by default).",
       inputZod: InputSchema3,
       inputJsonSchema: InputJsonSchema3,
       allowedModels: [...ALLOWED],
@@ -572,8 +592,13 @@ var init_storyboard = __esm({
   "src/skills/storyboard.ts"() {
     init_poll();
     init_types();
-    ALLOWED2 = ["veo-3.1", "kling-video", "seedance-2"];
-    DEFAULT_MODEL4 = "veo-3.1";
+    ALLOWED2 = [
+      "veo-3.1-fast",
+      "veo-3.1-standard",
+      "kling-video-v3-standard-text",
+      "seedance-2-high"
+    ];
+    DEFAULT_MODEL4 = "veo-3.1-fast";
     InputSchema4 = zod.z.object({
       brief: zod.z.string().min(1).max(8e3).describe("Storyboard brief: scene, shots, mood, action."),
       model: zod.z.string().optional().describe(`One of: ${ALLOWED2.join(", ")}. Defaults to ${DEFAULT_MODEL4}.`),
@@ -619,11 +644,20 @@ async function run5(input, ctx) {
   const apiId = requested;
   const body = {
     prompt: "",
-    image_url: input.image_url
+    image: input.image_url
   };
   if (apiId === "magnific-upscaler") {
-    body.scale_factor = input.scale_factor ?? 2;
+    const allowed = [2, 4, 8, 16];
+    const requestedScale = input.scale_factor ?? 2;
+    const nearest = allowed.reduce(
+      (a, b) => Math.abs(b - requestedScale) < Math.abs(a - requestedScale) ? b : a
+    );
+    body.scale_factor = `${nearest}x`;
     if (input.creativity !== void 0) body.creativity = input.creativity;
+  } else if (apiId === "creative-upscale") {
+    body.scale_factor = Math.min(10, Math.max(1, input.scale_factor ?? 2));
+  } else if (apiId === "face-enhance") {
+    body.scale_factor = Math.min(8, Math.max(1, input.scale_factor ?? 4));
   }
   const submitted = await ctx.client.run(apiId, body);
   const result = await pollJobUntilDoneOrHandoff(
@@ -650,7 +684,9 @@ var init_upscale = __esm({
     InputSchema5 = zod.z.object({
       image_url: zod.z.string().url().describe("URL of the source image to upscale."),
       model: zod.z.string().optional().describe(`One of: ${ALLOWED3.join(", ")}. Defaults to ${DEFAULT_MODEL5}.`),
-      scale_factor: zod.z.number().int().min(1).max(4).optional().describe("Magnific only: 1, 2, or 4. Defaults to 2."),
+      scale_factor: zod.z.number().min(1).max(16).optional().describe(
+        "Upscale multiplier. Magnific snaps to the nearest of 2/4/8/16 (default 2); creative-upscale accepts 1-10 (default 2); face-enhance accepts 1-8 (default 4)."
+      ),
       creativity: zod.z.number().min(0).max(1).optional().describe("Magnific only: 0 (preserve) \u2192 1 (creative). Defaults to 0.")
     });
     InputJsonSchema5 = {
@@ -658,7 +694,7 @@ var init_upscale = __esm({
       properties: {
         image_url: { type: "string", format: "uri" },
         model: { type: "string", enum: ALLOWED3 },
-        scale_factor: { type: "integer", minimum: 1, maximum: 4 },
+        scale_factor: { type: "number", minimum: 1, maximum: 16 },
         creativity: { type: "number", minimum: 0, maximum: 1 }
       },
       required: ["image_url"]
@@ -1287,7 +1323,24 @@ async function runWorkflow(input, ctx) {
         text: `Step ${i + 1} (${step.skill}) has invalid args: ${issue?.path.join(".") || "input"} ${issue?.message || "invalid"}.`
       };
     }
-    const result = await skill.run(parsed.data, ctx);
+    let result = await skill.run(parsed.data, ctx);
+    if (result.kind === "handoff") {
+      const handoffJobId = result.jobId;
+      const polled = await pollJobUntilDoneOrHandoff(
+        handoffJobId,
+        (id) => ctx.client.getJob(id),
+        ctx.buildStatusUrl,
+        {
+          syncBudgetMs: WORKFLOW_STEP_EXTRA_BUDGET_MS,
+          now: ctx.pollOverrides?.now,
+          sleep: ctx.pollOverrides?.sleep
+        }
+      );
+      result = pollResultToSkillResult(
+        polled,
+        (urls) => urls.length === 0 ? `Job ${handoffJobId} completed but returned no assets.` : `Done. Asset: ${urls[0]}`
+      );
+    }
     const url = firstUrl(result);
     if (url) {
       prevUrl = url;
@@ -1297,6 +1350,11 @@ async function runWorkflow(input, ctx) {
         kind: "text",
         text: `Workflow "${workflow.name}" stopped at step ${i + 1} (${step.skill}): ${result.text}`
       };
+    } else if (result.kind === "handoff") {
+      return {
+        kind: "text",
+        text: `Workflow "${workflow.name}" is still on step ${i + 1} (${step.skill}) after an extended wait. Poll with pixeldojo:status { jobId: "${result.jobId}" }, then re-run the workflow once it completes.`
+      };
     } else {
       outputs.push(`step ${i + 1} (${step.skill}): done (no asset URL)`);
     }
@@ -1307,9 +1365,12 @@ async function runWorkflow(input, ctx) {
 ${outputs.join("\n")}`
   };
 }
-var StepSchema, SaveInputSchema, RunInputSchema, ListInputSchema, stepJson, saveWorkflowSkill, runWorkflowSkill, listWorkflowsSkill;
+var WORKFLOW_STEP_EXTRA_BUDGET_MS, StepSchema, SaveInputSchema, RunInputSchema, ListInputSchema, stepJson, saveWorkflowSkill, runWorkflowSkill, listWorkflowsSkill;
 var init_workflows = __esm({
   "src/skills/workflows.ts"() {
+    init_poll();
+    init_types();
+    WORKFLOW_STEP_EXTRA_BUDGET_MS = 6e4;
     StepSchema = zod.z.object({
       skill: zod.z.string().regex(/^pixeldojo:(generate|edit|character|storyboard|upscale|audio|upload|from_url)$/, {
         message: "Runnable skills only: generate, edit, character, storyboard, upscale, audio, upload, from_url."
@@ -1494,7 +1555,7 @@ function makeRedactedLogger(target = console) {
 }
 
 // src/server.ts
-var PACKAGE_VERSION = "0.1.0";
+var PACKAGE_VERSION = "0.6.0" ;
 function buildServer(options) {
   if (!options.apiKey) {
     throw new Error(