comfyui-mcp 0.4.1 → 0.6.0

package/dist/services/api-nodes.d.ts

@@ -0,0 +1,77 @@
+import type { ObjectInfo, ComfyUINodeDef, WorkflowJSON } from "../comfyui/types.js";
+/** Injectable dependencies so the logic is unit-testable without a live server. */
+export interface ApiNodesDeps {
+    getObjectInfo: () => Promise<ObjectInfo>;
+    enqueue: (workflow: WorkflowJSON, options?: {
+        disable_random_seed?: boolean;
+        extra_data?: Record<string, unknown>;
+    }) => Promise<{
+        prompt_id: string;
+        queue_remaining?: number;
+    }>;
+}
+/** True if a node definition is a hosted partner/API node. */
+export declare function isApiNode(def: ComfyUINodeDef): boolean;
+export interface ApiNodeSummary {
+    class_type: string;
+    display_name: string;
+    category: string;
+    description: string;
+    output: string[];
+    deprecated: boolean;
+    experimental: boolean;
+}
+/**
+ * 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 declare function listApiNodes(filter?: string, deps?: ApiNodesDeps): Promise<ApiNodeSummary[]>;
+export interface ApiNodeInputDescriptor {
+    name: string;
+    type: string | string[];
+    required: boolean;
+    /** Extra config from /object_info (default, min, max, options, tooltip, ...). */
+    config: Record<string, unknown>;
+}
+export interface ApiNodeSchema {
+    class_type: string;
+    display_name: string;
+    category: string;
+    description: string;
+    is_api_node: boolean;
+    /** Whether the node is itself a terminal OUTPUT_NODE (drives execution). */
+    is_output_node: boolean;
+    /** Visible inputs the caller can/should supply. */
+    inputs: ApiNodeInputDescriptor[];
+    /**
+     * Hidden inputs (e.g. auth_token_comfy_org). The ComfyUI server fills these
+     * from the logged-in session — callers should NOT supply them.
+     */
+    hidden_inputs: string[];
+    output: string[];
+    output_name: string[];
+}
+/** Return the input schema for a given API node (from its /object_info entry). */
+export declare function getApiNodeSchema(classType: string, deps?: ApiNodesDeps): Promise<ApiNodeSchema>;
+export interface GenerateWithApiNodeArgs {
+    class_type: string;
+    inputs: Record<string, unknown>;
+    disable_random_seed?: boolean;
+}
+export interface GenerateWithApiNodeResult {
+    prompt_id: string;
+    queue_remaining?: number;
+    /** The minimal single-node workflow that was enqueued. */
+    workflow: WorkflowJSON;
+    /** Non-fatal guidance (e.g. unknown inputs, auth reminder). */
+    notes: string[];
+}
+/**
+ * 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 declare function generateWithApiNode(args: GenerateWithApiNodeArgs, deps?: ApiNodesDeps): Promise<GenerateWithApiNodeResult>;
+//# sourceMappingURL=api-nodes.d.ts.map

package/dist/services/api-nodes.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"api-nodes.d.ts","sourceRoot":"","sources":["../../src/services/api-nodes.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,UAAU,EACV,cAAc,EAEd,YAAY,EACb,MAAM,qBAAqB,CAAC;AAgC7B,mFAAmF;AACnF,MAAM,WAAW,YAAY;IAC3B,aAAa,EAAE,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC;IACzC,OAAO,EAAE,CACP,QAAQ,EAAE,YAAY,EACtB,OAAO,CAAC,EAAE;QACR,mBAAmB,CAAC,EAAE,OAAO,CAAC;QAC9B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACtC,KACE,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,eAAe,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAC/D;AAOD,8DAA8D;AAC9D,wBAAgB,SAAS,CAAC,GAAG,EAAE,cAAc,GAAG,OAAO,CAItD;AAED,MAAM,WAAW,cAAc;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,UAAU,EAAE,OAAO,CAAC;IACpB,YAAY,EAAE,OAAO,CAAC;CACvB;AAED;;;;GAIG;AACH,wBAAsB,YAAY,CAChC,MAAM,CAAC,EAAE,MAAM,EACf,IAAI,GAAE,YAA0B,GAC/B,OAAO,CAAC,cAAc,EAAE,CAAC,CA4B3B;AAED,MAAM,WAAW,sBAAsB;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IACxB,QAAQ,EAAE,OAAO,CAAC;IAClB,iFAAiF;IACjF,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACjC;AAED,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC;IACrB,4EAA4E;IAC5E,cAAc,EAAE,OAAO,CAAC;IACxB,mDAAmD;IACnD,MAAM,EAAE,sBAAsB,EAAE,CAAC;IACjC;;;OAGG;IACH,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB;AAqBD,kFAAkF;AAClF,wBAAsB,gBAAgB,CACpC,SAAS,EAAE,MAAM,EACjB,IAAI,GAAE,YAA0B,GAC/B,OAAO,CAAC,aAAa,CAAC,CAgCxB;AAED,MAAM,WAAW,uBAAuB;IACtC,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAED,MAAM,WAAW,yBAAyB;IACxC,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,0DAA0D;IAC1D,QAAQ,EAAE,YAAY,CAAC;IACvB,+DAA+D;IAC/D,KAAK,EAAE,MAAM,EAAE,CAAC;CACjB;AAED;;;;;GAKG;AACH,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,uBAAuB,EAC7B,IAAI,GAAE,YAA0B,GAC/B,OAAO,CAAC,yBAAyB,CAAC,CAsGpC"}

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/generate-conditioned.d.ts

@@ -0,0 +1,50 @@
+import type { WorkflowJSON } from "../comfyui/types.js";
+interface CommonArgs {
+    prompt: string;
+    negative_prompt?: string;
+    width?: number;
+    height?: number;
+    steps?: number;
+    cfg?: number;
+    sampler?: string;
+    scheduler?: string;
+    seed?: number;
+    checkpoint?: string;
+}
+export interface ControlNetArgs extends CommonArgs {
+    control_image: string;
+    controlnet_model?: string;
+    strength?: number;
+}
+export interface IpAdapterArgs extends CommonArgs {
+    reference_image: string;
+    weight?: number;
+    preset?: string;
+}
+export interface ConditionedDeps {
+    resolveCheckpoint: () => Promise<string | undefined>;
+    enqueue: (workflow: WorkflowJSON) => Promise<{
+        prompt_id: string;
+        queue_remaining?: number;
+    }>;
+    /** Optional: resolve a local ControlNet model when none is given/defaulted. */
+    resolveControlNetModel?: () => Promise<string | undefined>;
+}
+export interface ConditionedResult {
+    prompt_id: string;
+    queue_remaining?: number;
+    checkpoint: string;
+}
+/**
+ * Build + enqueue a ControlNet-conditioned txt2img workflow. `control_image`
+ * must already exist in ComfyUI's input dir (use upload_image first).
+ */
+export declare function generateWithControlNet(args: ControlNetArgs, deps: ConditionedDeps): Promise<ConditionedResult>;
+/**
+ * Build + enqueue an IP-Adapter-conditioned txt2img workflow. Requires the
+ * ComfyUI_IPAdapter_plus custom nodes. `reference_image` must already exist in
+ * ComfyUI's input dir (use upload_image first).
+ */
+export declare function generateWithIpAdapter(args: IpAdapterArgs, deps: ConditionedDeps): Promise<ConditionedResult>;
+export {};
+//# sourceMappingURL=generate-conditioned.d.ts.map

package/dist/services/generate-conditioned.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-conditioned.d.ts","sourceRoot":"","sources":["../../src/services/generate-conditioned.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAKxD,UAAU,UAAU;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,cAAe,SAAQ,UAAU;IAChD,aAAa,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,aAAc,SAAQ,UAAU;IAC/C,eAAe,EAAE,MAAM,CAAC;IACxB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,eAAe;IAC9B,iBAAiB,EAAE,MAAM,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;IACrD,OAAO,EAAE,CAAC,QAAQ,EAAE,YAAY,KAAK,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,eAAe,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9F,+EAA+E;IAC/E,sBAAsB,CAAC,EAAE,MAAM,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;CAC5D;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACpB;AA0DD;;;GAGG;AACH,wBAAsB,sBAAsB,CAC1C,IAAI,EAAE,cAAc,EACpB,IAAI,EAAE,eAAe,GACpB,OAAO,CAAC,iBAAiB,CAAC,CA2B5B;AAED;;;;GAIG;AACH,wBAAsB,qBAAqB,CACzC,IAAI,EAAE,aAAa,EACnB,IAAI,EAAE,eAAe,GACpB,OAAO,CAAC,iBAAiB,CAAC,CAgB5B"}

package/dist/services/generate-conditioned.js

@@ -0,0 +1,95 @@
+import { createWorkflow } from "./workflow-composer.js";
+import { DefaultsManager } from "./defaults-manager.js";
+import { ValidationError } from "../utils/errors.js";
+const DEFAULTABLE_KEYS = [
+    "negative_prompt",
+    "width",
+    "height",
+    "steps",
+    "cfg",
+    "sampler",
+    "scheduler",
+    "seed",
+    "checkpoint",
+];
+function withDefaults(args) {
+    const seed = {};
+    for (const key of DEFAULTABLE_KEYS) {
+        const v = args[key];
+        if (v !== undefined)
+            seed[key] = v;
+    }
+    return DefaultsManager.apply(seed);
+}
+async function resolveCheckpointOrThrow(explicit, resolved, resolveCheckpoint) {
+    let checkpoint = explicit ?? resolved.checkpoint;
+    if (!checkpoint)
+        checkpoint = await resolveCheckpoint();
+    if (!checkpoint) {
+        throw new ValidationError("No checkpoint specified, defaulted, or found locally. " +
+            "Pass `checkpoint`, set a default via set_defaults, or download one with download_model.");
+    }
+    return checkpoint;
+}
+function commonTemplateParams(args, resolved, checkpoint) {
+    return {
+        checkpoint,
+        positive_prompt: args.prompt,
+        negative_prompt: resolved.negative_prompt,
+        width: resolved.width,
+        height: resolved.height,
+        steps: resolved.steps,
+        cfg: resolved.cfg,
+        seed: resolved.seed,
+        sampler_name: resolved.sampler,
+        scheduler: resolved.scheduler,
+    };
+}
+/**
+ * Build + enqueue a ControlNet-conditioned txt2img workflow. `control_image`
+ * must already exist in ComfyUI's input dir (use upload_image first).
+ */
+export async function generateWithControlNet(args, deps) {
+    if (!args.control_image) {
+        throw new ValidationError("control_image is required (upload it first with upload_image)");
+    }
+    const resolved = withDefaults(args);
+    const checkpoint = await resolveCheckpointOrThrow(args.checkpoint, resolved, deps.resolveCheckpoint);
+    let controlnetModel = args.controlnet_model;
+    if (!controlnetModel && deps.resolveControlNetModel) {
+        controlnetModel = await deps.resolveControlNetModel();
+    }
+    if (!controlnetModel) {
+        throw new ValidationError("No controlnet_model specified and none found locally. Pass `controlnet_model` " +
+            "(a file in models/controlnet/) or download one with download_model.");
+    }
+    const workflow = createWorkflow("controlnet", {
+        ...commonTemplateParams(args, resolved, checkpoint),
+        control_image: args.control_image,
+        controlnet_model: controlnetModel,
+        strength: args.strength,
+    });
+    const { prompt_id, queue_remaining } = await deps.enqueue(workflow);
+    return { prompt_id, queue_remaining, checkpoint };
+}
+/**
+ * Build + enqueue an IP-Adapter-conditioned txt2img workflow. Requires the
+ * ComfyUI_IPAdapter_plus custom nodes. `reference_image` must already exist in
+ * ComfyUI's input dir (use upload_image first).
+ */
+export async function generateWithIpAdapter(args, deps) {
+    if (!args.reference_image) {
+        throw new ValidationError("reference_image is required (upload it first with upload_image)");
+    }
+    const resolved = withDefaults(args);
+    const checkpoint = await resolveCheckpointOrThrow(args.checkpoint, resolved, deps.resolveCheckpoint);
+    const workflow = createWorkflow("ip_adapter", {
+        ...commonTemplateParams(args, resolved, checkpoint),
+        reference_image: args.reference_image,
+        weight: args.weight,
+        preset: args.preset,
+    });
+    const { prompt_id, queue_remaining } = await deps.enqueue(workflow);
+    return { prompt_id, queue_remaining, checkpoint };
+}
+//# sourceMappingURL=generate-conditioned.js.map

package/dist/services/generate-conditioned.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-conditioned.js","sourceRoot":"","sources":["../../src/services/generate-conditioned.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAwCrD,MAAM,gBAAgB,GAAG;IACvB,iBAAiB;IACjB,OAAO;IACP,QAAQ;IACR,OAAO;IACP,KAAK;IACL,SAAS;IACT,WAAW;IACX,MAAM;IACN,YAAY;CACJ,CAAC;AAEX,SAAS,YAAY,CAAC,IAA6B;IACjD,MAAM,IAAI,GAA4B,EAAE,CAAC;IACzC,KAAK,MAAM,GAAG,IAAI,gBAAgB,EAAE,CAAC;QACnC,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;QACpB,IAAI,CAAC,KAAK,SAAS;YAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AACrC,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,QAA4B,EAC5B,QAAiC,EACjC,iBAAoD;IAEpD,IAAI,UAAU,GAAG,QAAQ,IAAK,QAAQ,CAAC,UAAiC,CAAC;IACzE,IAAI,CAAC,UAAU;QAAE,UAAU,GAAG,MAAM,iBAAiB,EAAE,CAAC;IACxD,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,eAAe,CACvB,wDAAwD;YACtD,yFAAyF,CAC5F,CAAC;IACJ,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,oBAAoB,CAC3B,IAAgB,EAChB,QAAiC,EACjC,UAAkB;IAElB,OAAO;QACL,UAAU;QACV,eAAe,EAAE,IAAI,CAAC,MAAM;QAC5B,eAAe,EAAE,QAAQ,CAAC,eAAqC;QAC/D,KAAK,EAAE,QAAQ,CAAC,KAA2B;QAC3C,MAAM,EAAE,QAAQ,CAAC,MAA4B;QAC7C,KAAK,EAAE,QAAQ,CAAC,KAA2B;QAC3C,GAAG,EAAE,QAAQ,CAAC,GAAyB;QACvC,IAAI,EAAE,QAAQ,CAAC,IAA0B;QACzC,YAAY,EAAE,QAAQ,CAAC,OAA6B;QACpD,SAAS,EAAE,QAAQ,CAAC,SAA+B;KACpD,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,IAAoB,EACpB,IAAqB;IAErB,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QACxB,MAAM,IAAI,eAAe,CAAC,+DAA+D,CAAC,CAAC;IAC7F,CAAC;IACD,MAAM,QAAQ,GAAG,YAAY,CAAC,IAA0C,CAAC,CAAC;IAC1E,MAAM,UAAU,GAAG,MAAM,wBAAwB,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAErG,IAAI,eAAe,GAAG,IAAI,CAAC,gBAAgB,CAAC;IAC5C,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;QACpD,eAAe,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;IACxD,CAAC;IACD,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,MAAM,IAAI,eAAe,CACvB,gFAAgF;YAC9E,qEAAqE,CACxE,CAAC;IACJ,CAAC;IAED,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,EAAE;QAC5C,GAAG,oBAAoB,CAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,CAAC;QACnD,aAAa,EAAE,IAAI,CAAC,aAAa;QACjC,gBAAgB,EAAE,eAAe;QACjC,QAAQ,EAAE,IAAI,CAAC,QAAQ;KACxB,CAAC,CAAC;IAEH,MAAM,EAAE,SAAS,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpE,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,UAAU,EAAE,CAAC;AACpD,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,IAAmB,EACnB,IAAqB;IAErB,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;QAC1B,MAAM,IAAI,eAAe,CAAC,iEAAiE,CAAC,CAAC;IAC/F,CAAC;IACD,MAAM,QAAQ,GAAG,YAAY,CAAC,IAA0C,CAAC,CAAC;IAC1E,MAAM,UAAU,GAAG,MAAM,wBAAwB,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAErG,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,EAAE;QAC5C,GAAG,oBAAoB,CAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,CAAC;QACnD,eAAe,EAAE,IAAI,CAAC,eAAe;QACrC,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC,CAAC;IAEH,MAAM,EAAE,SAAS,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpE,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,UAAU,EAAE,CAAC;AACpD,CAAC"}