comfyui-mcp 0.1.0

package/dist/services/workflow-composer.js

@@ -0,0 +1,333 @@
+import { ValidationError } from "../utils/errors.js";
+// --- Helpers ---
+export function getNextNodeId(workflow) {
+    const ids = Object.keys(workflow).map(Number).filter((n) => !Number.isNaN(n));
+    return String(ids.length === 0 ? 1 : Math.max(...ids) + 1);
+}
+function conn(nodeId, outputIndex) {
+    return [nodeId, outputIndex];
+}
+// --- Templates ---
+function buildTxt2Img(p) {
+    const ckpt = p.checkpoint ?? "sd_xl_base_1.0.safetensors";
+    const positive = p.positive_prompt ?? "";
+    const negative = p.negative_prompt ?? "";
+    const width = p.width ?? 1024;
+    const height = p.height ?? 1024;
+    const steps = p.steps ?? 20;
+    const cfg = p.cfg ?? 8.0;
+    const seed = p.seed ?? Math.floor(Math.random() * 2 ** 48);
+    const sampler = p.sampler_name ?? "euler";
+    const scheduler = p.scheduler ?? "normal";
+    return {
+        "1": {
+            class_type: "CheckpointLoaderSimple",
+            inputs: { ckpt_name: ckpt },
+        },
+        "2": {
+            class_type: "CLIPTextEncode",
+            inputs: { text: positive, clip: conn("1", 1) },
+            _meta: { title: "Positive Prompt" },
+        },
+        "3": {
+            class_type: "CLIPTextEncode",
+            inputs: { text: negative, clip: conn("1", 1) },
+            _meta: { title: "Negative Prompt" },
+        },
+        "4": {
+            class_type: "EmptyLatentImage",
+            inputs: { width, height, batch_size: 1 },
+        },
+        "5": {
+            class_type: "KSampler",
+            inputs: {
+                model: conn("1", 0),
+                positive: conn("2", 0),
+                negative: conn("3", 0),
+                latent_image: conn("4", 0),
+                seed,
+                steps,
+                cfg,
+                sampler_name: sampler,
+                scheduler,
+                denoise: 1.0,
+            },
+        },
+        "6": {
+            class_type: "VAEDecode",
+            inputs: { samples: conn("5", 0), vae: conn("1", 2) },
+        },
+        "7": {
+            class_type: "SaveImage",
+            inputs: { images: conn("6", 0), filename_prefix: "ComfyUI" },
+        },
+    };
+}
+function buildImg2Img(p) {
+    const ckpt = p.checkpoint ?? "sd_xl_base_1.0.safetensors";
+    const positive = p.positive_prompt ?? "";
+    const negative = p.negative_prompt ?? "";
+    const steps = p.steps ?? 20;
+    const cfg = p.cfg ?? 8.0;
+    const seed = p.seed ?? Math.floor(Math.random() * 2 ** 48);
+    const sampler = p.sampler_name ?? "euler";
+    const scheduler = p.scheduler ?? "normal";
+    const denoise = p.denoise ?? 0.75;
+    const imagePath = p.image_path ?? "input.png";
+    return {
+        "1": {
+            class_type: "CheckpointLoaderSimple",
+            inputs: { ckpt_name: ckpt },
+        },
+        "2": {
+            class_type: "LoadImage",
+            inputs: { image: imagePath },
+        },
+        "3": {
+            class_type: "VAEEncode",
+            inputs: { pixels: conn("2", 0), vae: conn("1", 2) },
+        },
+        "4": {
+            class_type: "CLIPTextEncode",
+            inputs: { text: positive, clip: conn("1", 1) },
+            _meta: { title: "Positive Prompt" },
+        },
+        "5": {
+            class_type: "CLIPTextEncode",
+            inputs: { text: negative, clip: conn("1", 1) },
+            _meta: { title: "Negative Prompt" },
+        },
+        "6": {
+            class_type: "KSampler",
+            inputs: {
+                model: conn("1", 0),
+                positive: conn("4", 0),
+                negative: conn("5", 0),
+                latent_image: conn("3", 0),
+                seed,
+                steps,
+                cfg,
+                sampler_name: sampler,
+                scheduler,
+                denoise,
+            },
+        },
+        "7": {
+            class_type: "VAEDecode",
+            inputs: { samples: conn("6", 0), vae: conn("1", 2) },
+        },
+        "8": {
+            class_type: "SaveImage",
+            inputs: { images: conn("7", 0), filename_prefix: "ComfyUI" },
+        },
+    };
+}
+function buildUpscale(p) {
+    const model = p.upscale_model ?? "RealESRGAN_x4plus.pth";
+    const imagePath = p.image_path ?? "input.png";
+    return {
+        "1": {
+            class_type: "LoadImage",
+            inputs: { image: imagePath },
+        },
+        "2": {
+            class_type: "UpscaleModelLoader",
+            inputs: { model_name: model },
+        },
+        "3": {
+            class_type: "ImageUpscaleWithModel",
+            inputs: { upscale_model: conn("2", 0), image: conn("1", 0) },
+        },
+        "4": {
+            class_type: "SaveImage",
+            inputs: { images: conn("3", 0), filename_prefix: "ComfyUI_upscale" },
+        },
+    };
+}
+function buildInpaint(p) {
+    const ckpt = p.checkpoint ?? "sd_xl_base_1.0.safetensors";
+    const positive = p.positive_prompt ?? "";
+    const negative = p.negative_prompt ?? "";
+    const steps = p.steps ?? 20;
+    const cfg = p.cfg ?? 8.0;
+    const seed = p.seed ?? Math.floor(Math.random() * 2 ** 48);
+    const sampler = p.sampler_name ?? "euler";
+    const scheduler = p.scheduler ?? "normal";
+    const denoise = p.denoise ?? 0.85;
+    const imagePath = p.image_path ?? "input.png";
+    const maskPath = p.mask_path ?? "mask.png";
+    return {
+        "1": {
+            class_type: "CheckpointLoaderSimple",
+            inputs: { ckpt_name: ckpt },
+        },
+        "2": {
+            class_type: "LoadImage",
+            inputs: { image: imagePath },
+            _meta: { title: "Input Image" },
+        },
+        "3": {
+            class_type: "LoadImage",
+            inputs: { image: maskPath },
+            _meta: { title: "Mask" },
+        },
+        "4": {
+            class_type: "VAEEncode",
+            inputs: { pixels: conn("2", 0), vae: conn("1", 2) },
+        },
+        "5": {
+            class_type: "SetLatentNoiseMask",
+            inputs: { samples: conn("4", 0), mask: conn("3", 1) },
+        },
+        "6": {
+            class_type: "CLIPTextEncode",
+            inputs: { text: positive, clip: conn("1", 1) },
+            _meta: { title: "Positive Prompt" },
+        },
+        "7": {
+            class_type: "CLIPTextEncode",
+            inputs: { text: negative, clip: conn("1", 1) },
+            _meta: { title: "Negative Prompt" },
+        },
+        "8": {
+            class_type: "KSampler",
+            inputs: {
+                model: conn("1", 0),
+                positive: conn("6", 0),
+                negative: conn("7", 0),
+                latent_image: conn("5", 0),
+                seed,
+                steps,
+                cfg,
+                sampler_name: sampler,
+                scheduler,
+                denoise,
+            },
+        },
+        "9": {
+            class_type: "VAEDecode",
+            inputs: { samples: conn("8", 0), vae: conn("1", 2) },
+        },
+        "10": {
+            class_type: "SaveImage",
+            inputs: { images: conn("9", 0), filename_prefix: "ComfyUI_inpaint" },
+        },
+    };
+}
+const TEMPLATES = {
+    txt2img: (p) => buildTxt2Img(p),
+    img2img: (p) => buildImg2Img(p),
+    upscale: (p) => buildUpscale(p),
+    inpaint: (p) => buildInpaint(p),
+};
+export const TEMPLATE_NAMES = Object.keys(TEMPLATES);
+export function createWorkflow(template, params = {}) {
+    const builder = TEMPLATES[template];
+    if (!builder) {
+        throw new ValidationError(`Unknown template "${template}". Available: ${TEMPLATE_NAMES.join(", ")}`);
+    }
+    return builder(params);
+}
+function applySetInput(wf, op) {
+    const node = wf[op.node_id];
+    if (!node)
+        throw new ValidationError(`Node "${op.node_id}" not found`);
+    node.inputs[op.input_name] = op.value;
+}
+function applyAddNode(wf, op) {
+    const id = op.id ?? getNextNodeId(wf);
+    if (wf[id])
+        throw new ValidationError(`Node ID "${id}" already exists`);
+    wf[id] = {
+        class_type: op.class_type,
+        inputs: op.inputs ?? {},
+    };
+    return id;
+}
+function applyRemoveNode(wf, op) {
+    if (!wf[op.node_id])
+        throw new ValidationError(`Node "${op.node_id}" not found`);
+    delete wf[op.node_id];
+    // Clean up any connections pointing to the removed node
+    for (const node of Object.values(wf)) {
+        for (const [key, val] of Object.entries(node.inputs)) {
+            if (Array.isArray(val) &&
+                val.length === 2 &&
+                typeof val[0] === "string" &&
+                val[0] === op.node_id) {
+                delete node.inputs[key];
+            }
+        }
+    }
+}
+function applyConnect(wf, op) {
+    if (!wf[op.source_id])
+        throw new ValidationError(`Source node "${op.source_id}" not found`);
+    if (!wf[op.target_id])
+        throw new ValidationError(`Target node "${op.target_id}" not found`);
+    wf[op.target_id].inputs[op.input_name] = [op.source_id, op.output_index];
+}
+function applyInsertBetween(wf, op) {
+    if (!wf[op.source_id])
+        throw new ValidationError(`Source node "${op.source_id}" not found`);
+    if (!wf[op.target_id])
+        throw new ValidationError(`Target node "${op.target_id}" not found`);
+    const newId = getNextNodeId(wf);
+    const newInputs = { ...(op.new_inputs ?? {}) };
+    // Connect the new node's first input to the original source
+    // Find the first input name that isn't already set -- use a convention-based approach
+    // The new node receives the source output on its primary input
+    // We'll figure out the right input name by looking for common patterns
+    const primaryInputNames = ["model", "clip", "samples", "latent_image", "image", "conditioning", "pixels"];
+    let connected = false;
+    for (const name of primaryInputNames) {
+        if (!(name in newInputs)) {
+            newInputs[name] = [op.source_id, op.output_index];
+            connected = true;
+            break;
+        }
+    }
+    if (!connected) {
+        // Fallback: add as first unused slot
+        newInputs["input"] = [op.source_id, op.output_index];
+    }
+    wf[newId] = {
+        class_type: op.new_class_type,
+        inputs: newInputs,
+    };
+    // Rewire: target's input now points to the new node's output 0
+    wf[op.target_id].inputs[op.input_name] = [newId, 0];
+    return newId;
+}
+export function modifyWorkflow(workflow, operations) {
+    // Deep clone to avoid mutating the original
+    const wf = JSON.parse(JSON.stringify(workflow));
+    const addedIds = [];
+    for (const op of operations) {
+        switch (op.op) {
+            case "set_input":
+                applySetInput(wf, op);
+                break;
+            case "add_node": {
+                const id = applyAddNode(wf, op);
+                addedIds.push(id);
+                break;
+            }
+            case "remove_node":
+                applyRemoveNode(wf, op);
+                break;
+            case "connect":
+                applyConnect(wf, op);
+                break;
+            case "insert_between": {
+                const id = applyInsertBetween(wf, op);
+                addedIds.push(id);
+                break;
+            }
+            default:
+                throw new ValidationError(`Unknown operation: ${op.op}`);
+        }
+    }
+    return { workflow: wf, added_ids: addedIds };
+}
+//# sourceMappingURL=workflow-composer.js.map

package/dist/services/workflow-composer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-composer.js","sourceRoot":"","sources":["../../src/services/workflow-composer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,kBAAkB;AAElB,MAAM,UAAU,aAAa,CAAC,QAAsB;IAClD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,OAAO,MAAM,CAAC,GAAG,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,IAAI,CAAC,MAAc,EAAE,WAAmB;IAC/C,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;AAC/B,CAAC;AAiCD,oBAAoB;AAEpB,SAAS,YAAY,CAAC,CAAgB;IACpC,MAAM,IAAI,GAAG,CAAC,CAAC,UAAU,IAAI,4BAA4B,CAAC;IAC1D,MAAM,QAAQ,GAAG,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC;IACzC,MAAM,QAAQ,GAAG,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC;IAC9B,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC;IAChC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;IAC5B,MAAM,GAAG,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC;IACzB,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3D,MAAM,OAAO,GAAG,CAAC,CAAC,YAAY,IAAI,OAAO,CAAC;IAC1C,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,IAAI,QAAQ,CAAC;IAE1C,OAAO;QACL,GAAG,EAAE;YACH,UAAU,EAAE,wBAAwB;YACpC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE;SAC5B;QACD,GAAG,EAAE;YACH,UAAU,EAAE,gBAAgB;YAC5B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;YAC9C,KAAK,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE;SACpC;QACD,GAAG,EAAE;YACH,UAAU,EAAE,gBAAgB;YAC5B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;YAC9C,KAAK,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE;SACpC;QACD,GAAG,EAAE;YACH,UAAU,EAAE,kBAAkB;YAC9B,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,EAAE;SACzC;QACD,GAAG,EAAE;YACH,UAAU,EAAE,UAAU;YACtB,MAAM,EAAE;gBACN,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACnB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACtB,YAAY,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC1B,IAAI;gBACJ,KAAK;gBACL,GAAG;gBACH,YAAY,EAAE,OAAO;gBACrB,SAAS;gBACT,OAAO,EAAE,GAAG;aACb;SACF;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;SACrD;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,SAAS,EAAE;SAC7D;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,CAAgB;IACpC,MAAM,IAAI,GAAG,CAAC,CAAC,UAAU,IAAI,4BAA4B,CAAC;IAC1D,MAAM,QAAQ,GAAG,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC;IACzC,MAAM,QAAQ,GAAG,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;IAC5B,MAAM,GAAG,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC;IACzB,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3D,MAAM,OAAO,GAAG,CAAC,CAAC,YAAY,IAAI,OAAO,CAAC;IAC1C,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,IAAI,QAAQ,CAAC;IAC1C,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC;IAClC,MAAM,SAAS,GAAG,CAAC,CAAC,UAAU,IAAI,WAAW,CAAC;IAE9C,OAAO;QACL,GAAG,EAAE;YACH,UAAU,EAAE,wBAAwB;YACpC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE;SAC5B;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;SAC7B;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;SACpD;QACD,GAAG,EAAE;YACH,UAAU,EAAE,gBAAgB;YAC5B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;YAC9C,KAAK,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE;SACpC;QACD,GAAG,EAAE;YACH,UAAU,EAAE,gBAAgB;YAC5B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;YAC9C,KAAK,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE;SACpC;QACD,GAAG,EAAE;YACH,UAAU,EAAE,UAAU;YACtB,MAAM,EAAE;gBACN,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACnB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACtB,YAAY,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC1B,IAAI;gBACJ,KAAK;gBACL,GAAG;gBACH,YAAY,EAAE,OAAO;gBACrB,SAAS;gBACT,OAAO;aACR;SACF;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;SACrD;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,SAAS,EAAE;SAC7D;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,CAAgB;IACpC,MAAM,KAAK,GAAG,CAAC,CAAC,aAAa,IAAI,uBAAuB,CAAC;IACzD,MAAM,SAAS,GAAG,CAAC,CAAC,UAAU,IAAI,WAAW,CAAC;IAE9C,OAAO;QACL,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;SAC7B;QACD,GAAG,EAAE;YACH,UAAU,EAAE,oBAAoB;YAChC,MAAM,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE;SAC9B;QACD,GAAG,EAAE;YACH,UAAU,EAAE,uBAAuB;YACnC,MAAM,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;SAC7D;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,iBAAiB,EAAE;SACrE;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,CAAgB;IACpC,MAAM,IAAI,GAAG,CAAC,CAAC,UAAU,IAAI,4BAA4B,CAAC;IAC1D,MAAM,QAAQ,GAAG,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC;IACzC,MAAM,QAAQ,GAAG,CAAC,CAAC,eAAe,IAAI,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;IAC5B,MAAM,GAAG,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC;IACzB,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3D,MAAM,OAAO,GAAG,CAAC,CAAC,YAAY,IAAI,OAAO,CAAC;IAC1C,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,IAAI,QAAQ,CAAC;IAC1C,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC;IAClC,MAAM,SAAS,GAAG,CAAC,CAAC,UAAU,IAAI,WAAW,CAAC;IAC9C,MAAM,QAAQ,GAAG,CAAC,CAAC,SAAS,IAAI,UAAU,CAAC;IAE3C,OAAO;QACL,GAAG,EAAE;YACH,UAAU,EAAE,wBAAwB;YACpC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE;SAC5B;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;YAC5B,KAAK,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE;SAChC;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;YAC3B,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE;SACzB;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;SACpD;QACD,GAAG,EAAE;YACH,UAAU,EAAE,oBAAoB;YAChC,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;SACtD;QACD,GAAG,EAAE;YACH,UAAU,EAAE,gBAAgB;YAC5B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;YAC9C,KAAK,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE;SACpC;QACD,GAAG,EAAE;YACH,UAAU,EAAE,gBAAgB;YAC5B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;YAC9C,KAAK,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE;SACpC;QACD,GAAG,EAAE;YACH,UAAU,EAAE,UAAU;YACtB,MAAM,EAAE;gBACN,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACnB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBACtB,YAAY,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC1B,IAAI;gBACJ,KAAK;gBACL,GAAG;gBACH,YAAY,EAAE,OAAO;gBACrB,SAAS;gBACT,OAAO;aACR;SACF;QACD,GAAG,EAAE;YACH,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE;SACrD;QACD,IAAI,EAAE;YACJ,UAAU,EAAE,WAAW;YACvB,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,eAAe,EAAE,iBAAiB,EAAE;SACrE;KACF,CAAC;AACJ,CAAC;AAED,MAAM,SAAS,GAAsE;IACnF,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAkB,CAAC;IAChD,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAkB,CAAC;IAChD,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAkB,CAAC;IAChD,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAkB,CAAC;CACjD,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AAErD,MAAM,UAAU,cAAc,CAC5B,QAAgB,EAChB,SAAkC,EAAE;IAEpC,MAAM,OAAO,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;IACpC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,eAAe,CACvB,qBAAqB,QAAQ,iBAAiB,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC1E,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC;AACzB,CAAC;AAgDD,SAAS,aAAa,CAAC,EAAgB,EAAE,EAAc;IACrD,MAAM,IAAI,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;IAC5B,IAAI,CAAC,IAAI;QAAE,MAAM,IAAI,eAAe,CAAC,SAAS,EAAE,CAAC,OAAO,aAAa,CAAC,CAAC;IACvE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;AACxC,CAAC;AAED,SAAS,YAAY,CAAC,EAAgB,EAAE,EAAa;IACnD,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;IACtC,IAAI,EAAE,CAAC,EAAE,CAAC;QAAE,MAAM,IAAI,eAAe,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;IACxE,EAAE,CAAC,EAAE,CAAC,GAAG;QACP,UAAU,EAAE,EAAE,CAAC,UAAU;QACzB,MAAM,EAAE,EAAE,CAAC,MAAM,IAAI,EAAE;KACxB,CAAC;IACF,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,eAAe,CAAC,EAAgB,EAAE,EAAgB;IACzD,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC;QAAE,MAAM,IAAI,eAAe,CAAC,SAAS,EAAE,CAAC,OAAO,aAAa,CAAC,CAAC;IACjF,OAAO,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;IAEtB,wDAAwD;IACxD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACrD,IACE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;gBAClB,GAAG,CAAC,MAAM,KAAK,CAAC;gBAChB,OAAO,GAAG,CAAC,CAAC,CAAC,KAAK,QAAQ;gBAC1B,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,OAAO,EACrB,CAAC;gBACD,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,EAAgB,EAAE,EAAa;IACnD,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QAAE,MAAM,IAAI,eAAe,CAAC,gBAAgB,EAAE,CAAC,SAAS,aAAa,CAAC,CAAC;IAC5F,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QAAE,MAAM,IAAI,eAAe,CAAC,gBAAgB,EAAE,CAAC,SAAS,aAAa,CAAC,CAAC;IAC5F,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC;AAC3E,CAAC;AAED,SAAS,kBAAkB,CAAC,EAAgB,EAAE,EAAmB;IAC/D,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QAAE,MAAM,IAAI,eAAe,CAAC,gBAAgB,EAAE,CAAC,SAAS,aAAa,CAAC,CAAC;IAC5F,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QAAE,MAAM,IAAI,eAAe,CAAC,gBAAgB,EAAE,CAAC,SAAS,aAAa,CAAC,CAAC;IAE5F,MAAM,KAAK,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;IAChC,MAAM,SAAS,GAA4B,EAAE,GAAG,CAAC,EAAE,CAAC,UAAU,IAAI,EAAE,CAAC,EAAE,CAAC;IAExE,4DAA4D;IAC5D,sFAAsF;IACtF,+DAA+D;IAC/D,uEAAuE;IACvE,MAAM,iBAAiB,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;IAC1G,IAAI,SAAS,GAAG,KAAK,CAAC;IACtB,KAAK,MAAM,IAAI,IAAI,iBAAiB,EAAE,CAAC;QACrC,IAAI,CAAC,CAAC,IAAI,IAAI,SAAS,CAAC,EAAE,CAAC;YACzB,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC;YAClD,SAAS,GAAG,IAAI,CAAC;YACjB,MAAM;QACR,CAAC;IACH,CAAC;IACD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,qCAAqC;QACrC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC;IACvD,CAAC;IAED,EAAE,CAAC,KAAK,CAAC,GAAG;QACV,UAAU,EAAE,EAAE,CAAC,cAAc;QAC7B,MAAM,EAAE,SAAS;KAClB,CAAC;IAEF,+DAA+D;IAC/D,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAEpD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,QAAsB,EACtB,UAA6B;IAE7B,4CAA4C;IAC5C,MAAM,EAAE,GAAiB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC9D,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,KAAK,MAAM,EAAE,IAAI,UAAU,EAAE,CAAC;QAC5B,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;YACd,KAAK,WAAW;gBACd,aAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACtB,MAAM;YACR,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,MAAM,EAAE,GAAG,YAAY,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAChC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClB,MAAM;YACR,CAAC;YACD,KAAK,aAAa;gBAChB,eAAe,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACxB,MAAM;YACR,KAAK,SAAS;gBACZ,YAAY,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACrB,MAAM;YACR,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,MAAM,EAAE,GAAG,kBAAkB,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACtC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClB,MAAM;YACR,CAAC;YACD;gBACE,MAAM,IAAI,eAAe,CAAC,sBAAuB,EAAqB,CAAC,EAAE,EAAE,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;AAC/C,CAAC"}

package/dist/services/workflow-executor.d.ts

@@ -0,0 +1,15 @@
+import type { WorkflowJSON, SystemStats, QueueStatus, JobResult } from "../comfyui/types.js";
+export interface ExecuteWorkflowOptions {
+    timeout_ms?: number;
+    disable_random_seed?: boolean;
+}
+export declare function executeWorkflow(workflowJson: WorkflowJSON, options?: ExecuteWorkflowOptions): Promise<JobResult>;
+export declare function getJobStatus(promptId: string): Promise<{
+    running: boolean;
+    pending: boolean;
+    done: boolean;
+}>;
+export declare function getQueueStatus(): Promise<QueueStatus>;
+export declare function cancelCurrentJob(): Promise<void>;
+export declare function getSystemInfo(): Promise<SystemStats>;
+//# sourceMappingURL=workflow-executor.d.ts.map

package/dist/services/workflow-executor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-executor.d.ts","sourceRoot":"","sources":["../../src/services/workflow-executor.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EACV,YAAY,EACZ,WAAW,EACX,WAAW,EACX,SAAS,EACV,MAAM,qBAAqB,CAAC;AAK7B,MAAM,WAAW,sBAAsB;IACrC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAED,wBAAsB,eAAe,CACnC,YAAY,EAAE,YAAY,EAC1B,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,SAAS,CAAC,CAsDpB;AAED,wBAAsB,YAAY,CAChC,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,IAAI,EAAE,OAAO,CAAA;CAAE,CAAC,CAGhE;AAED,wBAAsB,cAAc,IAAI,OAAO,CAAC,WAAW,CAAC,CAE3D;AAED,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC,CAGtD;AAED,wBAAsB,aAAa,IAAI,OAAO,CAAC,WAAW,CAAC,CAE1D"}

package/dist/services/workflow-executor.js

@@ -0,0 +1,63 @@
+import { getClient, connectClient, getSystemStats as clientGetSystemStats, getQueue as clientGetQueue, interrupt as clientInterrupt, } from "../comfyui/client.js";
+import { WorkflowExecutionError, ConnectionError } from "../utils/errors.js";
+import { arrayBufferToBase64 } from "../utils/image.js";
+import { logger } from "../utils/logger.js";
+export async function executeWorkflow(workflowJson, options) {
+    const client = getClient();
+    // Ensure WebSocket is connected for enqueue (uses WS internally)
+    try {
+        await connectClient();
+    }
+    catch (err) {
+        if (err instanceof ConnectionError)
+            throw err;
+        throw new ConnectionError(`Failed to connect: ${err instanceof Error ? err.message : err}`);
+    }
+    logger.info("Enqueueing workflow for execution");
+    try {
+        const result = await client.enqueue(workflowJson, {
+            timeout_ms: options?.timeout_ms ?? 10 * 60 * 1000,
+            disable_random_seed: options?.disable_random_seed,
+            progress: (p) => {
+                logger.debug("Execution progress", {
+                    value: p.value,
+                    max: p.max,
+                });
+            },
+        });
+        const images = result.images
+            .filter((img) => img.type === "buff")
+            .map((img) => ({
+            data: arrayBufferToBase64(img.data),
+            mime: img.mime,
+        }));
+        logger.info("Workflow execution completed", {
+            prompt_id: result.prompt_id,
+            image_count: images.length,
+        });
+        return {
+            prompt_id: result.prompt_id,
+            images,
+            node_outputs: {},
+        };
+    }
+    catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        throw new WorkflowExecutionError(`Workflow execution failed: ${message}`);
+    }
+}
+export async function getJobStatus(promptId) {
+    const client = getClient();
+    return client.getPromptStatus(promptId);
+}
+export async function getQueueStatus() {
+    return clientGetQueue();
+}
+export async function cancelCurrentJob() {
+    await clientInterrupt();
+    logger.info("Current job interrupted");
+}
+export async function getSystemInfo() {
+    return clientGetSystemStats();
+}
+//# sourceMappingURL=workflow-executor.js.map

package/dist/services/workflow-executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-executor.js","sourceRoot":"","sources":["../../src/services/workflow-executor.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,EACT,aAAa,EACb,cAAc,IAAI,oBAAoB,EACtC,QAAQ,IAAI,cAAc,EAC1B,SAAS,IAAI,eAAe,GAC7B,MAAM,sBAAsB,CAAC;AAO9B,OAAO,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC7E,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAO5C,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,YAA0B,EAC1B,OAAgC;IAEhC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,iEAAiE;IACjE,IAAI,CAAC;QACH,MAAM,aAAa,EAAE,CAAC;IACxB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,eAAe;YAAE,MAAM,GAAG,CAAC;QAC9C,MAAM,IAAI,eAAe,CACvB,sBAAsB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,CACjE,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;IAEjD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CACjC,YAAuC,EACvC;YACE,UAAU,EAAE,OAAO,EAAE,UAAU,IAAI,EAAE,GAAG,EAAE,GAAG,IAAI;YACjD,mBAAmB,EAAE,OAAO,EAAE,mBAAmB;YACjD,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE;gBACd,MAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE;oBACjC,KAAK,EAAE,CAAC,CAAC,KAAK;oBACd,GAAG,EAAE,CAAC,CAAC,GAAG;iBACX,CAAC,CAAC;YACL,CAAC;SACF,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM;aACzB,MAAM,CACL,CAAC,GAAG,EAA4D,EAAE,CAChE,GAAG,CAAC,IAAI,KAAK,MAAM,CACtB;aACA,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACb,IAAI,EAAE,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC;YACnC,IAAI,EAAE,GAAG,CAAC,IAAI;SACf,CAAC,CAAC,CAAC;QAEN,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE;YAC1C,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,WAAW,EAAE,MAAM,CAAC,MAAM;SAC3B,CAAC,CAAC;QAEH,OAAO;YACL,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,MAAM;YACN,YAAY,EAAE,EAAE;SACjB,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjE,MAAM,IAAI,sBAAsB,CAAC,8BAA8B,OAAO,EAAE,CAAC,CAAC;IAC5E,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,QAAgB;IAEhB,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc;IAClC,OAAO,cAAc,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB;IACpC,MAAM,eAAe,EAAE,CAAC;IACxB,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa;IACjC,OAAO,oBAAoB,EAAE,CAAC;AAChC,CAAC"}

package/dist/tools/index.d.ts

@@ -0,0 +1,3 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+export declare function registerAllTools(server: McpServer): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/tools/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAQzE,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI,CAOxD"}

package/dist/tools/index.js

@@ -0,0 +1,15 @@
+import { registerWorkflowExecuteTools } from "./workflow-execute.js";
+import { registerWorkflowVisualizeTools } from "./workflow-visualize.js";
+import { registerWorkflowComposeTools } from "./workflow-compose.js";
+import { registerRegistrySearchTools } from "./registry-search.js";
+import { registerModelManagementTools } from "./model-management.js";
+import { registerSkillGeneratorTools } from "./skill-generator.js";
+export function registerAllTools(server) {
+    registerWorkflowExecuteTools(server);
+    registerWorkflowVisualizeTools(server);
+    registerWorkflowComposeTools(server);
+    registerRegistrySearchTools(server);
+    registerModelManagementTools(server);
+    registerSkillGeneratorTools(server);
+}
+//# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,4BAA4B,EAAE,MAAM,uBAAuB,CAAC;AACrE,OAAO,EAAE,8BAA8B,EAAE,MAAM,yBAAyB,CAAC;AACzE,OAAO,EAAE,4BAA4B,EAAE,MAAM,uBAAuB,CAAC;AACrE,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAAE,4BAA4B,EAAE,MAAM,uBAAuB,CAAC;AACrE,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AAEnE,MAAM,UAAU,gBAAgB,CAAC,MAAiB;IAChD,4BAA4B,CAAC,MAAM,CAAC,CAAC;IACrC,8BAA8B,CAAC,MAAM,CAAC,CAAC;IACvC,4BAA4B,CAAC,MAAM,CAAC,CAAC;IACrC,2BAA2B,CAAC,MAAM,CAAC,CAAC;IACpC,4BAA4B,CAAC,MAAM,CAAC,CAAC;IACrC,2BAA2B,CAAC,MAAM,CAAC,CAAC;AACtC,CAAC"}

package/dist/tools/model-management.d.ts

@@ -0,0 +1,3 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+export declare function registerModelManagementTools(server: McpServer): void;
+//# sourceMappingURL=model-management.d.ts.map

package/dist/tools/model-management.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-management.d.ts","sourceRoot":"","sources":["../../src/tools/model-management.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAWzE,wBAAgB,4BAA4B,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI,CA2HpE"}

package/dist/tools/model-management.js

@@ -0,0 +1,97 @@
+import { z } from "zod";
+import { searchHuggingFaceModels, listLocalModels, downloadModel, MODEL_SUBDIRS, } from "../services/model-resolver.js";
+import { errorToToolResult } from "../utils/errors.js";
+const modelTypeEnum = z.enum(MODEL_SUBDIRS);
+export function registerModelManagementTools(server) {
+    server.tool("search_models", "Search HuggingFace for models compatible with ComfyUI (checkpoints, LoRAs, VAEs, etc.)", {
+        query: z.string().describe("Search query (e.g. 'SDXL', 'flux', 'controlnet')"),
+        filter: z
+            .string()
+            .optional()
+            .describe("HuggingFace tag filter (e.g. 'diffusers', 'text-to-image')"),
+        limit: z
+            .number()
+            .int()
+            .min(1)
+            .max(50)
+            .optional()
+            .describe("Max results to return (default 10)"),
+    }, async (args) => {
+        try {
+            const results = await searchHuggingFaceModels(args.query, {
+                filter: args.filter,
+                limit: args.limit,
+            });
+            const text = results.length === 0
+                ? `No models found for "${args.query}".`
+                : results
+                    .map((m, i) => `${i + 1}. **${m.modelId}** by ${m.author || "unknown"}\n` +
+                    `   Downloads: ${m.downloads.toLocaleString()} | Likes: ${m.likes}\n` +
+                    `   Tags: ${m.tags.slice(0, 5).join(", ") || "none"}`)
+                    .join("\n\n");
+            return { content: [{ type: "text", text }] };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    server.tool("download_model", "Download a model file to the ComfyUI models directory from a URL (HuggingFace or direct link)", {
+        url: z.string().url().describe("Direct download URL for the model file"),
+        target_subfolder: modelTypeEnum.describe("Target subfolder under ComfyUI models/ (e.g. 'checkpoints', 'loras', 'vae')"),
+        filename: z
+            .string()
+            .optional()
+            .describe("Override filename (auto-detected from URL if omitted)"),
+    }, async (args) => {
+        try {
+            const savedPath = await downloadModel(args.url, args.target_subfolder, args.filename);
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: `Model downloaded successfully to:\n${savedPath}`,
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    server.tool("list_local_models", "List model files installed in the local ComfyUI models directory", {
+        model_type: modelTypeEnum
+            .optional()
+            .describe("Filter by model type (e.g. 'checkpoints', 'loras'). Lists all types if omitted."),
+    }, async (args) => {
+        try {
+            const models = await listLocalModels(args.model_type);
+            if (models.length === 0) {
+                const scope = args.model_type
+                    ? `No ${args.model_type} models found.`
+                    : "No local models found.";
+                return { content: [{ type: "text", text: scope }] };
+            }
+            // Group by type
+            const grouped = new Map();
+            for (const m of models) {
+                const list = grouped.get(m.type) ?? [];
+                list.push(m);
+                grouped.set(m.type, list);
+            }
+            const lines = [];
+            for (const [type, list] of grouped) {
+                lines.push(`## ${type} (${list.length})`);
+                for (const m of list) {
+                    const sizeMB = (m.size / 1024 / 1024).toFixed(1);
+                    lines.push(`- ${m.name} (${sizeMB} MB) — modified ${m.modified}`);
+                }
+                lines.push("");
+            }
+            return { content: [{ type: "text", text: lines.join("\n") }] };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+}
+//# sourceMappingURL=model-management.js.map

package/dist/tools/model-management.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-management.js","sourceRoot":"","sources":["../../src/tools/model-management.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EACL,uBAAuB,EACvB,eAAe,EACf,aAAa,EACb,aAAa,GACd,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAEvD,MAAM,aAAa,GAAG,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;AAE5C,MAAM,UAAU,4BAA4B,CAAC,MAAiB;IAC5D,MAAM,CAAC,IAAI,CACT,eAAe,EACf,wFAAwF,EACxF;QACE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kDAAkD,CAAC;QAC9E,MAAM,EAAE,CAAC;aACN,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,4DAA4D,CAAC;QACzE,KAAK,EAAE,CAAC;aACL,MAAM,EAAE;aACR,GAAG,EAAE;aACL,GAAG,CAAC,CAAC,CAAC;aACN,GAAG,CAAC,EAAE,CAAC;aACP,QAAQ,EAAE;aACV,QAAQ,CAAC,oCAAoC,CAAC;KAClD,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,uBAAuB,CAAC,IAAI,CAAC,KAAK,EAAE;gBACxD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;aAClB,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,KAAK,CAAC;gBAC/B,CAAC,CAAC,wBAAwB,IAAI,CAAC,KAAK,IAAI;gBACxC,CAAC,CAAC,OAAO;qBACJ,GAAG,CACF,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACP,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,OAAO,SAAS,CAAC,CAAC,MAAM,IAAI,SAAS,IAAI;oBAC1D,iBAAiB,CAAC,CAAC,SAAS,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC,KAAK,IAAI;oBACrE,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,MAAM,EAAE,CACxD;qBACA,IAAI,CAAC,MAAM,CAAC,CAAC;YAEpB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAC/C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,gBAAgB,EAChB,+FAA+F,EAC/F;QACE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;QACxE,gBAAgB,EAAE,aAAa,CAAC,QAAQ,CACtC,6EAA6E,CAC9E;QACD,QAAQ,EAAE,CAAC;aACR,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,uDAAuD,CAAC;KACrE,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,aAAa,CACnC,IAAI,CAAC,GAAG,EACR,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,QAAQ,CACd,CAAC;YAEF,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,sCAAsC,SAAS,EAAE;qBACxD;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,mBAAmB,EACnB,kEAAkE,EAClE;QACE,UAAU,EAAE,aAAa;aACtB,QAAQ,EAAE;aACV,QAAQ,CACP,iFAAiF,CAClF;KACJ,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEtD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU;oBAC3B,CAAC,CAAC,MAAM,IAAI,CAAC,UAAU,gBAAgB;oBACvC,CAAC,CAAC,wBAAwB,CAAC;gBAC7B,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;YACtD,CAAC;YAED,gBAAgB;YAChB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAyB,CAAC;YACjD,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;gBACvB,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACvC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACb,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC5B,CAAC;YAED,MAAM,KAAK,GAAa,EAAE,CAAC;YAC3B,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,OAAO,EAAE,CAAC;gBACnC,KAAK,CAAC,IAAI,CAAC,MAAM,IAAI,KAAK,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;gBAC1C,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;oBACrB,MAAM,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;oBACjD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,MAAM,mBAAmB,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACpE,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACjB,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;QACjE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/dist/tools/registry-search.d.ts

@@ -0,0 +1,3 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+export declare function registerRegistrySearchTools(server: McpServer): void;
+//# sourceMappingURL=registry-search.d.ts.map

package/dist/tools/registry-search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry-search.d.ts","sourceRoot":"","sources":["../../src/tools/registry-search.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,wBAAgB,2BAA2B,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI,CAyFnE"}