@hypercli.com/sdk 0.9.1

package/dist/job/base.js

@@ -0,0 +1,206 @@
+import { findJob } from '../jobs.js';
+import { requestWithRetry } from '../http.js';
+/**
+ * Base class for managed GPU jobs with lifecycle helpers
+ */
+export class BaseJob {
+    client;
+    job;
+    static DEFAULT_IMAGE = '';
+    static DEFAULT_GPU_TYPE = 'l40s';
+    static HEALTH_ENDPOINT = '/';
+    static HEALTH_TIMEOUT = 5000;
+    /**
+     * Consumer-side hostname settling window.  The API returns after committing
+     * the Cloudflare record; waiting before the first lookup avoids caching a
+     * transient NXDOMAIN in recursive resolvers.
+     */
+    static DEFAULT_HOSTNAME_SETTLE_DELAY = 15000;
+    _baseUrl = null;
+    constructor(client, job) {
+        this.client = client;
+        this.job = job;
+    }
+    get jobId() {
+        return this.job.jobId;
+    }
+    get hostname() {
+        return this.job.hostname;
+    }
+    get baseUrl() {
+        if (!this._baseUrl && this.hostname) {
+            this._baseUrl = `http://${this.hostname}`;
+        }
+        return this._baseUrl || '';
+    }
+    get authHeaders() {
+        return { 'Authorization': `Bearer ${this.client.apiKey}` };
+    }
+    /**
+     * Find an existing running job, optionally filtering by image
+     */
+    static async getRunning(client, imageFilter) {
+        const jobs = await client.jobs.list('running');
+        for (const job of jobs) {
+            if (imageFilter && !job.dockerImage.includes(imageFilter)) {
+                continue;
+            }
+            return new this(client, job);
+        }
+        return null;
+    }
+    /**
+     * Get a job by ID, hostname, or IP address
+     */
+    static async getByInstance(client, instance, state = 'running') {
+        const job = await findJob(client.jobs, instance, state);
+        if (!job) {
+            throw new Error(`No job found matching: ${instance}`);
+        }
+        return new this(client, job);
+    }
+    /**
+     * Create a new job
+     */
+    static async create(client, options = {}) {
+        const { image, gpuType, gpuCount = 1, runtime = 3600, ...kwargs } = options;
+        const job = await client.jobs.create({
+            image: image || this.DEFAULT_IMAGE,
+            gpuType: gpuType || this.DEFAULT_GPU_TYPE,
+            gpuCount,
+            runtime,
+            ...kwargs,
+        });
+        return new this(client, job);
+    }
+    /**
+     * Get existing running job or create new one
+     */
+    static async getOrCreate(client, options = {}) {
+        const { reuse = true, image, ...restOptions } = options;
+        if (reuse) {
+            const existing = await this.getRunning(client, image || this.DEFAULT_IMAGE);
+            if (existing) {
+                return existing;
+            }
+        }
+        return this.create(client, { image, ...restOptions });
+    }
+    /**
+     * Refresh job state from API
+     */
+    async refresh() {
+        this.job = await this.client.jobs.get(this.jobId);
+        this._baseUrl = null;
+        return this;
+    }
+    /**
+     * Wait for job to reach running state via API polling
+     */
+    async waitForRunning(timeout = 300000, pollInterval = 5000) {
+        const start = Date.now();
+        while (Date.now() - start < timeout) {
+            await this.refresh();
+            if (this.job.state === 'running' && this.hostname) {
+                return true;
+            }
+            if (['failed', 'cancelled', 'completed', 'terminated'].includes(this.job.state)) {
+                return false;
+            }
+            await new Promise(resolve => setTimeout(resolve, pollInterval));
+        }
+        return false;
+    }
+    /**
+     * Alias for waitForRunning
+     */
+    async waitForHostname(timeout = 300000, pollInterval = 5000) {
+        return this.waitForRunning(timeout, pollInterval);
+    }
+    /**
+     * Check if the service is responding
+     */
+    async checkHealth() {
+        if (!this.baseUrl || this.job.state !== 'running') {
+            return false;
+        }
+        try {
+            const Constructor = this.constructor;
+            const response = await requestWithRetry({
+                method: 'GET',
+                url: `${this.baseUrl}${Constructor.HEALTH_ENDPOINT}`,
+                headers: this.authHeaders,
+                timeout: Constructor.HEALTH_TIMEOUT,
+                retries: 3,
+            });
+            return response.status === 200;
+        }
+        catch {
+            return false;
+        }
+    }
+    /**
+     * Wait for an EXISTING running job to respond to health checks
+     */
+    async waitExisting(timeout = 15000) {
+        await this.refresh();
+        if (this.job.state !== 'running' || !this.hostname) {
+            return false;
+        }
+        const start = Date.now();
+        while (Date.now() - start < timeout) {
+            if (await this.checkHealth()) {
+                return true;
+            }
+            await new Promise(resolve => setTimeout(resolve, 2000));
+        }
+        return false;
+    }
+    /**
+     * Wait for a NEW job to be ready (running state + health check passing)
+     */
+    async waitReady(timeout = 300000, pollInterval = 5000, dnsDelay = this.constructor.DEFAULT_HOSTNAME_SETTLE_DELAY) {
+        const start = Date.now();
+        // Wait for running state via API
+        await this.refresh();
+        if (['failed', 'cancelled', 'completed', 'terminated'].includes(this.job.state)) {
+            return false;
+        }
+        if (this.job.state !== 'running' || !this.hostname) {
+            if (!await this.waitForRunning(timeout, pollInterval)) {
+                return false;
+            }
+        }
+        // DNS propagation delay
+        if (dnsDelay > 0) {
+            await new Promise(resolve => setTimeout(resolve, dnsDelay));
+        }
+        // Job is running, check health
+        const elapsed = Date.now() - start;
+        const remaining = timeout - elapsed;
+        while (remaining > 0) {
+            if (await this.checkHealth()) {
+                return true;
+            }
+            await new Promise(resolve => setTimeout(resolve, 2000));
+            const newRemaining = timeout - (Date.now() - start);
+            if (newRemaining <= 0)
+                break;
+        }
+        return false;
+    }
+    /**
+     * Cancel the job
+     */
+    async shutdown() {
+        return await this.client.jobs.cancel(this.jobId);
+    }
+    /**
+     * Extend job runtime
+     */
+    async extend(runtime) {
+        this.job = await this.client.jobs.extend(this.jobId, runtime);
+        return this;
+    }
+}
+//# sourceMappingURL=base.js.map

package/dist/job/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/job/base.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAU9C;;GAEG;AACH,MAAM,OAAO,OAAO;IAgBT;IACA;IAhBT,MAAM,CAAC,aAAa,GAAW,EAAE,CAAC;IAClC,MAAM,CAAC,gBAAgB,GAAW,MAAM,CAAC;IACzC,MAAM,CAAC,eAAe,GAAW,GAAG,CAAC;IACrC,MAAM,CAAC,cAAc,GAAW,IAAI,CAAC;IAErC;;;;OAIG;IACH,MAAM,CAAC,6BAA6B,GAAW,KAAK,CAAC;IAE3C,QAAQ,GAAkB,IAAI,CAAC;IAEzC,YACS,MAAgB,EAChB,GAAQ;QADR,WAAM,GAAN,MAAM,CAAU;QAChB,QAAG,GAAH,GAAG,CAAK;IACd,CAAC;IAEJ,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC;IACxB,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC;IAC3B,CAAC;IAED,IAAI,OAAO;QACT,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpC,IAAI,CAAC,QAAQ,GAAG,UAAU,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC5C,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,IAAI,WAAW;QACb,OAAO,EAAE,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;IAC7D,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,UAAU,CAErB,MAAgB,EAChB,WAAoB;QAEpB,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/C,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,IAAI,WAAW,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC1D,SAAS;YACX,CAAC;YACD,OAAO,IAAI,IAAI,CAAC,MAAM,EAAE,GAAG,CAAoB,CAAC;QAClD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,aAAa,CAExB,MAAgB,EAChB,QAAgB,EAChB,QAAgB,SAAS;QAEzB,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QACxD,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,EAAE,CAAC,CAAC;QACxD,CAAC;QACD,OAAO,IAAI,IAAI,CAAC,MAAM,EAAE,GAAG,CAAoB,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAEjB,MAAgB,EAChB,UAA0B,EAAE;QAE5B,MAAM,EACJ,KAAK,EACL,OAAO,EACP,QAAQ,GAAG,CAAC,EACZ,OAAO,GAAG,IAAI,EACd,GAAG,MAAM,EACV,GAAG,OAAO,CAAC;QAEZ,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;YACnC,KAAK,EAAE,KAAK,IAAI,IAAI,CAAC,aAAa;YAClC,OAAO,EAAE,OAAO,IAAI,IAAI,CAAC,gBAAgB;YACzC,QAAQ;YACR,OAAO;YACP,GAAG,MAAM;SACV,CAAC,CAAC;QAEH,OAAO,IAAI,IAAI,CAAC,MAAM,EAAE,GAAG,CAAoB,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CAEtB,MAAgB,EAChB,UAAgD,EAAE;QAElD,MAAM,EAAE,KAAK,GAAG,IAAI,EAAE,KAAK,EAAE,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC;QAExD,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,KAAK,IAAI,IAAI,CAAC,aAAa,CAAC,CAAC;YAC5E,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,QAA2B,CAAC;YACrC,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC;IACxD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,UAAkB,MAAM,EAAE,eAAuB,IAAI;QACxE,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,OAAO,EAAE,CAAC;YACpC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YACrB,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClD,OAAO,IAAI,CAAC;YACd,CAAC;YACD,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChF,OAAO,KAAK,CAAC;YACf,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC;QAClE,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,UAAkB,MAAM,EAAE,eAAuB,IAAI;QACzE,OAAO,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW;QACf,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAClD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,WAA6B,CAAC;YACvD,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC;gBACtC,MAAM,EAAE,KAAK;gBACb,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,GAAG,WAAW,CAAC,eAAe,EAAE;gBACpD,OAAO,EAAE,IAAI,CAAC,WAAW;gBACzB,OAAO,EAAE,WAAW,CAAC,cAAc;gBACnC,OAAO,EAAE,CAAC;aACX,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,MAAM,KAAK,GAAG,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,UAAkB,KAAK;QACxC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,OAAO,EAAE,CAAC;YACpC,IAAI,MAAM,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC7B,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CACb,UAAkB,MAAM,EACxB,eAAuB,IAAI,EAC3B,WAAoB,IAAI,CAAC,WAA8B,CAAC,6BAA6B;QAErF,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,iCAAiC;QACjC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YAChF,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnD,IAAI,CAAC,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE,CAAC;gBACtD,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAED,wBAAwB;QACxB,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;YACjB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC9D,CAAC;QAED,+BAA+B;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;QACnC,MAAM,SAAS,GAAG,OAAO,GAAG,OAAO,CAAC;QAEpC,OAAO,SAAS,GAAG,CAAC,EAAE,CAAC;YACrB,IAAI,MAAM,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC7B,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YACxD,MAAM,YAAY,GAAG,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;YACpD,IAAI,YAAY,IAAI,CAAC;gBAAE,MAAM;QAC/B,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,OAAe;QAC1B,IAAI,CAAC,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC;IACd,CAAC"}

package/dist/job/comfyui.d.ts

@@ -0,0 +1,259 @@
+/**
+ * ComfyUI job helpers and workflow conversion utilities
+ */
+import type { HyperCLI } from '../client.js';
+import type { Job } from '../jobs.js';
+import { BaseJob } from './base.js';
+/**
+ * Load a workflow template JSON in graph format (nodes array, links array).
+ *
+ * The Python SDK resolves IDs via the comfyui-workflow-templates pip package;
+ * Node has no equivalent package, so `templateId` must resolve to a JSON file:
+ * an explicit path, `<templateId>.json`, or — when `templatesDir` is given —
+ * `<templatesDir>/<templateId>.json` or `<templatesDir>/<templateId>/<templateId>.json`.
+ */
+export declare function loadTemplate(templateId: string, options?: {
+    templatesDir?: string;
+}): Record<string, any>;
+/** Check whether a widget value matches the expected input type. */
+export declare function valueMatchesType(value: any, inputSpec: any): boolean;
+/**
+ * Expand subgraph/group nodes into their constituent nodes.
+ *
+ * ComfyUI supports "workflow components" (group nodes) where a subgraph is
+ * collapsed into a single node with a UUID type. These must be expanded before
+ * the workflow can be executed. Returns a new graph; the input is not modified.
+ */
+export declare function expandSubgraphs(graph: Record<string, any>, debug?: boolean): Record<string, any>;
+export declare const DEFAULT_OBJECT_INFO: Record<string, any>;
+/**
+ * Find nodes in API-format workflow by class_type and optional title pattern
+ */
+export declare function findNodes(workflow: Record<string, any>, classType: string, titleContains?: string): Array<[string, any]>;
+/**
+ * Find first node matching class_type and optional title pattern
+ */
+export declare function findNode(workflow: Record<string, any>, classType: string, titleContains?: string): [string, any] | [null, null];
+/**
+ * Apply parameters to workflow nodes
+ */
+export declare function applyParams(workflow: Record<string, any>, params: Record<string, any>): Record<string, any>;
+/**
+ * Convert ComfyUI graph format (from UI) to API format (for /prompt endpoint).
+ */
+export declare function graphToApi(graph: any, objectInfo?: Record<string, any>, debug?: boolean): Record<string, any>;
+/**
+ * Apply graph modes (enable/disable nodes)
+ */
+export declare function applyGraphModes(graph: any, nodesConfig: Record<string, any>): any;
+/**
+ * ComfyUI-specific job with workflow execution helpers
+ */
+export declare class ComfyUIJob extends BaseJob {
+    static DEFAULT_IMAGE: string;
+    static DEFAULT_GPU_TYPE: string;
+    static HEALTH_ENDPOINT: string;
+    static COMFYUI_PORT: number;
+    private _objectInfo;
+    private _jobToken;
+    private _useLb;
+    useAuth: boolean;
+    template: string | null;
+    constructor(client: HyperCLI, job: Job, template?: string, useLb?: boolean, useAuth?: boolean);
+    get useLb(): boolean;
+    set useLb(value: boolean);
+    get baseUrl(): string;
+    get authHeaders(): Record<string, string>;
+    jobToken(): Promise<string>;
+    /**
+     * Create a new ComfyUI job configured for a specific template
+     */
+    static createForTemplate(client: HyperCLI, template: string, options?: {
+        gpuType?: string;
+        gpuCount?: number;
+        runtime?: number;
+        lb?: number;
+        auth?: boolean;
+        [key: string]: any;
+    }): Promise<ComfyUIJob>;
+    /**
+     * Get object_info from ComfyUI (cached)
+     */
+    getObjectInfo(refresh?: boolean): Promise<Record<string, any>>;
+    /**
+     * Convert graph format workflow to API format
+     */
+    convertWorkflow(graph: any, debug?: boolean): Promise<Record<string, any>>;
+    /**
+     * Connect to a specific ComfyUI instance by job ID or hostname.
+     */
+    static getInstance(client: HyperCLI, instance: string, options?: {
+        useLb?: boolean;
+        useAuth?: boolean;
+    }): Promise<ComfyUIJob>;
+    /**
+     * Get existing running job or create new one for a template.
+     *
+     * If reuse is true and a ComfyUI job is already running, it will be reused
+     * (note: the existing job may have different models loaded).
+     */
+    static getOrCreateForTemplate(client: HyperCLI, template: string, options?: {
+        gpuType?: string;
+        gpuCount?: number;
+        runtime?: number;
+        reuse?: boolean;
+        lb?: number;
+        auth?: boolean;
+        [key: string]: any;
+    }): Promise<ComfyUIJob>;
+    /**
+     * Load workflow template JSON in graph format.
+     *
+     * Unlike the Python SDK, Node has no comfyui-workflow-templates package, so
+     * the ID must resolve to a JSON file (see the module-level loadTemplate).
+     */
+    loadTemplate(templateId: string, options?: {
+        templatesDir?: string;
+    }): Record<string, any>;
+    private comfyJson;
+    private resolvedAuthHeaders;
+    private static isTransientNetworkError;
+    /**
+     * Get set of available node class_types from ComfyUI's /object_info.
+     */
+    getAvailableNodeTypes(): Promise<Set<string>>;
+    /**
+     * Extract all class_type values from a workflow (API format or UI format).
+     */
+    getWorkflowNodeTypes(workflow: Record<string, any>): Set<string>;
+    /**
+     * Find node types in workflow that aren't available in ComfyUI.
+     */
+    getMissingNodeTypes(workflow: Record<string, any>): Promise<Set<string>>;
+    /**
+     * Get node class_type to package mappings from ComfyUI Manager.
+     *
+     * Returns a mapping of package URL to [node_list, metadata].
+     */
+    getNodeMappings(): Promise<Record<string, any>>;
+    /**
+     * Look up which packages provide the given node types.
+     *
+     * Returns a mapping of package URL to the node types it provides.
+     */
+    lookupPackagesForNodes(nodeTypes: Iterable<string>): Promise<Record<string, string[]>>;
+    /**
+     * Get the full list of available custom node packages.
+     */
+    getCustomNodeList(): Promise<Record<string, any>>;
+    /**
+     * Install custom node packages by their repository URLs. Returns
+     * { queued, failed, notFound } package titles/URLs.
+     */
+    installPackagesByUrl(packageUrls: string[]): Promise<{
+        queued: string[];
+        failed: string[];
+        notFound: string[];
+    }>;
+    /**
+     * Automatically install missing custom nodes for a workflow.
+     */
+    autoInstallWorkflowNodes(workflow: Record<string, any>, options?: {
+        wait?: boolean;
+        reboot?: boolean;
+    }): Promise<{
+        missingNodes: string[];
+        packagesToInstall: Record<string, string[]>;
+        installed: string[];
+        failed: string[];
+        notFoundNodes: string[];
+    }>;
+    /**
+     * Get list of installed custom nodes from ComfyUI Manager.
+     */
+    getInstalledNodes(): Promise<Record<string, any>>;
+    /**
+     * Install a custom node package via ComfyUI Manager.
+     * Returns true if installation was queued successfully.
+     */
+    installNode(nodeName: string): Promise<boolean>;
+    /**
+     * Install multiple custom node packages. Returns { queued, failed }.
+     */
+    installNodes(nodeNames: string[], options?: {
+        wait?: boolean;
+    }): Promise<{
+        queued: string[];
+        failed: string[];
+    }>;
+    /**
+     * Reboot ComfyUI server (required after installing nodes).
+     */
+    reboot(options?: {
+        waitReady?: boolean;
+        timeoutSeconds?: number;
+    }): Promise<boolean>;
+    /**
+     * Ensure custom nodes are installed, installing missing ones. Returns
+     * { alreadyInstalled, installed, failed }.
+     */
+    ensureNodesInstalled(nodeNames: string[]): Promise<{
+        alreadyInstalled: string[];
+        installed: string[];
+        failed: string[];
+    }>;
+    /**
+     * Submit workflow to ComfyUI, returns prompt_id.
+     * Retries on connection errors including DNS failures.
+     */
+    queuePrompt(workflow: Record<string, any>, retries?: number): Promise<string>;
+    /**
+     * Get execution history for a prompt. Retries on connection errors including
+     * DNS failures.
+     */
+    getHistory(promptId: string, retries?: number): Promise<Record<string, any> | null>;
+    /**
+     * Wait for prompt execution to complete, returns the history entry.
+     */
+    waitForCompletion(promptId: string, options?: {
+        timeoutSeconds?: number;
+        pollIntervalSeconds?: number;
+    }): Promise<Record<string, any>>;
+    /**
+     * Download output file from ComfyUI server.
+     * If the file already exists, increments the name (file_1.png, file_2.png, etc.)
+     */
+    downloadOutput(filename: string, options?: {
+        outputDir?: string;
+        subfolder?: string;
+        retries?: number;
+    }): Promise<string>;
+    /**
+     * Run a workflow and wait for completion.
+     */
+    run(workflow: Record<string, any>, options?: {
+        timeoutSeconds?: number;
+        convert?: boolean;
+    }): Promise<Record<string, any>>;
+    /**
+     * Run a template workflow with parameter overrides.
+     */
+    runTemplate(templateId: string, options?: {
+        timeoutSeconds?: number;
+        templatesDir?: string;
+        params?: Record<string, any>;
+    }): Promise<Record<string, any>>;
+    /**
+     * Extract output file info from a history entry (images, videos, gifs).
+     */
+    getOutputImages(history: Record<string, any>): Array<Record<string, any>>;
+    /**
+     * Upload image to ComfyUI server
+     */
+    uploadImage(filePath: string, filename?: string): Promise<string>;
+    /**
+     * Upload audio to ComfyUI server
+     */
+    uploadAudio(filePath: string, filename?: string): Promise<string>;
+}
+//# sourceMappingURL=comfyui.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"comfyui.d.ts","sourceRoot":"","sources":["../../src/job/comfyui.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAUpC;;;;;;;GAOG;AACH,wBAAgB,YAAY,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,GAAE;IAAE,YAAY,CAAC,EAAE,MAAM,CAAA;CAAO,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAkB7G;AAED,oEAAoE;AACpE,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,GAAG,OAAO,CAqDpE;AAED;;;;;;GAMG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,KAAK,GAAE,OAAe,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAmMvG;AAID,eAAO,MAAM,mBAAmB,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAqKnD,CAAC;AAEF;;GAEG;AACH,wBAAgB,SAAS,CACvB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC7B,SAAS,EAAE,MAAM,EACjB,aAAa,CAAC,EAAE,MAAM,GACrB,KAAK,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAiBtB;AAED;;GAEG;AACH,wBAAgB,QAAQ,CACtB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC7B,SAAS,EAAE,MAAM,EACjB,aAAa,CAAC,EAAE,MAAM,GACrB,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAG9B;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAkK3G;AAED;;GAEG;AACH,wBAAgB,UAAU,CAAC,KAAK,EAAE,GAAG,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,KAAK,GAAE,OAAe,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAoJpH;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,GAAG,EAAE,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,GAAG,CAmBjF;AAED;;GAEG;AACH,qBAAa,UAAW,SAAQ,OAAO;IACrC,OAAgB,aAAa,SAAiB;IAC9C,OAAgB,gBAAgB,SAAU;IAC1C,OAAgB,eAAe,SAAmB;IAClD,MAAM,CAAC,YAAY,SAAQ;IAE3B,OAAO,CAAC,WAAW,CAAoC;IACvD,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,MAAM,CAAU;IACjB,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAQ;gBAGpC,MAAM,EAAE,QAAQ,EAChB,GAAG,EAAE,GAAG,EACR,QAAQ,CAAC,EAAE,MAAM,EACjB,KAAK,GAAE,OAAe,EACtB,OAAO,GAAE,OAAe;IAQ1B,IAAI,KAAK,IAAI,OAAO,CAEnB;IAED,IAAI,KAAK,CAAC,KAAK,EAAE,OAAO,EAGvB;IAED,IAAa,OAAO,IAAI,MAAM,CAS7B;IAED,IAAa,WAAW,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAQjD;IAEK,QAAQ,IAAI,OAAO,CAAC,MAAM,CAAC;IAOjC;;OAEG;WACU,iBAAiB,CAC5B,MAAM,EAAE,QAAQ,EAChB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QACP,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,IAAI,CAAC,EAAE,OAAO,CAAC;QACf,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;KACf,GACL,OAAO,CAAC,UAAU,CAAC;IA2BtB;;OAEG;IACG,aAAa,CAAC,OAAO,GAAE,OAAe,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAe3E;;OAEG;IACG,eAAe,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,GAAE,OAAe,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAKvF;;OAEG;WACU,WAAW,CACtB,MAAM,EAAE,QAAQ,EAChB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAA;KAAO,GACnD,OAAO,CAAC,UAAU,CAAC;IAqBtB;;;;;OAKG;WACU,sBAAsB,CACjC,MAAM,EAAE,QAAQ,EAChB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QACP,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,CAAC,EAAE,OAAO,CAAC;QAChB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,IAAI,CAAC,EAAE,OAAO,CAAC;QACf,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;KACf,GACL,OAAO,CAAC,UAAU,CAAC;IAuBtB;;;;;OAKG;IACH,YAAY,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,GAAE;QAAE,YAAY,CAAC,EAAE,MAAM,CAAA;KAAO,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;YAQhF,SAAS;YAwBT,mBAAmB;IAOjC,OAAO,CAAC,MAAM,CAAC,uBAAuB;IAMtC;;OAEG;IACG,qBAAqB,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAKnD;;OAEG;IACH,oBAAoB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC;IAqBhE;;OAEG;IACG,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAY9E;;;;OAIG;IACG,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAIrD;;;;OAIG;IACG,sBAAsB,CAAC,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAwB5F;;OAEG;IACG,iBAAiB,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAOvD;;;OAGG;IACG,oBAAoB,CAAC,WAAW,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAAE,MAAM,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAC;QAAC,QAAQ,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IAmDtH;;OAEG;IACG,wBAAwB,CAC5B,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC7B,OAAO,GAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,OAAO,CAAA;KAAO,GACjD,OAAO,CAAC;QACT,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5C,SAAS,EAAE,MAAM,EAAE,CAAC;QACpB,MAAM,EAAE,MAAM,EAAE,CAAC;QACjB,aAAa,EAAE,MAAM,EAAE,CAAC;KACzB,CAAC;IA0DF;;OAEG;IACG,iBAAiB,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAIvD;;;OAGG;IACG,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAuBrD;;OAEG;IACG,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,OAAO,GAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAA;KAAO,GAAG,OAAO,CAAC;QAAE,MAAM,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IAwD1H;;OAEG;IACG,MAAM,CAAC,OAAO,GAAE;QAAE,SAAS,CAAC,EAAE,OAAO,CAAC;QAAC,cAAc,CAAC,EAAE,MAAM,CAAA;KAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAsB9F;;;OAGG;IACG,oBAAoB,CACxB,SAAS,EAAE,MAAM,EAAE,GAClB,OAAO,CAAC;QAAE,gBAAgB,EAAE,MAAM,EAAE,CAAC;QAAC,SAAS,EAAE,MAAM,EAAE,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IAiCjF;;;OAGG;IACG,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,GAAE,MAAU,GAAG,OAAO,CAAC,MAAM,CAAC;IA0CtF;;;OAGG;IACG,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,GAAE,MAAU,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC;IA0B5F;;OAEG;IACG,iBAAiB,CACrB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,mBAAmB,CAAC,EAAE,MAAM,CAAA;KAAO,GACtE,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAoC/B;;;OAGG;IACG,cAAc,CAClB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAO,GACzE,OAAO,CAAC,MAAM,CAAC;IAsDlB;;OAEG;IACG,GAAG,CACP,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAC7B,OAAO,GAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAA;KAAO,GAC3D,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAY/B;;OAEG;IACG,WAAW,CACf,UAAU,EAAE,MAAM,EAClB,OAAO,GAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAO,GAC7F,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAY/B;;OAEG;IACH,eAAe,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAazE;;OAEG;IACG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAsBvE;;OAEG;IACG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAqBxE"}