comfyui-mcp 0.1.0

package/dist/tools/registry-search.js

@@ -0,0 +1,70 @@
+import { z } from "zod";
+import { searchNodes, getNodePackDetails } from "../services/registry-client.js";
+import { errorToToolResult } from "../utils/errors.js";
+export function registerRegistrySearchTools(server) {
+    server.tool("search_custom_nodes", "Search the ComfyUI Registry for custom node packs by keyword", {
+        query: z.string().describe("Search query for custom node packs"),
+        limit: z
+            .number()
+            .int()
+            .min(1)
+            .max(50)
+            .optional()
+            .describe("Max results to return (default 10)"),
+        page: z
+            .number()
+            .int()
+            .min(1)
+            .optional()
+            .describe("Page number for pagination (default 1)"),
+    }, async (args) => {
+        try {
+            const results = await searchNodes(args.query, {
+                limit: args.limit,
+                page: args.page,
+            });
+            const text = results.length === 0
+                ? `No custom nodes found for "${args.query}".`
+                : results
+                    .map((r, i) => `${i + 1}. **${r.name}** (${r.id})\n` +
+                    `   ${r.description ?? "No description"}\n` +
+                    `   Author: ${r.author} | Installs: ${r.total_install ?? "N/A"} | Version: ${r.latest_version ?? "N/A"}`)
+                    .join("\n\n");
+            return { content: [{ type: "text", text }] };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    server.tool("get_node_pack_details", "Get detailed information about a specific ComfyUI custom node pack from the Registry", {
+        id: z.string().describe("Node pack ID (e.g. 'comfyui-impact-pack')"),
+    }, async (args) => {
+        try {
+            const details = await getNodePackDetails(args.id);
+            const lines = [
+                `# ${details.name}`,
+                "",
+                details.description ?? "",
+                "",
+                `- **Author**: ${details.author}`,
+                `- **License**: ${details.license ?? "N/A"}`,
+                `- **Repository**: ${details.repository ?? "N/A"}`,
+                `- **Total Installs**: ${details.total_install ?? "N/A"}`,
+                `- **Latest Version**: ${details.latest_version ?? "N/A"}`,
+                `- **Created**: ${details.created_at ?? "N/A"}`,
+                `- **Updated**: ${details.updated_at ?? "N/A"}`,
+            ];
+            if (details.nodes?.length) {
+                lines.push("", "## Nodes Provided", ...details.nodes.map((n) => `- ${n}`));
+            }
+            if (details.versions?.length) {
+                lines.push("", "## Recent Versions", ...details.versions.slice(0, 5).map((v) => `- **${v.version}**${v.changelog ? `: ${v.changelog}` : ""}`));
+            }
+            return { content: [{ type: "text", text: lines.join("\n") }] };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+}
+//# sourceMappingURL=registry-search.js.map

package/dist/tools/registry-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry-search.js","sourceRoot":"","sources":["../../src/tools/registry-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AACjF,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAEvD,MAAM,UAAU,2BAA2B,CAAC,MAAiB;IAC3D,MAAM,CAAC,IAAI,CACT,qBAAqB,EACrB,8DAA8D,EAC9D;QACE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oCAAoC,CAAC;QAChE,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;QACjD,IAAI,EAAE,CAAC;aACJ,MAAM,EAAE;aACR,GAAG,EAAE;aACL,GAAG,CAAC,CAAC,CAAC;aACN,QAAQ,EAAE;aACV,QAAQ,CAAC,wCAAwC,CAAC;KACtD,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE;gBAC5C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,IAAI,EAAE,IAAI,CAAC,IAAI;aAChB,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,KAAK,CAAC;gBAC/B,CAAC,CAAC,8BAA8B,IAAI,CAAC,KAAK,IAAI;gBAC9C,CAAC,CAAC,OAAO;qBACJ,GAAG,CACF,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACP,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,EAAE,KAAK;oBACrC,MAAM,CAAC,CAAC,WAAW,IAAI,gBAAgB,IAAI;oBAC3C,cAAc,CAAC,CAAC,MAAM,gBAAgB,CAAC,CAAC,aAAa,IAAI,KAAK,eAAe,CAAC,CAAC,cAAc,IAAI,KAAK,EAAE,CAC3G;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,uBAAuB,EACvB,sFAAsF,EACtF;QACE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;KACrE,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAElD,MAAM,KAAK,GAAG;gBACZ,KAAK,OAAO,CAAC,IAAI,EAAE;gBACnB,EAAE;gBACF,OAAO,CAAC,WAAW,IAAI,EAAE;gBACzB,EAAE;gBACF,iBAAiB,OAAO,CAAC,MAAM,EAAE;gBACjC,kBAAkB,OAAO,CAAC,OAAO,IAAI,KAAK,EAAE;gBAC5C,qBAAqB,OAAO,CAAC,UAAU,IAAI,KAAK,EAAE;gBAClD,yBAAyB,OAAO,CAAC,aAAa,IAAI,KAAK,EAAE;gBACzD,yBAAyB,OAAO,CAAC,cAAc,IAAI,KAAK,EAAE;gBAC1D,kBAAkB,OAAO,CAAC,UAAU,IAAI,KAAK,EAAE;gBAC/C,kBAAkB,OAAO,CAAC,UAAU,IAAI,KAAK,EAAE;aAChD,CAAC;YAEF,IAAI,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;gBAC1B,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,mBAAmB,EAAE,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YAC7E,CAAC;YAED,IAAI,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;gBAC7B,KAAK,CAAC,IAAI,CACR,EAAE,EACF,oBAAoB,EACpB,GAAG,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CACjC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CACpE,CACF,CAAC;YACJ,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/skill-generator.d.ts

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

package/dist/tools/skill-generator.d.ts.map

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

package/dist/tools/skill-generator.js

@@ -0,0 +1,42 @@
+import { z } from "zod";
+import { generateSkill } from "../services/skill-generator.js";
+import { errorToToolResult } from "../utils/errors.js";
+import { writeFile, mkdir } from "node:fs/promises";
+import { join } from "node:path";
+export function registerSkillGeneratorTools(server) {
+    server.tool("generate_node_skill", "Analyze a ComfyUI custom node pack and generate a Claude skill (.md) file describing all its nodes, inputs/outputs, and usage examples. Accepts a ComfyUI Registry ID or a GitHub repository URL as the source.", {
+        source: z
+            .string()
+            .describe("ComfyUI Registry node ID (e.g. 'comfyui-impact-pack') or GitHub repository URL"),
+        install_in: z
+            .string()
+            .optional()
+            .describe("Optional directory path to save the generated SKILL.md file"),
+    }, async (args) => {
+        try {
+            const markdown = await generateSkill(args.source);
+            // Optionally write to disk
+            if (args.install_in) {
+                const dir = args.install_in;
+                await mkdir(dir, { recursive: true });
+                const filePath = join(dir, "SKILL.md");
+                await writeFile(filePath, markdown, "utf-8");
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: `Skill file written to ${filePath}\n\n${markdown}`,
+                        },
+                    ],
+                };
+            }
+            return {
+                content: [{ type: "text", text: markdown }],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+}
+//# sourceMappingURL=skill-generator.js.map

package/dist/tools/skill-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"skill-generator.js","sourceRoot":"","sources":["../../src/tools/skill-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAW,IAAI,EAAE,MAAM,WAAW,CAAC;AAE1C,MAAM,UAAU,2BAA2B,CAAC,MAAiB;IAC3D,MAAM,CAAC,IAAI,CACT,qBAAqB,EACrB,iNAAiN,EACjN;QACE,MAAM,EAAE,CAAC;aACN,MAAM,EAAE;aACR,QAAQ,CACP,gFAAgF,CACjF;QACH,UAAU,EAAE,CAAC;aACV,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CACP,6DAA6D,CAC9D;KACJ,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAElD,2BAA2B;YAC3B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC;gBAC5B,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;gBACvC,MAAM,SAAS,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;gBAC7C,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,yBAAyB,QAAQ,OAAO,QAAQ,EAAE;yBACzD;qBACF;iBACF,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;aACrD,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/dist/tools/workflow-compose.d.ts

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

package/dist/tools/workflow-compose.d.ts.map

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

package/dist/tools/workflow-compose.js

@@ -0,0 +1,180 @@
+import { z } from "zod";
+import { createWorkflow, modifyWorkflow, TEMPLATE_NAMES, } from "../services/workflow-composer.js";
+import { getObjectInfo } from "../comfyui/client.js";
+import { errorToToolResult, ValidationError } from "../utils/errors.js";
+import { logger } from "../utils/logger.js";
+function parseWorkflow(input) {
+    if (typeof input === "string") {
+        try {
+            const parsed = JSON.parse(input);
+            if (typeof parsed !== "object" || parsed === null || Array.isArray(parsed)) {
+                throw new ValidationError("Workflow JSON must be an object with node IDs as keys");
+            }
+            return parsed;
+        }
+        catch (err) {
+            if (err instanceof ValidationError)
+                throw err;
+            throw new ValidationError(`Invalid JSON string: ${err.message}`);
+        }
+    }
+    if (typeof input === "object" && input !== null && !Array.isArray(input)) {
+        return input;
+    }
+    throw new ValidationError("Workflow must be a JSON string or object");
+}
+const operationSchema = z.discriminatedUnion("op", [
+    z.object({
+        op: z.literal("set_input"),
+        node_id: z.string(),
+        input_name: z.string(),
+        value: z.any(),
+    }),
+    z.object({
+        op: z.literal("add_node"),
+        class_type: z.string(),
+        inputs: z.record(z.any()).optional(),
+        id: z.string().optional(),
+    }),
+    z.object({
+        op: z.literal("remove_node"),
+        node_id: z.string(),
+    }),
+    z.object({
+        op: z.literal("connect"),
+        source_id: z.string(),
+        output_index: z.number(),
+        target_id: z.string(),
+        input_name: z.string(),
+    }),
+    z.object({
+        op: z.literal("insert_between"),
+        source_id: z.string(),
+        output_index: z.number(),
+        target_id: z.string(),
+        input_name: z.string(),
+        new_class_type: z.string(),
+        new_inputs: z.record(z.any()).optional(),
+    }),
+]);
+export function registerWorkflowComposeTools(server) {
+    // 1. create_workflow
+    server.tool("create_workflow", `Create a ComfyUI workflow from a named template. Available templates: ${TEMPLATE_NAMES.join(", ")}. Returns the complete workflow JSON ready for execution or further modification.`, {
+        template: z
+            .enum(TEMPLATE_NAMES)
+            .describe("Template name: txt2img, img2img, upscale, or inpaint"),
+        params: z
+            .record(z.any())
+            .optional()
+            .default({})
+            .describe("Template parameters (e.g. checkpoint, positive_prompt, negative_prompt, width, height, steps, cfg, seed, sampler_name, scheduler, denoise, image_path, mask_path, upscale_model)"),
+    }, async ({ template, params }) => {
+        try {
+            logger.info("Creating workflow", { template, params });
+            const workflow = createWorkflow(template, params);
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify(workflow, null, 2),
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    // 2. modify_workflow
+    server.tool("modify_workflow", "Apply modification operations to an existing ComfyUI workflow. Supports: set_input, add_node, remove_node, connect, insert_between. Returns the modified workflow JSON and IDs of any newly added nodes.", {
+        workflow: z
+            .union([z.string(), z.record(z.any())])
+            .describe("ComfyUI workflow JSON (as a JSON string or object)"),
+        operations: z
+            .array(operationSchema)
+            .describe("Array of operations to apply in order. Each has an 'op' field: set_input, add_node, remove_node, connect, or insert_between"),
+    }, async ({ workflow, operations }) => {
+        try {
+            logger.info("Modifying workflow", { opCount: operations.length });
+            const parsed = parseWorkflow(workflow);
+            const result = modifyWorkflow(parsed, operations);
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            workflow: result.workflow,
+                            added_node_ids: result.added_ids,
+                        }, null, 2),
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    // 3. get_node_info
+    server.tool("get_node_info", "Query ComfyUI's /object_info endpoint to get available node type definitions. Optionally filter by node type name (substring match). Returns node inputs, outputs, and descriptions.", {
+        node_type: z
+            .string()
+            .optional()
+            .describe("Filter by node class_type name (case-insensitive substring match). Omit to list all available nodes."),
+    }, async ({ node_type }) => {
+        try {
+            logger.info("Getting node info", { filter: node_type });
+            const info = await getObjectInfo();
+            let entries = Object.entries(info);
+            if (node_type) {
+                const lower = node_type.toLowerCase();
+                entries = entries.filter(([name]) => name.toLowerCase().includes(lower));
+            }
+            if (entries.length === 0) {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: node_type
+                                ? `No nodes found matching "${node_type}"`
+                                : "No node definitions returned from ComfyUI",
+                        },
+                    ],
+                };
+            }
+            // For large result sets, return just names + descriptions
+            if (entries.length > 20) {
+                const summary = entries.map(([name, def]) => ({
+                    name,
+                    display_name: def.display_name,
+                    category: def.category,
+                    description: def.description || "",
+                }));
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: JSON.stringify({
+                                count: summary.length,
+                                nodes: summary,
+                                hint: "Use a more specific node_type filter to see full definitions with inputs/outputs",
+                            }, null, 2),
+                        },
+                    ],
+                };
+            }
+            const result = Object.fromEntries(entries);
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify(result, null, 2),
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+}
+//# sourceMappingURL=workflow-compose.js.map

package/dist/tools/workflow-compose.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-compose.js","sourceRoot":"","sources":["../../src/tools/workflow-compose.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EACL,cAAc,EACd,cAAc,EACd,cAAc,GAEf,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACxE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3E,MAAM,IAAI,eAAe,CAAC,uDAAuD,CAAC,CAAC;YACrF,CAAC;YACD,OAAO,MAAsB,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,eAAe;gBAAE,MAAM,GAAG,CAAC;YAC9C,MAAM,IAAI,eAAe,CAAC,wBAAyB,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzE,OAAO,KAAqB,CAAC;IAC/B,CAAC;IACD,MAAM,IAAI,eAAe,CAAC,0CAA0C,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,eAAe,GAAG,CAAC,CAAC,kBAAkB,CAAC,IAAI,EAAE;IACjD,CAAC,CAAC,MAAM,CAAC;QACP,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;QAC1B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;QACnB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;QACtB,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE;KACf,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC;QACzB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;QACtB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,EAAE;QACpC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KAC1B,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;QAC5B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;KACpB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC;QACxB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;QACxB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;KACvB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;QAC/B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;QACxB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;QACtB,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;QAC1B,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,EAAE;KACzC,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,UAAU,4BAA4B,CAAC,MAAiB;IAC5D,qBAAqB;IACrB,MAAM,CAAC,IAAI,CACT,iBAAiB,EACjB,yEAAyE,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,mFAAmF,EACrL;QACE,QAAQ,EAAE,CAAC;aACR,IAAI,CAAC,cAAuC,CAAC;aAC7C,QAAQ,CAAC,sDAAsD,CAAC;QACnE,MAAM,EAAE,CAAC;aACN,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;aACf,QAAQ,EAAE;aACV,OAAO,CAAC,EAAE,CAAC;aACX,QAAQ,CACP,kLAAkL,CACnL;KACJ,EACD,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE;QAC7B,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,mBAAmB,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YACvD,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAElD,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;qBACxC;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,qBAAqB;IACrB,MAAM,CAAC,IAAI,CACT,iBAAiB,EACjB,0MAA0M,EAC1M;QACE,QAAQ,EAAE,CAAC;aACR,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;aACtC,QAAQ,CAAC,oDAAoD,CAAC;QACjE,UAAU,EAAE,CAAC;aACV,KAAK,CAAC,eAAe,CAAC;aACtB,QAAQ,CACP,6HAA6H,CAC9H;KACJ,EACD,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,OAAO,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;YAClE,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;YACvC,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,EAAE,UAA+B,CAAC,CAAC;YAEvE,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB;4BACE,QAAQ,EAAE,MAAM,CAAC,QAAQ;4BACzB,cAAc,EAAE,MAAM,CAAC,SAAS;yBACjC,EACD,IAAI,EACJ,CAAC,CACF;qBACF;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,mBAAmB;IACnB,MAAM,CAAC,IAAI,CACT,eAAe,EACf,sLAAsL,EACtL;QACE,SAAS,EAAE,CAAC;aACT,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CACP,sGAAsG,CACvG;KACJ,EACD,KAAK,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;QACtB,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,mBAAmB,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YACxD,MAAM,IAAI,GAAG,MAAM,aAAa,EAAE,CAAC;YAEnC,IAAI,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACnC,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,KAAK,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;gBACtC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;YAC3E,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,SAAS;gCACb,CAAC,CAAC,4BAA4B,SAAS,GAAG;gCAC1C,CAAC,CAAC,2CAA2C;yBAChD;qBACF;iBACF,CAAC;YACJ,CAAC;YAED,0DAA0D;YAC1D,IAAI,OAAO,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC5C,IAAI;oBACJ,YAAY,EAAE,GAAG,CAAC,YAAY;oBAC9B,QAAQ,EAAE,GAAG,CAAC,QAAQ;oBACtB,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE;iBACnC,CAAC,CAAC,CAAC;gBACJ,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB;gCACE,KAAK,EAAE,OAAO,CAAC,MAAM;gCACrB,KAAK,EAAE,OAAO;gCACd,IAAI,EAAE,kFAAkF;6BACzF,EACD,IAAI,EACJ,CAAC,CACF;yBACF;qBACF;iBACF,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAC3C,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;qBACtC;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/dist/tools/workflow-execute.d.ts

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

package/dist/tools/workflow-execute.d.ts.map

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

package/dist/tools/workflow-execute.js

@@ -0,0 +1,115 @@
+import { z } from "zod";
+import { executeWorkflow, getJobStatus, getQueueStatus, cancelCurrentJob, getSystemInfo, } from "../services/workflow-executor.js";
+import { errorToToolResult } from "../utils/errors.js";
+export function registerWorkflowExecuteTools(server) {
+    server.tool("run_workflow", "Execute a ComfyUI workflow (API format JSON). Returns generated images as base64. The workflow must be in ComfyUI's API/prompt format — a mapping of node IDs to {class_type, inputs}.", {
+        workflow: z
+            .record(z.string(), z.any())
+            .describe("ComfyUI workflow in API format (node ID -> {class_type, inputs})"),
+        timeout_ms: z
+            .number()
+            .int()
+            .positive()
+            .optional()
+            .describe("Execution timeout in milliseconds (default: 10 minutes)"),
+        disable_random_seed: z
+            .boolean()
+            .optional()
+            .describe("If true, do not randomize seed values"),
+    }, async (args) => {
+        try {
+            const result = await executeWorkflow(args.workflow, {
+                timeout_ms: args.timeout_ms,
+                disable_random_seed: args.disable_random_seed,
+            });
+            const content = [];
+            content.push({
+                type: "text",
+                text: `Workflow executed successfully. prompt_id: ${result.prompt_id}, images: ${result.images.length}`,
+            });
+            for (const img of result.images) {
+                content.push({
+                    type: "image",
+                    data: img.data,
+                    mimeType: img.mime,
+                });
+            }
+            return { content };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    server.tool("get_job_status", "Check the execution status of a ComfyUI prompt/job by its ID.", {
+        prompt_id: z.string().describe("The prompt ID returned by run_workflow"),
+    }, async (args) => {
+        try {
+            const status = await getJobStatus(args.prompt_id);
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify(status, null, 2),
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    server.tool("get_queue", "Get the current ComfyUI execution queue showing running and pending jobs.", {}, async () => {
+        try {
+            const queue = await getQueueStatus();
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            running: queue.queue_running.length,
+                            pending: queue.queue_pending.length,
+                            queue_running: queue.queue_running,
+                            queue_pending: queue.queue_pending,
+                        }, null, 2),
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    server.tool("cancel_job", "Interrupt/cancel the currently running ComfyUI job.", {}, async () => {
+        try {
+            await cancelCurrentJob();
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: "Current job cancelled successfully.",
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    server.tool("get_system_stats", "Get ComfyUI system information including GPU, VRAM, Python version, and OS details.", {}, async () => {
+        try {
+            const stats = await getSystemInfo();
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify(stats, null, 2),
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+}
+//# sourceMappingURL=workflow-execute.js.map

package/dist/tools/workflow-execute.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-execute.js","sourceRoot":"","sources":["../../src/tools/workflow-execute.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EACL,eAAe,EACf,YAAY,EACZ,cAAc,EACd,gBAAgB,EAChB,aAAa,GACd,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAEvD,MAAM,UAAU,4BAA4B,CAAC,MAAiB;IAC5D,MAAM,CAAC,IAAI,CACT,cAAc,EACd,wLAAwL,EACxL;QACE,QAAQ,EAAE,CAAC;aACR,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;aAC3B,QAAQ,CAAC,kEAAkE,CAAC;QAC/E,UAAU,EAAE,CAAC;aACV,MAAM,EAAE;aACR,GAAG,EAAE;aACL,QAAQ,EAAE;aACV,QAAQ,EAAE;aACV,QAAQ,CAAC,yDAAyD,CAAC;QACtE,mBAAmB,EAAE,CAAC;aACnB,OAAO,EAAE;aACT,QAAQ,EAAE;aACV,QAAQ,CAAC,uCAAuC,CAAC;KACrD,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAClD,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;aAC9C,CAAC,CAAC;YAEH,MAAM,OAAO,GAET,EAAE,CAAC;YAEP,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,8CAA8C,MAAM,CAAC,SAAS,aAAa,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE;aACxG,CAAC,CAAC;YAEH,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAChC,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,QAAQ,EAAE,GAAG,CAAC,IAAI;iBACnB,CAAC,CAAC;YACL,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,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,+DAA+D,EAC/D;QACE,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;KACzE,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClD,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;qBACtC;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,WAAW,EACX,2EAA2E,EAC3E,EAAE,EACF,KAAK,IAAI,EAAE;QACT,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;YACrC,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB;4BACE,OAAO,EAAE,KAAK,CAAC,aAAa,CAAC,MAAM;4BACnC,OAAO,EAAE,KAAK,CAAC,aAAa,CAAC,MAAM;4BACnC,aAAa,EAAE,KAAK,CAAC,aAAa;4BAClC,aAAa,EAAE,KAAK,CAAC,aAAa;yBACnC,EACD,IAAI,EACJ,CAAC,CACF;qBACF;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,YAAY,EACZ,qDAAqD,EACrD,EAAE,EACF,KAAK,IAAI,EAAE;QACT,IAAI,CAAC;YACH,MAAM,gBAAgB,EAAE,CAAC;YACzB,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,qCAAqC;qBAC5C;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,kBAAkB,EAClB,qFAAqF,EACrF,EAAE,EACF,KAAK,IAAI,EAAE;QACT,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,aAAa,EAAE,CAAC;YACpC,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;qBACrC;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/dist/tools/workflow-visualize.d.ts

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

package/dist/tools/workflow-visualize.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-visualize.d.ts","sourceRoot":"","sources":["../../src/tools/workflow-visualize.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AA2BpE,wBAAgB,8BAA8B,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI,CAkHtE"}

package/dist/tools/workflow-visualize.js

@@ -0,0 +1,111 @@
+import { z } from "zod";
+import { convertToMermaid } from "../services/mermaid-converter.js";
+import { parseMermaid, resolveWorkflow } from "../services/mermaid-parser.js";
+import { getObjectInfo } from "../comfyui/client.js";
+import { errorToToolResult, ValidationError } from "../utils/errors.js";
+import { logger } from "../utils/logger.js";
+function parseWorkflow(input) {
+    if (typeof input === "string") {
+        try {
+            const parsed = JSON.parse(input);
+            if (typeof parsed !== "object" || parsed === null || Array.isArray(parsed)) {
+                throw new ValidationError("Workflow JSON must be an object with node IDs as keys");
+            }
+            return parsed;
+        }
+        catch (err) {
+            if (err instanceof ValidationError)
+                throw err;
+            throw new ValidationError(`Invalid JSON string: ${err.message}`);
+        }
+    }
+    if (typeof input === "object" && input !== null && !Array.isArray(input)) {
+        return input;
+    }
+    throw new ValidationError("Workflow must be a JSON string or object");
+}
+export function registerWorkflowVisualizeTools(server) {
+    server.tool("visualize_workflow", "Convert a ComfyUI workflow JSON into a Mermaid flowchart diagram. Returns mermaid syntax showing nodes grouped by category (loading, conditioning, sampling, image, output) with connections labeled by data type.", {
+        workflow: z
+            .union([z.string(), z.record(z.any())])
+            .describe("ComfyUI workflow JSON (as a JSON string or object)"),
+        show_values: z
+            .boolean()
+            .optional()
+            .default(true)
+            .describe("Include widget values (seed, steps, cfg, etc.) in node labels"),
+        direction: z
+            .enum(["LR", "TB"])
+            .optional()
+            .default("LR")
+            .describe("Flowchart direction: LR (left-to-right) or TB (top-to-bottom)"),
+    }, async ({ workflow, show_values, direction }) => {
+        try {
+            logger.info("Visualizing workflow");
+            const parsed = parseWorkflow(workflow);
+            const nodeCount = Object.keys(parsed).length;
+            if (nodeCount === 0) {
+                throw new ValidationError("Workflow contains no nodes");
+            }
+            const mermaid = convertToMermaid(parsed, {
+                showValues: show_values,
+                direction,
+            });
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: `\`\`\`mermaid\n${mermaid}\n\`\`\``,
+                    },
+                ],
+            };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+    server.tool("mermaid_to_workflow", "Convert a Mermaid flowchart diagram back into a ComfyUI workflow JSON. " +
+        "Parses node definitions, connections (with data type labels), and widget values from the mermaid syntax. " +
+        "Resolves node types and wires connections using ComfyUI's /object_info schemas. " +
+        "Fills missing inputs with defaults. Returns a valid, executable ComfyUI API workflow.", {
+        mermaid: z
+            .string()
+            .describe("Mermaid flowchart text (with or without ```mermaid code fence). " +
+            "Nodes should use ComfyUI class_type names as labels. " +
+            "Connections should be labeled with data types (e.g., -->|MODEL|)."),
+    }, async ({ mermaid }) => {
+        try {
+            logger.info("Converting mermaid to workflow");
+            // Parse the mermaid text into nodes and edges
+            const parsed = parseMermaid(mermaid);
+            logger.info(`Parsed ${parsed.nodes.size} nodes and ${parsed.edges.length} edges`);
+            // Fetch node definitions from ComfyUI for schema resolution
+            const objectInfo = await getObjectInfo();
+            // Resolve into a valid workflow
+            const { workflow, warnings } = resolveWorkflow(parsed, objectInfo);
+            const nodeCount = Object.keys(workflow).length;
+            const connectionCount = Object.values(workflow).reduce((count, node) => count +
+                Object.values(node.inputs).filter((v) => Array.isArray(v) &&
+                    v.length === 2 &&
+                    typeof v[0] === "string" &&
+                    typeof v[1] === "number").length, 0);
+            const content = [];
+            // Summary
+            let summary = `Converted mermaid to workflow: ${nodeCount} nodes, ${connectionCount} connections.`;
+            if (warnings.length > 0) {
+                summary += `\n\n**Warnings (${warnings.length}):**\n${warnings.map((w) => `- ${w}`).join("\n")}`;
+            }
+            content.push({ type: "text", text: summary });
+            // The workflow JSON
+            content.push({
+                type: "text",
+                text: JSON.stringify(workflow, null, 2),
+            });
+            return { content };
+        }
+        catch (err) {
+            return errorToToolResult(err);
+        }
+    });
+}
+//# sourceMappingURL=workflow-visualize.js.map

package/dist/tools/workflow-visualize.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-visualize.js","sourceRoot":"","sources":["../../src/tools/workflow-visualize.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAC9E,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACxE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3E,MAAM,IAAI,eAAe,CAAC,uDAAuD,CAAC,CAAC;YACrF,CAAC;YACD,OAAO,MAAsB,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,eAAe;gBAAE,MAAM,GAAG,CAAC;YAC9C,MAAM,IAAI,eAAe,CAAC,wBAAyB,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzE,OAAO,KAAqB,CAAC;IAC/B,CAAC;IACD,MAAM,IAAI,eAAe,CAAC,0CAA0C,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,8BAA8B,CAAC,MAAiB;IAC9D,MAAM,CAAC,IAAI,CACT,oBAAoB,EACpB,oNAAoN,EACpN;QACE,QAAQ,EAAE,CAAC;aACR,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;aACtC,QAAQ,CAAC,oDAAoD,CAAC;QACjE,WAAW,EAAE,CAAC;aACX,OAAO,EAAE;aACT,QAAQ,EAAE;aACV,OAAO,CAAC,IAAI,CAAC;aACb,QAAQ,CAAC,+DAA+D,CAAC;QAC5E,SAAS,EAAE,CAAC;aACT,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;aAClB,QAAQ,EAAE;aACV,OAAO,CAAC,IAAI,CAAC;aACb,QAAQ,CAAC,+DAA+D,CAAC;KAC7E,EACD,KAAK,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,EAAE,EAAE;QAC7C,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YACpC,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;YAEvC,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC;YAC7C,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;gBACpB,MAAM,IAAI,eAAe,CAAC,4BAA4B,CAAC,CAAC;YAC1D,CAAC;YAED,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;gBACvC,UAAU,EAAE,WAAW;gBACvB,SAAS;aACV,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,kBAAkB,OAAO,UAAU;qBAC1C;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,qBAAqB,EACrB,yEAAyE;QACvE,2GAA2G;QAC3G,kFAAkF;QAClF,uFAAuF,EACzF;QACE,OAAO,EAAE,CAAC;aACP,MAAM,EAAE;aACR,QAAQ,CACP,kEAAkE;YAChE,uDAAuD;YACvD,mEAAmE,CACtE;KACJ,EACD,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;QACpB,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;YAE9C,8CAA8C;YAC9C,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,CAAC,IAAI,CACT,UAAU,MAAM,CAAC,KAAK,CAAC,IAAI,cAAc,MAAM,CAAC,KAAK,CAAC,MAAM,QAAQ,CACrE,CAAC;YAEF,4DAA4D;YAC5D,MAAM,UAAU,GAAG,MAAM,aAAa,EAAE,CAAC;YAEzC,gCAAgC;YAChC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,eAAe,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAEnE,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;YAC/C,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,MAAM,CACpD,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CACd,KAAK;gBACL,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAC/B,CAAC,CAAC,EAAE,EAAE,CACJ,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;oBAChB,CAAC,CAAC,MAAM,KAAK,CAAC;oBACd,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,QAAQ;oBACxB,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,QAAQ,CAC3B,CAAC,MAAM,EACV,CAAC,CACF,CAAC;YAEF,MAAM,OAAO,GAA0C,EAAE,CAAC;YAE1D,UAAU;YACV,IAAI,OAAO,GAAG,kCAAkC,SAAS,WAAW,eAAe,eAAe,CAAC;YACnG,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,OAAO,IAAI,mBAAmB,QAAQ,CAAC,MAAM,SAAS,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACnG,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YAE9C,oBAAoB;YACpB,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;aACxC,CAAC,CAAC;YAEH,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/dist/utils/errors.d.ts

@@ -0,0 +1,24 @@
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+export declare class ComfyUIError extends Error {
+    readonly code: string;
+    readonly details?: unknown | undefined;
+    constructor(message: string, code: string, details?: unknown | undefined);
+    toToolResult(): CallToolResult;
+}
+export declare class ConnectionError extends ComfyUIError {
+    constructor(message: string);
+}
+export declare class WorkflowExecutionError extends ComfyUIError {
+    constructor(message: string, details?: unknown);
+}
+export declare class ValidationError extends ComfyUIError {
+    constructor(message: string);
+}
+export declare class RegistryError extends ComfyUIError {
+    constructor(message: string, details?: unknown);
+}
+export declare class ModelError extends ComfyUIError {
+    constructor(message: string, details?: unknown);
+}
+export declare function errorToToolResult(err: unknown): CallToolResult;
+//# sourceMappingURL=errors.d.ts.map

package/dist/utils/errors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../../src/utils/errors.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAEzE,qBAAa,YAAa,SAAQ,KAAK;aAGnB,IAAI,EAAE,MAAM;aACZ,OAAO,CAAC,EAAE,OAAO;gBAFjC,OAAO,EAAE,MAAM,EACC,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,OAAO,YAAA;IAMnC,YAAY,IAAI,cAAc;CAe/B;AAED,qBAAa,eAAgB,SAAQ,YAAY;gBACnC,OAAO,EAAE,MAAM;CAI5B;AAED,qBAAa,sBAAuB,SAAQ,YAAY;gBAC1C,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO;CAI/C;AAED,qBAAa,eAAgB,SAAQ,YAAY;gBACnC,OAAO,EAAE,MAAM;CAI5B;AAED,qBAAa,aAAc,SAAQ,YAAY;gBACjC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO;CAI/C;AAED,qBAAa,UAAW,SAAQ,YAAY;gBAC9B,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO;CAI/C;AAED,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,OAAO,GAAG,cAAc,CAO9D"}