comfyui-mcp 0.1.0

package/.mcp.json

@@ -0,0 +1,11 @@
+{
+  "comfyui": {
+    "command": "npx",
+    "args": ["-y", "comfyui-mcp"],
+    "env": {
+      "COMFYUI_HOST": "127.0.0.1",
+      "COMFYUI_PORT": "8000",
+      "CIVITAI_API_TOKEN": ""
+    }
+  }
+}

package/dist/comfyui/client.d.ts

@@ -0,0 +1,16 @@
+import { Client } from "@stable-canvas/comfyui-client";
+import type { ObjectInfo, SystemStats, QueueStatus } from "./types.js";
+export declare function getClient(): Client;
+export declare function connectClient(): Promise<Client>;
+export declare function getSystemStats(): Promise<SystemStats>;
+export declare function getObjectInfo(): Promise<ObjectInfo>;
+export declare function getQueue(): Promise<QueueStatus>;
+export declare function interrupt(): Promise<void>;
+export declare function getSamplers(): Promise<string[]>;
+export declare function getSchedulers(): Promise<string[]>;
+export declare function getCheckpoints(): Promise<string[]>;
+export declare function getLoRAs(): Promise<string[]>;
+export declare function getVAEs(): Promise<string[]>;
+export declare function getUpscaleModels(): Promise<string[]>;
+export declare function getComfyUIPath(): string | undefined;
+//# sourceMappingURL=client.d.ts.map

package/dist/comfyui/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/comfyui/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAC;AAIvD,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAIvE,wBAAgB,SAAS,IAAI,MAAM,CAWlC;AAED,wBAAsB,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC,CAWrD;AAED,wBAAsB,cAAc,IAAI,OAAO,CAAC,WAAW,CAAC,CAI3D;AAED,wBAAsB,aAAa,IAAI,OAAO,CAAC,UAAU,CAAC,CAIzD;AAED,wBAAsB,QAAQ,IAAI,OAAO,CAAC,WAAW,CAAC,CAIrD;AAED,wBAAsB,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC,CAG/C;AAED,wBAAsB,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAGrD;AAED,wBAAsB,aAAa,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAGvD;AAED,wBAAsB,cAAc,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAGxD;AAED,wBAAsB,QAAQ,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAGlD;AAED,wBAAsB,OAAO,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAGjD;AAED,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAG1D;AAED,wBAAgB,cAAc,IAAI,MAAM,GAAG,SAAS,CAEnD"}

package/dist/comfyui/client.js

@@ -0,0 +1,75 @@
+import { Client } from "@stable-canvas/comfyui-client";
+import { config, getComfyUIApiHost } from "../config.js";
+import { logger } from "../utils/logger.js";
+import { ConnectionError } from "../utils/errors.js";
+let clientInstance = null;
+export function getClient() {
+    if (!clientInstance) {
+        clientInstance = new Client({
+            api_host: getComfyUIApiHost(),
+            // Node 22+ provides global WebSocket
+        });
+        logger.info("ComfyUI client created", {
+            host: getComfyUIApiHost(),
+        });
+    }
+    return clientInstance;
+}
+export async function connectClient() {
+    const client = getClient();
+    try {
+        await client.connect();
+        logger.info("Connected to ComfyUI via WebSocket");
+        return client;
+    }
+    catch (err) {
+        throw new ConnectionError(`Failed to connect to ComfyUI at ${getComfyUIApiHost()}: ${err instanceof Error ? err.message : err}`);
+    }
+}
+export async function getSystemStats() {
+    const client = getClient();
+    const stats = await client.getSystemStats();
+    return stats;
+}
+export async function getObjectInfo() {
+    const client = getClient();
+    const info = await client.getNodeDefs();
+    return info;
+}
+export async function getQueue() {
+    const client = getClient();
+    const queue = await client.getQueue();
+    return queue;
+}
+export async function interrupt() {
+    const client = getClient();
+    await client.interrupt();
+}
+export async function getSamplers() {
+    const client = getClient();
+    return client.getSamplers();
+}
+export async function getSchedulers() {
+    const client = getClient();
+    return client.getSchedulers();
+}
+export async function getCheckpoints() {
+    const client = getClient();
+    return client.getSDModels();
+}
+export async function getLoRAs() {
+    const client = getClient();
+    return client.getLoRAs();
+}
+export async function getVAEs() {
+    const client = getClient();
+    return client.getVAEs();
+}
+export async function getUpscaleModels() {
+    const client = getClient();
+    return client.getUpscaleModels();
+}
+export function getComfyUIPath() {
+    return config.comfyuiPath;
+}
+//# sourceMappingURL=client.js.map

package/dist/comfyui/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/comfyui/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAC;AACvD,OAAO,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAGrD,IAAI,cAAc,GAAkB,IAAI,CAAC;AAEzC,MAAM,UAAU,SAAS;IACvB,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,cAAc,GAAG,IAAI,MAAM,CAAC;YAC1B,QAAQ,EAAE,iBAAiB,EAAE;YAC7B,qCAAqC;SACtC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE;YACpC,IAAI,EAAE,iBAAiB,EAAE;SAC1B,CAAC,CAAC;IACL,CAAC;IACD,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa;IACjC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;QAClD,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,IAAI,eAAe,CACvB,mCAAmC,iBAAiB,EAAE,KAAK,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,CACtG,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc;IAClC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;IAC5C,OAAO,KAA+B,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa;IACjC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;IACxC,OAAO,IAA6B,CAAC;AACvC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ;IAC5B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,QAAQ,EAAE,CAAC;IACtC,OAAO,KAA+B,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS;IAC7B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;AAC3B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW;IAC/B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,MAAM,CAAC,WAAW,EAAE,CAAC;AAC9B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa;IACjC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,MAAM,CAAC,aAAa,EAAE,CAAC;AAChC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc;IAClC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,MAAM,CAAC,WAAW,EAAE,CAAC;AAC9B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ;IAC5B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,MAAM,CAAC,QAAQ,EAAE,CAAC;AAC3B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,OAAO;IAC3B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB;IACpC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,MAAM,CAAC,gBAAgB,EAAE,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,cAAc;IAC5B,OAAO,MAAM,CAAC,WAAW,CAAC;AAC5B,CAAC"}

package/dist/comfyui/events.d.ts

@@ -0,0 +1,9 @@
+import type { Client } from "@stable-canvas/comfyui-client";
+import type { JobProgress } from "./types.js";
+export interface ExecutionCallbacks {
+    onProgress?: (progress: JobProgress) => void;
+    onComplete?: (promptId: string) => void;
+    onError?: (promptId: string, error: string) => void;
+}
+export declare function attachExecutionListeners(client: Client, promptId: string, callbacks: ExecutionCallbacks): () => void;
+//# sourceMappingURL=events.d.ts.map

package/dist/comfyui/events.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.d.ts","sourceRoot":"","sources":["../../src/comfyui/events.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAC;AAE5D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,WAAW,kBAAkB;IACjC,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,WAAW,KAAK,IAAI,CAAC;IAC7C,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;IACxC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACrD;AAED,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,kBAAkB,GAC5B,MAAM,IAAI,CAwCZ"}

package/dist/comfyui/events.js

@@ -0,0 +1,40 @@
+import { logger } from "../utils/logger.js";
+export function attachExecutionListeners(client, promptId, callbacks) {
+    const unsubscribers = [];
+    if (callbacks.onProgress) {
+        const unsub = client.on("progress", (data) => {
+            if (data.prompt_id === promptId) {
+                callbacks.onProgress({
+                    value: data.value,
+                    max: data.max,
+                    node: data.node,
+                    prompt_id: data.prompt_id,
+                });
+            }
+        });
+        unsubscribers.push(unsub);
+    }
+    if (callbacks.onComplete) {
+        const unsub = client.on("execution_success", (data) => {
+            if (data.prompt_id === promptId) {
+                logger.info(`Execution completed: ${promptId}`);
+                callbacks.onComplete(promptId);
+            }
+        });
+        unsubscribers.push(unsub);
+    }
+    if (callbacks.onError) {
+        const unsub = client.on("execution_error", (data) => {
+            if (data.prompt_id === promptId) {
+                logger.error(`Execution error: ${promptId}`, data.exception_message);
+                callbacks.onError(promptId, data.exception_message);
+            }
+        });
+        unsubscribers.push(unsub);
+    }
+    return () => {
+        for (const unsub of unsubscribers)
+            unsub();
+    };
+}
+//# sourceMappingURL=events.js.map

package/dist/comfyui/events.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.js","sourceRoot":"","sources":["../../src/comfyui/events.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAS5C,MAAM,UAAU,wBAAwB,CACtC,MAAc,EACd,QAAgB,EAChB,SAA6B;IAE7B,MAAM,aAAa,GAAsB,EAAE,CAAC;IAE5C,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,IAAI,EAAE,EAAE;YAC3C,IAAI,IAAI,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAChC,SAAS,CAAC,UAAW,CAAC;oBACpB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,GAAG,EAAE,IAAI,CAAC,GAAG;oBACb,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,SAAS,EAAE,IAAI,CAAC,SAAS;iBAC1B,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QACH,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,IAAI,EAAE,EAAE;YACpD,IAAI,IAAI,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAChC,MAAM,CAAC,IAAI,CAAC,wBAAwB,QAAQ,EAAE,CAAC,CAAC;gBAChD,SAAS,CAAC,UAAW,CAAC,QAAQ,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QACH,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;QACtB,MAAM,KAAK,GAAG,MAAM,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,IAAI,EAAE,EAAE;YAClD,IAAI,IAAI,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAChC,MAAM,CAAC,KAAK,CAAC,oBAAoB,QAAQ,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;gBACrE,SAAS,CAAC,OAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACvD,CAAC;QACH,CAAC,CAAC,CAAC;QACH,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO,GAAG,EAAE;QACV,KAAK,MAAM,KAAK,IAAI,aAAa;YAAE,KAAK,EAAE,CAAC;IAC7C,CAAC,CAAC;AACJ,CAAC"}

package/dist/comfyui/types.d.ts

@@ -0,0 +1,66 @@
+export interface ComfyUINodeDef {
+    input: {
+        required?: Record<string, NodeInputSpec>;
+        optional?: Record<string, NodeInputSpec>;
+        hidden?: Record<string, NodeInputSpec>;
+    };
+    input_order?: {
+        required?: string[];
+        optional?: string[];
+    };
+    output: string[];
+    output_is_list: boolean[];
+    output_name: string[];
+    name: string;
+    display_name: string;
+    description: string;
+    category: string;
+    output_node: boolean;
+    python_module?: string;
+}
+export type NodeInputSpec = [string | string[], Record<string, unknown>?];
+export type ObjectInfo = Record<string, ComfyUINodeDef>;
+export interface WorkflowNode {
+    class_type: string;
+    inputs: Record<string, unknown>;
+    _meta?: {
+        title?: string;
+    };
+}
+export type WorkflowJSON = Record<string, WorkflowNode>;
+export interface QueueStatus {
+    queue_running: QueueItem[];
+    queue_pending: QueueItem[];
+}
+export type QueueItem = [number, string, Record<string, unknown>, unknown, unknown];
+export interface SystemStats {
+    system: {
+        os: string;
+        python_version: string;
+        embedded_python: boolean;
+    };
+    devices: Array<{
+        name: string;
+        type: string;
+        index: number;
+        vram_total: number;
+        vram_free: number;
+        torch_vram_total: number;
+        torch_vram_free: number;
+    }>;
+}
+export interface JobResult {
+    prompt_id: string;
+    images: Array<{
+        data: string;
+        mime: string;
+    }>;
+    node_outputs: Record<string, unknown>;
+}
+export interface JobProgress {
+    value: number;
+    max: number;
+    node?: string;
+    prompt_id?: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/comfyui/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/comfyui/types.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE;QACL,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACzC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACzC,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;KACxC,CAAC;IACF,WAAW,CAAC,EAAE;QACZ,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;KACrB,CAAC;IACF,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,cAAc,EAAE,OAAO,EAAE,CAAC;IAC1B,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,OAAO,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,MAAM,aAAa,GAAG,CAAC,MAAM,GAAG,MAAM,EAAE,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;AAE1E,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;AAExD,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,KAAK,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CAC5B;AAED,MAAM,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;AAExD,MAAM,WAAW,WAAW;IAC1B,aAAa,EAAE,SAAS,EAAE,CAAC;IAC3B,aAAa,EAAE,SAAS,EAAE,CAAC;CAC5B;AAED,MAAM,MAAM,SAAS,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AAEpF,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE;QACN,EAAE,EAAE,MAAM,CAAC;QACX,cAAc,EAAE,MAAM,CAAC;QACvB,eAAe,EAAE,OAAO,CAAC;KAC1B,CAAC;IACF,OAAO,EAAE,KAAK,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,gBAAgB,EAAE,MAAM,CAAC;QACzB,eAAe,EAAE,MAAM,CAAC;KACzB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,SAAS;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;KACd,CAAC,CAAC;IACH,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACvC;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB"}

package/dist/comfyui/types.js

@@ -0,0 +1,3 @@
+// Extended types for ComfyUI operations
+export {};
+//# sourceMappingURL=types.js.map

package/dist/comfyui/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/comfyui/types.ts"],"names":[],"mappings":"AAAA,wCAAwC"}

package/dist/config.d.ts

@@ -0,0 +1,28 @@
+import { z } from "zod";
+declare const configSchema: z.ZodObject<{
+    comfyuiHost: z.ZodDefault<z.ZodString>;
+    comfyuiPort: z.ZodDefault<z.ZodNumber>;
+    comfyuiPath: z.ZodOptional<z.ZodString>;
+    huggingfaceToken: z.ZodOptional<z.ZodString>;
+    githubToken: z.ZodOptional<z.ZodString>;
+    civitaiApiToken: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    comfyuiHost: string;
+    comfyuiPort: number;
+    comfyuiPath?: string | undefined;
+    huggingfaceToken?: string | undefined;
+    githubToken?: string | undefined;
+    civitaiApiToken?: string | undefined;
+}, {
+    comfyuiHost?: string | undefined;
+    comfyuiPort?: number | undefined;
+    comfyuiPath?: string | undefined;
+    huggingfaceToken?: string | undefined;
+    githubToken?: string | undefined;
+    civitaiApiToken?: string | undefined;
+}>;
+export type Config = z.infer<typeof configSchema>;
+export declare const config: Config;
+export declare function getComfyUIApiHost(): string;
+export {};
+//# sourceMappingURL=config.d.ts.map

package/dist/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAsFxB,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;EAOhB,CAAC;AAEH,MAAM,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAElD,eAAO,MAAM,MAAM,EAAE,MAOnB,CAAC;AAEH,wBAAgB,iBAAiB,IAAI,MAAM,CAE1C"}

package/dist/config.js

@@ -0,0 +1,93 @@
+import { z } from "zod";
+import dotenv from "dotenv";
+import { fileURLToPath } from "url";
+import { dirname, resolve, join } from "path";
+import { existsSync, readdirSync } from "node:fs";
+import { homedir } from "node:os";
+// Resolve .env from the package root, not process.cwd()
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+const packageRoot = resolve(__dirname, "..");
+dotenv.config({ path: resolve(packageRoot, ".env") });
+/**
+ * Auto-detect ComfyUI installation directories.
+ * Checks common locations on macOS, Linux, and Windows.
+ * Returns all found paths sorted by preference.
+ */
+function detectComfyUIPaths() {
+    const home = homedir();
+    const candidates = [];
+    // macOS: ComfyUI Desktop app stores data here
+    candidates.push(join(home, "Documents", "ComfyUI"));
+    // macOS: Application Support
+    candidates.push(join(home, "Library", "Application Support", "ComfyUI"));
+    // Common manual install locations
+    candidates.push(join(home, "ComfyUI"));
+    candidates.push(join(home, "code", "ComfyUI"));
+    candidates.push(join(home, "projects", "ComfyUI"));
+    candidates.push(join(home, "src", "ComfyUI"));
+    // Linux common paths
+    candidates.push("/opt/ComfyUI");
+    candidates.push(join(home, ".local", "share", "ComfyUI"));
+    // Windows common paths (via WSL or native)
+    candidates.push(join(home, "AppData", "Local", "ComfyUI"));
+    candidates.push(join(home, "Desktop", "ComfyUI"));
+    // Also scan ~/Documents for any ComfyUI-named directories
+    const documentsDir = join(home, "Documents");
+    try {
+        if (existsSync(documentsDir)) {
+            const entries = readdirSync(documentsDir, { withFileTypes: true });
+            for (const entry of entries) {
+                if (entry.isDirectory() && entry.name.toLowerCase().includes("comfyui")) {
+                    const fullPath = join(documentsDir, entry.name);
+                    if (!candidates.includes(fullPath)) {
+                        candidates.push(fullPath);
+                    }
+                }
+            }
+        }
+    }
+    catch {
+        // Ignore permission errors
+    }
+    return candidates.filter((p) => existsSync(p));
+}
+/**
+ * Resolve the ComfyUI path, with auto-detection fallback.
+ * Priority: COMFYUI_PATH env var > auto-detected paths.
+ * Logs a warning if multiple installations found.
+ */
+function resolveComfyUIPath(envPath) {
+    if (envPath)
+        return envPath;
+    const detected = detectComfyUIPaths();
+    if (detected.length === 0)
+        return undefined;
+    if (detected.length > 1) {
+        console.error(`[comfyui-mcp] Multiple ComfyUI installations detected:\n` +
+            detected.map((p, i) => `  ${i + 1}. ${p}`).join("\n") +
+            `\nUsing: ${detected[0]}\n` +
+            `Set COMFYUI_PATH to override.`);
+    }
+    return detected[0];
+}
+const configSchema = z.object({
+    comfyuiHost: z.string().default("127.0.0.1"),
+    comfyuiPort: z.coerce.number().int().positive().default(8188),
+    comfyuiPath: z.string().optional(),
+    huggingfaceToken: z.string().optional(),
+    githubToken: z.string().optional(),
+    civitaiApiToken: z.string().optional(),
+});
+export const config = configSchema.parse({
+    comfyuiHost: process.env.COMFYUI_HOST,
+    comfyuiPort: process.env.COMFYUI_PORT,
+    comfyuiPath: resolveComfyUIPath(process.env.COMFYUI_PATH),
+    huggingfaceToken: process.env.HUGGINGFACE_TOKEN,
+    githubToken: process.env.GITHUB_TOKEN,
+    civitaiApiToken: process.env.CIVITAI_API_TOKEN,
+});
+export function getComfyUIApiHost() {
+    return `${config.comfyuiHost}:${config.comfyuiPort}`;
+}
+//# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAClD,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAElC,wDAAwD;AACxD,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AACtC,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;AAC7C,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;AAEtD;;;;GAIG;AACH,SAAS,kBAAkB;IACzB,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,8CAA8C;IAC9C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC;IAEpD,6BAA6B;IAC7B,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,qBAAqB,EAAE,SAAS,CAAC,CAAC,CAAC;IAEzE,kCAAkC;IAClC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC;IACvC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC;IAC/C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC;IACnD,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;IAE9C,qBAAqB;IACrB,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAChC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;IAE1D,2CAA2C;IAC3C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;IAC3D,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;IAElD,0DAA0D;IAC1D,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAC7C,IAAI,CAAC;QACH,IAAI,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAG,WAAW,CAAC,YAAY,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;YACnE,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;oBACxE,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;oBAChD,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;wBACnC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBAC5B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,2BAA2B;IAC7B,CAAC;IAED,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;AACjD,CAAC;AAED;;;;GAIG;AACH,SAAS,kBAAkB,CAAC,OAAgB;IAC1C,IAAI,OAAO;QAAE,OAAO,OAAO,CAAC;IAE5B,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;IACtC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAE5C,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,KAAK,CACX,0DAA0D;YACxD,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;YACrD,YAAY,QAAQ,CAAC,CAAC,CAAC,IAAI;YAC3B,+BAA+B,CAClC,CAAC;IACJ,CAAC;IAED,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC;AACrB,CAAC;AAED,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5B,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC;IAC5C,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;IAC7D,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACvC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACvC,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,MAAM,GAAW,YAAY,CAAC,KAAK,CAAC;IAC/C,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,YAAY;IACrC,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,YAAY;IACrC,WAAW,EAAE,kBAAkB,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IACzD,gBAAgB,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;IAC/C,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,YAAY;IACrC,eAAe,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;CAC/C,CAAC,CAAC;AAEH,MAAM,UAAU,iBAAiB;IAC/B,OAAO,GAAG,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;AACvD,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":""}

package/dist/index.js

@@ -0,0 +1,22 @@
+#!/usr/bin/env node
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { registerAllTools } from "./tools/index.js";
+import { logger } from "./utils/logger.js";
+const server = new McpServer({
+    name: "comfyui-mcp",
+    version: "0.1.0",
+}, {
+    capabilities: { tools: {} },
+});
+registerAllTools(server);
+async function main() {
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    logger.info("ComfyUI MCP server running on stdio");
+}
+main().catch((err) => {
+    logger.error("Fatal error", err);
+    process.exit(1);
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAE3C,MAAM,MAAM,GAAG,IAAI,SAAS,CAC1B;IACE,IAAI,EAAE,aAAa;IACnB,OAAO,EAAE,OAAO;CACjB,EACD;IACE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;CAC5B,CACF,CAAC;AAEF,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAEzB,KAAK,UAAU,IAAI;IACjB,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,MAAM,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;AACrD,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnB,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;IACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/services/mermaid-converter.d.ts

@@ -0,0 +1,8 @@
+import type { WorkflowJSON } from "../comfyui/types.js";
+interface MermaidOptions {
+    showValues?: boolean;
+    direction?: "LR" | "TB";
+}
+export declare function convertToMermaid(workflow: WorkflowJSON, options?: MermaidOptions): string;
+export {};
+//# sourceMappingURL=mermaid-converter.d.ts.map

package/dist/services/mermaid-converter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mermaid-converter.d.ts","sourceRoot":"","sources":["../../src/services/mermaid-converter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAgB,MAAM,qBAAqB,CAAC;AAUtE,UAAU,cAAc;IACtB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;CACzB;AAqMD,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,YAAY,EACtB,OAAO,GAAE,cAAmB,GAC3B,MAAM,CA0ER"}