comfyui-mcp 0.5.0 → 0.6.0

package/dist/services/api-nodes.js

@@ -0,0 +1,205 @@
+import { getObjectInfo } from "../comfyui/client.js";
+import { enqueueWorkflow } from "./workflow-executor.js";
+import { config } from "../config.js";
+import { ValidationError } from "../utils/errors.js";
+/**
+ * API / partner-node support, mirroring `comfy-cli generate` (list / schema / run).
+ *
+ * MECHANISM & ASSUMPTIONS (confirmed against ComfyUI + comfy-cli source):
+ * - These nodes run entirely on the *connected* ComfyUI server via its normal
+ *   HTTP API (/object_info + /prompt), so this works against remote servers too.
+ *   We reuse the existing client (getObjectInfo) and enqueue helper rather than
+ *   re-implementing anything.
+ * - ComfyUI marks hosted partner/API nodes in /object_info with `api_node: true`
+ *   (server.py emits this from the node class's `API_NODE` attribute). Their
+ *   `category` also conventionally starts with "api node/" (e.g.
+ *   "api node/image/BFL", "api node/video/Kling"). We treat EITHER signal as a
+ *   match so older/newer ComfyUI builds both work.
+ *   Ref: comfyanonymous/ComfyUI server.py node_info() and comfy_api_nodes/*.py.
+ * - AUTH: API nodes require a Comfy account / API key, but that credential is
+ *   supplied to the *ComfyUI server* out-of-band — it is injected by the server
+ *   into the node's HIDDEN inputs (`auth_token_comfy_org` / `api_key_comfy_org`)
+ *   from the logged-in session. The MCP client does NOT (and should not) place
+ *   the key in the workflow JSON. If the server isn't authenticated, the job
+ *   will be enqueued but fail server-side; we surface that as guidance rather
+ *   than trying to pass credentials ourselves. (Lower-confidence area — see
+ *   docs.comfy.org/tutorials/api-nodes.)
+ */
+const API_CATEGORY_PREFIX = "api node";
+const defaultDeps = {
+    getObjectInfo,
+    enqueue: enqueueWorkflow,
+};
+/** True if a node definition is a hosted partner/API node. */
+export function isApiNode(def) {
+    if (def.api_node === true)
+        return true;
+    const category = (def.category ?? "").toLowerCase();
+    return category === API_CATEGORY_PREFIX || category.startsWith(`${API_CATEGORY_PREFIX}/`);
+}
+/**
+ * Discover hosted partner/API nodes available on the connected ComfyUI.
+ * Optionally narrow by a case-insensitive substring matched against the
+ * class_type, display name, or category (e.g. "image", "kling", "video").
+ */
+export async function listApiNodes(filter, deps = defaultDeps) {
+    const objectInfo = await deps.getObjectInfo();
+    const needle = filter?.trim().toLowerCase();
+    const results = [];
+    for (const [classType, def] of Object.entries(objectInfo)) {
+        if (!def || !isApiNode(def))
+            continue;
+        const summary = {
+            class_type: classType,
+            display_name: def.display_name || classType,
+            category: def.category ?? "",
+            description: def.description ?? "",
+            output: Array.isArray(def.output) ? def.output : [],
+            deprecated: def.deprecated === true,
+            experimental: def.experimental === true,
+        };
+        if (needle) {
+            const haystack = `${classType} ${summary.display_name} ${summary.category}`.toLowerCase();
+            if (!haystack.includes(needle))
+                continue;
+        }
+        results.push(summary);
+    }
+    results.sort((a, b) => a.class_type.localeCompare(b.class_type));
+    return results;
+}
+function describeInputs(specs, required) {
+    if (!specs)
+        return [];
+    return Object.entries(specs).map(([name, spec]) => {
+        // Spec shape: [type, config?] where type is a string (e.g. "STRING") or a
+        // string[] of enum options (e.g. ["fast", "quality"]).
+        const type = Array.isArray(spec) ? spec[0] : spec;
+        const config = (Array.isArray(spec) ? spec[1] : undefined) ?? {};
+        return {
+            name,
+            type: type,
+            required,
+            config: config,
+        };
+    });
+}
+/** Return the input schema for a given API node (from its /object_info entry). */
+export async function getApiNodeSchema(classType, deps = defaultDeps) {
+    const objectInfo = await deps.getObjectInfo();
+    const def = objectInfo[classType];
+    if (!def) {
+        throw new ValidationError(`Node "${classType}" was not found on the connected ComfyUI. ` +
+            `Use list_api_nodes to see available API nodes.`);
+    }
+    if (!isApiNode(def)) {
+        throw new ValidationError(`Node "${classType}" exists but is not an API/partner node ` +
+            `(category "${def.category ?? ""}"). Use list_api_nodes to find API nodes.`);
+    }
+    return {
+        class_type: classType,
+        display_name: def.display_name || classType,
+        category: def.category ?? "",
+        description: def.description ?? "",
+        is_api_node: true,
+        is_output_node: def.output_node === true,
+        inputs: [
+            ...describeInputs(def.input?.required, true),
+            ...describeInputs(def.input?.optional, false),
+        ],
+        hidden_inputs: def.input?.hidden ? Object.keys(def.input.hidden) : [],
+        output: Array.isArray(def.output) ? def.output : [],
+        output_name: Array.isArray(def.output_name) ? def.output_name : [],
+    };
+}
+/**
+ * Build a minimal single-node workflow that runs a chosen API node with the
+ * provided inputs and enqueue it. Returns the prompt_id (reuses the existing
+ * enqueue path). The API node is typically an output node, so it can stand
+ * alone as a complete graph.
+ */
+export async function generateWithApiNode(args, deps = defaultDeps) {
+    const schema = await getApiNodeSchema(args.class_type, deps);
+    const notes = [];
+    const provided = args.inputs ?? {};
+    const knownNames = new Set(schema.inputs.map((i) => i.name));
+    // Warn about inputs that aren't part of the node's visible schema (typos,
+    // or hidden auth fields the server should fill itself).
+    for (const key of Object.keys(provided)) {
+        if (!knownNames.has(key)) {
+            if (schema.hidden_inputs.includes(key)) {
+                notes.push(`Ignoring "${key}": it is a hidden input filled by the ComfyUI server, not the client.`);
+            }
+            else {
+                notes.push(`Unknown input "${key}" for ${args.class_type} (passing through anyway).`);
+            }
+        }
+    }
+    // Check required inputs are present (skip enum/combo selectors that ComfyUI
+    // can default, but still flag genuinely missing scalars).
+    const missingRequired = schema.inputs
+        .filter((i) => i.required && !(i.name in provided))
+        .map((i) => i.name);
+    if (missingRequired.length > 0) {
+        notes.push(`Missing required input(s): ${missingRequired.join(", ")}. ` +
+            `The server may reject the job. Use get_api_node_schema for details.`);
+    }
+    // Drop any hidden-input values the caller mistakenly supplied — the server
+    // injects auth credentials itself.
+    const inputs = {};
+    for (const [key, value] of Object.entries(provided)) {
+        if (schema.hidden_inputs.includes(key))
+            continue;
+        inputs[key] = value;
+    }
+    const workflow = {
+        "1": {
+            class_type: args.class_type,
+            inputs,
+            _meta: { title: schema.display_name },
+        },
+    };
+    // ComfyUI only executes graphs that reach a terminal OUTPUT_NODE; a bare
+    // non-output API node fails validation with "prompt_no_outputs". If the API
+    // node isn't itself an output node, wire its IMAGE output into a SaveImage.
+    if (!schema.is_output_node) {
+        const imageIdx = schema.output.findIndex((o) => String(o).toUpperCase() === "IMAGE");
+        if (imageIdx >= 0) {
+            workflow["2"] = {
+                class_type: "SaveImage",
+                inputs: { images: ["1", imageIdx], filename_prefix: "ComfyUI" },
+                _meta: { title: "Save Image" },
+            };
+            notes.push("Added a SaveImage output node — ComfyUI requires a terminal output node " +
+                "for the prompt to execute.");
+        }
+        else {
+            notes.push(`"${args.class_type}" is not an output node and has no IMAGE output, so no ` +
+                "output node was auto-added; the prompt may fail with 'prompt_no_outputs'. " +
+                "Wire a terminal output node yourself if needed.");
+        }
+    }
+    // comfy.org API-node credentials travel in /prompt extra_data (the server
+    // injects them into the node's hidden inputs). Supply the configured key so
+    // API nodes work even when the ComfyUI server isn't logged in itself.
+    const extraData = {};
+    if (config.comfyApiKey) {
+        extraData.api_key_comfy_org = config.comfyApiKey;
+        notes.push("Attached COMFY_API_KEY to the request (extra_data.api_key_comfy_org).");
+    }
+    else {
+        notes.push("No COMFY_API_KEY configured — relying on the ComfyUI server's own logged-in " +
+            "comfy.org session for API-node auth. Set COMFY_API_KEY if the server isn't logged in.");
+    }
+    const result = await deps.enqueue(workflow, {
+        disable_random_seed: args.disable_random_seed,
+        extra_data: Object.keys(extraData).length > 0 ? extraData : undefined,
+    });
+    return {
+        prompt_id: result.prompt_id,
+        queue_remaining: result.queue_remaining,
+        workflow,
+        notes,
+    };
+}
+//# sourceMappingURL=api-nodes.js.map

package/dist/services/api-nodes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api-nodes.js","sourceRoot":"","sources":["../../src/services/api-nodes.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD;;;;;;;;;;;;;;;;;;;;;;GAsBG;AAEH,MAAM,mBAAmB,GAAG,UAAU,CAAC;AAcvC,MAAM,WAAW,GAAiB;IAChC,aAAa;IACb,OAAO,EAAE,eAAe;CACzB,CAAC;AAEF,8DAA8D;AAC9D,MAAM,UAAU,SAAS,CAAC,GAAmB;IAC3C,IAAI,GAAG,CAAC,QAAQ,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IACvC,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;IACpD,OAAO,QAAQ,KAAK,mBAAmB,IAAI,QAAQ,CAAC,UAAU,CAAC,GAAG,mBAAmB,GAAG,CAAC,CAAC;AAC5F,CAAC;AAYD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,MAAe,EACf,OAAqB,WAAW;IAEhC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;IAC9C,MAAM,MAAM,GAAG,MAAM,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAE5C,MAAM,OAAO,GAAqB,EAAE,CAAC;IACrC,KAAK,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC1D,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;YAAE,SAAS;QAEtC,MAAM,OAAO,GAAmB;YAC9B,UAAU,EAAE,SAAS;YACrB,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,SAAS;YAC3C,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE;YAC5B,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE;YAClC,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE;YACnD,UAAU,EAAE,GAAG,CAAC,UAAU,KAAK,IAAI;YACnC,YAAY,EAAE,GAAG,CAAC,YAAY,KAAK,IAAI;SACxC,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,QAAQ,GAAG,GAAG,SAAS,IAAI,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC,WAAW,EAAE,CAAC;YAC1F,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAAE,SAAS;QAC3C,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;IAED,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;IACjE,OAAO,OAAO,CAAC;AACjB,CAAC;AA6BD,SAAS,cAAc,CACrB,KAAgD,EAChD,QAAiB;IAEjB,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IACtB,OAAO,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QAChD,0EAA0E;QAC1E,uDAAuD;QACvD,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,IAA0B,CAAC;QACzE,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QACjE,OAAO;YACL,IAAI;YACJ,IAAI,EAAE,IAAyB;YAC/B,QAAQ;YACR,MAAM,EAAE,MAAiC;SAC1C,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,kFAAkF;AAClF,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,SAAiB,EACjB,OAAqB,WAAW;IAEhC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;IAC9C,MAAM,GAAG,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IAElC,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,eAAe,CACvB,SAAS,SAAS,4CAA4C;YAC5D,gDAAgD,CACnD,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;QACpB,MAAM,IAAI,eAAe,CACvB,SAAS,SAAS,0CAA0C;YAC1D,cAAc,GAAG,CAAC,QAAQ,IAAI,EAAE,2CAA2C,CAC9E,CAAC;IACJ,CAAC;IAED,OAAO;QACL,UAAU,EAAE,SAAS;QACrB,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,SAAS;QAC3C,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE;QAC5B,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE;QAClC,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,GAAG,CAAC,WAAW,KAAK,IAAI;QACxC,MAAM,EAAE;YACN,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC;YAC5C,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC;SAC9C;QACD,aAAa,EAAE,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;QACrE,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE;QACnD,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE;KACnE,CAAC;AACJ,CAAC;AAiBD;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,IAA6B,EAC7B,OAAqB,WAAW;IAEhC,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IAC7D,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;IACnC,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAE7D,0EAA0E;IAC1E,wDAAwD;IACxD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACzB,IAAI,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACvC,KAAK,CAAC,IAAI,CACR,aAAa,GAAG,uEAAuE,CACxF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,IAAI,CAAC,kBAAkB,GAAG,SAAS,IAAI,CAAC,UAAU,4BAA4B,CAAC,CAAC;YACxF,CAAC;QACH,CAAC;IACH,CAAC;IAED,4EAA4E;IAC5E,0DAA0D;IAC1D,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM;SAClC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,QAAQ,CAAC,CAAC;SAClD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtB,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,KAAK,CAAC,IAAI,CACR,8BAA8B,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI;YAC1D,qEAAqE,CACxE,CAAC;IACJ,CAAC;IAED,2EAA2E;IAC3E,mCAAmC;IACnC,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpD,IAAI,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,SAAS;QACjD,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IACtB,CAAC;IAED,MAAM,QAAQ,GAAiB;QAC7B,GAAG,EAAE;YACH,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM;YACN,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,YAAY,EAAE;SACtC;KACF,CAAC;IAEF,yEAAyE;IACzE,4EAA4E;IAC5E,4EAA4E;IAC5E,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,OAAO,CAC3C,CAAC;QACF,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;YAClB,QAAQ,CAAC,GAAG,CAAC,GAAG;gBACd,UAAU,EAAE,WAAW;gBACvB,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,eAAe,EAAE,SAAS,EAAE;gBAC/D,KAAK,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;aAC/B,CAAC;YACF,KAAK,CAAC,IAAI,CACR,0EAA0E;gBACxE,4BAA4B,CAC/B,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,IAAI,CACR,IAAI,IAAI,CAAC,UAAU,yDAAyD;gBAC1E,4EAA4E;gBAC5E,iDAAiD,CACpD,CAAC;QACJ,CAAC;IACH,CAAC;IAED,0EAA0E;IAC1E,4EAA4E;IAC5E,sEAAsE;IACtE,MAAM,SAAS,GAA4B,EAAE,CAAC;IAC9C,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,SAAS,CAAC,iBAAiB,GAAG,MAAM,CAAC,WAAW,CAAC;QACjD,KAAK,CAAC,IAAI,CACR,uEAAuE,CACxE,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,IAAI,CACR,8EAA8E;YAC5E,uFAAuF,CAC1F,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;QAC1C,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;QAC7C,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;KACtE,CAAC,CAAC;IAEH,OAAO;QACL,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,eAAe,EAAE,MAAM,CAAC,eAAe;QACvC,QAAQ;QACR,KAAK;KACN,CAAC;AACJ,CAAC"}

package/dist/services/civitai-resolver.d.ts

@@ -0,0 +1,27 @@
+export interface CivitaiResolved {
+    /**
+     * Direct download URL. No credentials are embedded — `downloadModel` attaches
+     * the CivitAI token as an `Authorization` request header, so the token never
+     * leaks into logs, error messages, or redirect URLs.
+     */
+    downloadUrl: string;
+    /** Suggested filename from CivitAI metadata, when available. */
+    filename?: string;
+    /** Resolved model-version id. */
+    versionId: number;
+    /** Model name, when resolvable. */
+    modelName?: string;
+}
+/**
+ * Resolve a CivitAI model-version id directly to a download URL.
+ * Uses GET /api/v1/model-versions/{id}.
+ */
+export declare function resolveCivitaiModelVersion(versionId: number): Promise<CivitaiResolved>;
+/**
+ * Resolve a CivitAI model id to a download URL.
+ * Uses GET /api/v1/models/{id} and picks a model version.
+ * If `versionId` is supplied, that specific version is used; otherwise the
+ * latest (first listed) version is chosen.
+ */
+export declare function resolveCivitaiModel(modelId: number, versionId?: number): Promise<CivitaiResolved>;
+//# sourceMappingURL=civitai-resolver.d.ts.map

package/dist/services/civitai-resolver.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"civitai-resolver.d.ts","sourceRoot":"","sources":["../../src/services/civitai-resolver.ts"],"names":[],"mappings":"AA+BA,MAAM,WAAW,eAAe;IAC9B;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB,gEAAgE;IAChE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,iCAAiC;IACjC,SAAS,EAAE,MAAM,CAAC;IAClB,mCAAmC;IACnC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAwDD;;;GAGG;AACH,wBAAsB,0BAA0B,CAC9C,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,eAAe,CAAC,CAK1B;AAED;;;;;GAKG;AACH,wBAAsB,mBAAmB,CACvC,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,eAAe,CAAC,CAyB1B"}

package/dist/services/civitai-resolver.js

@@ -0,0 +1,82 @@
+import { config } from "../config.js";
+import { ModelError, ValidationError } from "../utils/errors.js";
+import { logger } from "../utils/logger.js";
+const CIVITAI_API_BASE = "https://civitai.com/api/v1";
+function authHeaders() {
+    const headers = {};
+    if (config.civitaiApiToken) {
+        headers["Authorization"] = `Bearer ${config.civitaiApiToken}`;
+    }
+    return headers;
+}
+async function civitaiGet(path) {
+    const url = `${CIVITAI_API_BASE}${path}`;
+    logger.debug("CivitAI API request", { url });
+    const res = await fetch(url, { headers: authHeaders() });
+    if (res.status === 404) {
+        throw new ModelError(`CivitAI resource not found: ${path}`, {
+            url,
+            status: 404,
+        });
+    }
+    if (!res.ok) {
+        const body = await res.text().catch(() => "");
+        throw new ModelError(`CivitAI API ${res.status}: ${res.statusText}`, {
+            url,
+            status: res.status,
+            body,
+        });
+    }
+    return (await res.json());
+}
+/** Pick the best file from a version's file list: primary first, else the first. */
+function pickFile(version) {
+    const files = version.files ?? [];
+    return files.find((f) => f.primary) ?? files[0];
+}
+function resolveFromVersion(version, modelName) {
+    const file = pickFile(version);
+    const downloadUrl = file?.downloadUrl ??
+        version.downloadUrl ??
+        `https://civitai.com/api/download/models/${version.id}`;
+    return {
+        downloadUrl,
+        filename: file?.name,
+        versionId: version.id,
+        modelName,
+    };
+}
+/**
+ * Resolve a CivitAI model-version id directly to a download URL.
+ * Uses GET /api/v1/model-versions/{id}.
+ */
+export async function resolveCivitaiModelVersion(versionId) {
+    const version = await civitaiGet(`/model-versions/${versionId}`);
+    return resolveFromVersion(version);
+}
+/**
+ * Resolve a CivitAI model id to a download URL.
+ * Uses GET /api/v1/models/{id} and picks a model version.
+ * If `versionId` is supplied, that specific version is used; otherwise the
+ * latest (first listed) version is chosen.
+ */
+export async function resolveCivitaiModel(modelId, versionId) {
+    const model = await civitaiGet(`/models/${modelId}`);
+    const versions = model.modelVersions ?? [];
+    if (versions.length === 0) {
+        throw new ModelError(`CivitAI model ${modelId} has no downloadable versions.`, { modelId });
+    }
+    let version;
+    if (versionId !== undefined) {
+        version = versions.find((v) => v.id === versionId);
+        if (!version) {
+            throw new ValidationError(`Model ${modelId} has no version with id ${versionId}. ` +
+                `Available versions: ${versions.map((v) => v.id).join(", ")}.`);
+        }
+    }
+    else {
+        version = versions[0];
+    }
+    return resolveFromVersion(version, model.name);
+}
+//# sourceMappingURL=civitai-resolver.js.map

package/dist/services/civitai-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"civitai-resolver.js","sourceRoot":"","sources":["../../src/services/civitai-resolver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,MAAM,gBAAgB,GAAG,4BAA4B,CAAC;AA0CtD,SAAS,WAAW;IAClB,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;QAC3B,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,MAAM,CAAC,eAAe,EAAE,CAAC;IAChE,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,UAAU,CAAI,IAAY;IACvC,MAAM,GAAG,GAAG,GAAG,gBAAgB,GAAG,IAAI,EAAE,CAAC;IACzC,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;IAE7C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;IACzD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACvB,MAAM,IAAI,UAAU,CAAC,+BAA+B,IAAI,EAAE,EAAE;YAC1D,GAAG;YACH,MAAM,EAAE,GAAG;SACZ,CAAC,CAAC;IACL,CAAC;IACD,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9C,MAAM,IAAI,UAAU,CAAC,eAAe,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,EAAE,EAAE;YACnE,GAAG;YACH,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,IAAI;SACL,CAAC,CAAC;IACL,CAAC;IACD,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAM,CAAC;AACjC,CAAC;AAED,oFAAoF;AACpF,SAAS,QAAQ,CAAC,OAA4B;IAC5C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;IAClC,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,SAAS,kBAAkB,CACzB,OAA4B,EAC5B,SAAkB;IAElB,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC/B,MAAM,WAAW,GACf,IAAI,EAAE,WAAW;QACjB,OAAO,CAAC,WAAW;QACnB,2CAA2C,OAAO,CAAC,EAAE,EAAE,CAAC;IAE1D,OAAO;QACL,WAAW;QACX,QAAQ,EAAE,IAAI,EAAE,IAAI;QACpB,SAAS,EAAE,OAAO,CAAC,EAAE;QACrB,SAAS;KACV,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,SAAiB;IAEjB,MAAM,OAAO,GAAG,MAAM,UAAU,CAC9B,mBAAmB,SAAS,EAAE,CAC/B,CAAC;IACF,OAAO,kBAAkB,CAAC,OAAO,CAAC,CAAC;AACrC,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,OAAe,EACf,SAAkB;IAElB,MAAM,KAAK,GAAG,MAAM,UAAU,CAAe,WAAW,OAAO,EAAE,CAAC,CAAC;IACnE,MAAM,QAAQ,GAAG,KAAK,CAAC,aAAa,IAAI,EAAE,CAAC;IAE3C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,UAAU,CAClB,iBAAiB,OAAO,gCAAgC,EACxD,EAAE,OAAO,EAAE,CACZ,CAAC;IACJ,CAAC;IAED,IAAI,OAAwC,CAAC;IAC7C,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC5B,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,eAAe,CACvB,SAAS,OAAO,2BAA2B,SAAS,IAAI;gBACtD,uBAAuB,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CACjE,CAAC;QACJ,CAAC;IACH,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC;IAED,OAAO,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;AACjD,CAAC"}

package/dist/services/install-comfyui.d.ts

@@ -0,0 +1,89 @@
+export declare const COMFYUI_REPO_URL = "https://github.com/comfyanonymous/ComfyUI";
+export declare const COMFYUI_MANAGER_REPO_URL = "https://github.com/Comfy-Org/ComfyUI-Manager";
+/** Sub-path under the ComfyUI clone where the Manager lives (clone fallback). */
+export declare const MANAGER_SUBDIR: string;
+export interface InstallComfyUIOptions {
+    /** Target workspace directory to install ComfyUI into. */
+    targetPath: string;
+    /** Skip cloning ComfyUI-Manager. Default false (Manager is installed). */
+    skipManager?: boolean;
+    /** Prefer `uv pip` over plain `pip` when uv is available. Default false. */
+    useUv?: boolean;
+    /**
+     * ComfyUI git ref to check out (tag, branch, or commit). When omitted the
+     * default branch HEAD is used.
+     */
+    version?: string;
+}
+export interface StepResult {
+    step: string;
+    command: string;
+    ok: boolean;
+    output?: string;
+}
+export interface InstallComfyUIResult {
+    installed: boolean;
+    targetPath: string;
+    /** Path to the workspace virtualenv python that deps were installed into. */
+    venvPath: string;
+    comfyuiUrl: string;
+    managerUrl: string | null;
+    managerInstalled: boolean;
+    /** How the Manager was installed: pip "requirements", "git-clone", or null. */
+    managerVia: "requirements" | "git-clone" | null;
+    version: string | null;
+    pythonInstaller: "uv" | "pip";
+    steps: StepResult[];
+    message: string;
+}
+export interface InstallDeps {
+    /** Run a command, throwing on non-zero exit. Returns combined stdout. */
+    run: (cmd: string, args: string[], cwd?: string) => string;
+    /** Detect whether a CLI tool is on PATH. */
+    hasCommand: (cmd: string) => boolean;
+    existsSync: (p: string) => boolean;
+    /** True if the path exists AND contains at least one entry. */
+    isNonEmptyDir: (p: string) => boolean;
+    mkdirp: (p: string) => void;
+}
+/** Build the `git clone` argv (without the leading `git`). */
+export declare function buildCloneArgs(url: string, dest: string, version?: string): string[];
+/** Path to the workspace venv's python interpreter. */
+export declare function venvPythonPath(targetPath: string): string;
+/** Build the argv that creates the workspace venv (`<target>/.venv`). */
+export declare function buildVenvArgs(installer: "uv" | "pip", targetPath: string): {
+    cmd: string;
+    args: string[];
+};
+/**
+ * Build the pip/uv install argv for a requirements file, ALWAYS targeting the
+ * workspace venv's interpreter (`venvPython`) — never the Python running this
+ * MCP server. uv targets the venv explicitly via `--python`.
+ */
+export declare function buildPipInstallArgs(installer: "uv" | "pip", venvPython: string, requirementsFile: string): {
+    cmd: string;
+    args: string[];
+};
+/**
+ * Validate the requested ComfyUI version, mirroring comfy-cli's
+ * `validate_version`: only "nightly", "latest", or a full semantic version
+ * (optionally v-prefixed). Raw git refs / branch names are rejected.
+ */
+export declare function validateVersion(version: string): {
+    kind: "nightly";
+} | {
+    kind: "latest";
+} | {
+    kind: "semver";
+    tag: string;
+};
+/**
+ * Mirrors `comfy-cli install`: clones ComfyUI (and optionally ComfyUI-Manager)
+ * into a target workspace, then installs Python requirements via pip or uv.
+ *
+ * This is a LOCAL, subprocess-only operation. It never touches a remote
+ * ComfyUI server and ignores `config.comfyuiPath` in favour of the explicit
+ * `targetPath`.
+ */
+export declare function installComfyUI(options: InstallComfyUIOptions, deps?: InstallDeps): InstallComfyUIResult;
+//# sourceMappingURL=install-comfyui.d.ts.map

package/dist/services/install-comfyui.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"install-comfyui.d.ts","sourceRoot":"","sources":["../../src/services/install-comfyui.ts"],"names":[],"mappings":"AAoBA,eAAO,MAAM,gBAAgB,8CAA8C,CAAC;AAC5E,eAAO,MAAM,wBAAwB,iDACW,CAAC;AACjD,iFAAiF;AACjF,eAAO,MAAM,cAAc,QAA0C,CAAC;AAQtE,MAAM,WAAW,qBAAqB;IACpC,0DAA0D;IAC1D,UAAU,EAAE,MAAM,CAAC;IACnB,0EAA0E;IAC1E,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,4EAA4E;IAC5E,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,EAAE,EAAE,OAAO,CAAC;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,6EAA6E;IAC7E,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,gBAAgB,EAAE,OAAO,CAAC;IAC1B,+EAA+E;IAC/E,UAAU,EAAE,cAAc,GAAG,WAAW,GAAG,IAAI,CAAC;IAChD,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,eAAe,EAAE,IAAI,GAAG,KAAK,CAAC;IAC9B,KAAK,EAAE,UAAU,EAAE,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAMD,MAAM,WAAW,WAAW;IAC1B,yEAAyE;IACzE,GAAG,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,GAAG,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC;IAC3D,4CAA4C;IAC5C,UAAU,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC;IACrC,UAAU,EAAE,CAAC,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC;IACnC,+DAA+D;IAC/D,aAAa,EAAE,CAAC,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC;IACtC,MAAM,EAAE,CAAC,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;CAC7B;AAoED,8DAA8D;AAC9D,wBAAgB,cAAc,CAC5B,GAAG,EAAE,MAAM,EACX,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,GACf,MAAM,EAAE,CASV;AAED,uDAAuD;AACvD,wBAAgB,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAIzD;AAED,yEAAyE;AACzE,wBAAgB,aAAa,CAC3B,SAAS,EAAE,IAAI,GAAG,KAAK,EACvB,UAAU,EAAE,MAAM,GACjB;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,EAAE,CAAA;CAAE,CAMjC;AAED;;;;GAIG;AACH,wBAAgB,mBAAmB,CACjC,SAAS,EAAE,IAAI,GAAG,KAAK,EACvB,UAAU,EAAE,MAAM,EAClB,gBAAgB,EAAE,MAAM,GACvB;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,EAAE,CAAA;CAAE,CAWjC;AAED;;;;GAIG;AACH,wBAAgB,eAAe,CAC7B,OAAO,EAAE,MAAM,GACd;IAAE,IAAI,EAAE,SAAS,CAAA;CAAE,GAAG;IAAE,IAAI,EAAE,QAAQ,CAAA;CAAE,GAAG;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAAE,CAa5E;AAMD;;;;;;;GAOG;AACH,wBAAgB,cAAc,CAC5B,OAAO,EAAE,qBAAqB,EAC9B,IAAI,GAAE,WAAyB,GAC9B,oBAAoB,CA+JtB"}