comfyui-node 1.4.4 → 1.6.0

package/dist/pool/WorkflowPool.d.ts

@@ -4,10 +4,26 @@ import type { QueueAdapter, QueueStats } from "./queue/QueueAdapter.js";
 import type { FailoverStrategy } from "./failover/Strategy.js";
 import type { JobRecord, WorkflowInput, WorkflowJobOptions, JobId } from "./types/job.js";
 import type { WorkflowPoolEventMap } from "./types/events.js";
+import type { WorkflowAffinity } from "./types/affinity.js";
 /**
  * Configuration options for WorkflowPool.
  */
 export interface WorkflowPoolOpts {
+    /**
+     * An array of workflow affinity rules to establish a default mapping
+     * between workflows and specific clients.
+     *
+     * @example
+     * ```ts
+     * const pool = new WorkflowPool(clients, {
+     *   workflowAffinities: [
+     *     { workflowHash: "hash1", preferredClientIds: ["client-a"] },
+     *     { workflowHash: "hash2", excludeClientIds: ["client-b"] },
+     *   ]
+     * });
+     * ```
+     */
+    workflowAffinities?: WorkflowAffinity[];
     /**
      * Queue adapter for managing job queue operations.
      *
@@ -59,7 +75,7 @@ export interface WorkflowPoolOpts {
      *   executionStartTimeoutMs: 10000 // 10 seconds
      * });
      * ```
-     * @since 1.4.4
+     * @since 1.5.0
      */
     executionStartTimeoutMs?: number;
     /**
@@ -90,7 +106,7 @@ export interface WorkflowPoolOpts {
      * - Progress events (e.g., KSampler steps) reset the timeout
      * - First generation with model loading often needs longer timeout
      * - Cached nodes complete instantly and don't trigger timeout
-     * @since 1.4.4
+     * @since 1.5.0
      */
     nodeExecutionTimeoutMs?: number;
     /**
@@ -152,11 +168,19 @@ export declare class WorkflowPool extends TypedEventTarget<WorkflowPoolEventMap>
     private clientManager;
     private opts;
     private jobStore;
+    private jobFailureAnalysis;
+    private affinities;
     private initPromise;
     private processing;
+    private processQueued;
     private activeJobs;
+    private readonly queueDebug;
+    private debugLog;
     constructor(clients: ComfyApi[], opts?: WorkflowPoolOpts);
     ready(): Promise<void>;
+    setAffinity(affinity: WorkflowAffinity): void;
+    removeAffinity(workflowHash: string): boolean;
+    getAffinities(): WorkflowAffinity[];
     enqueue(workflowInput: WorkflowInput, options?: WorkflowJobOptions): Promise<JobId>;
     getJob(jobId: string): JobRecord | undefined;
     cancel(jobId: string): Promise<boolean>;
@@ -167,6 +191,12 @@ export declare class WorkflowPool extends TypedEventTarget<WorkflowPoolEventMap>
     private static fallbackId;
     private scheduleProcess;
     private applyAutoSeed;
+    private rememberJobFailure;
+    private clearJobFailures;
+    private collectFailureReasons;
+    private addPermanentExclusion;
+    private hasRetryPath;
+    private createWorkflowNotSupportedError;
     private processQueue;
     private runJob;
 }

package/dist/pool/WorkflowPool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"WorkflowPool.d.ts","sourceRoot":"","sources":["../../src/pool/WorkflowPool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAK7C,OAAO,KAAK,EAAE,YAAY,EAAoB,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAC1F,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAK/D,OAAO,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,kBAAkB,EAAsB,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAC9G,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAG9D;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B;;;;;;;;;;;OAWG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;IAE5B;;;;;;;;;;;OAWG;IACH,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IAEpC;;;;;OAKG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB;;;;;;;;;;;;;;;;;;;OAmBG;IACH,uBAAuB,CAAC,EAAE,MAAM,CAAC;IAEjC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BG;IACH,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAEhC;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAE/B;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B;AAaD,qBAAa,YAAa,SAAQ,gBAAgB,CAAC,oBAAoB,CAAC;IACtE,OAAO,CAAC,KAAK,CAAe;IAC5B,OAAO,CAAC,QAAQ,CAAmB;IACnC,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,IAAI,CAAmB;IAC/B,OAAO,CAAC,QAAQ,CAAoC;IACpD,OAAO,CAAC,WAAW,CAAgB;IACnC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,UAAU,CAA2C;gBAEjD,OAAO,EAAE,QAAQ,EAAE,EAAE,IAAI,CAAC,EAAE,gBAAgB;IA6BlD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAItB,OAAO,CAAC,aAAa,EAAE,aAAa,EAAE,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC;IAsDzF,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAItC,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAyBvC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IASzB,aAAa,IAAI,OAAO,CAAC,UAAU,CAAC;IAI1C,OAAO,CAAC,iBAAiB;IAazB,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,MAAM,CAAC,UAAU;IAMzB,OAAO,CAAC,eAAe;IAOvB,OAAO,CAAC,aAAa;YAeP,YAAY;YA+BZ,MAAM;CA8brB"}
+{"version":3,"file":"WorkflowPool.d.ts","sourceRoot":"","sources":["../../src/pool/WorkflowPool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAK7C,OAAO,KAAK,EAAE,YAAY,EAAoB,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAC1F,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAK/D,OAAO,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,kBAAkB,EAAsB,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAC9G,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAK9D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE5D;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B;;;;;;;;;;;;;OAaG;IACH,kBAAkB,CAAC,EAAE,gBAAgB,EAAE,CAAC;IACxC;;;;;;;;;;;OAWG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;IAE5B;;;;;;;;;;;OAWG;IACH,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IAEpC;;;;;OAKG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB;;;;;;;;;;;;;;;;;;;OAmBG;IACH,uBAAuB,CAAC,EAAE,MAAM,CAAC;IAEjC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BG;IACH,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAEhC;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAE/B;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B;AAaD,qBAAa,YAAa,SAAQ,gBAAgB,CAAC,oBAAoB,CAAC;IACtE,OAAO,CAAC,KAAK,CAAe;IAC5B,OAAO,CAAC,QAAQ,CAAmB;IACnC,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,IAAI,CAAmB;IAC/B,OAAO,CAAC,QAAQ,CAAoC;IACpD,OAAO,CAAC,kBAAkB,CAA+D;IACzF,OAAO,CAAC,UAAU,CAA4C;IAC9D,OAAO,CAAC,WAAW,CAAgB;IACnC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,UAAU,CAA2C;IAC7D,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA2C;IAEtE,OAAO,CAAC,QAAQ;gBAMJ,OAAO,EAAE,QAAQ,EAAE,EAAE,IAAI,CAAC,EAAE,gBAAgB;IAoClD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAIrB,WAAW,CAAC,QAAQ,EAAE,gBAAgB,GAAG,IAAI;IAI7C,cAAc,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO;IAI7C,aAAa,IAAI,gBAAgB,EAAE;IAIpC,OAAO,CAAC,aAAa,EAAE,aAAa,EAAE,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,KAAK,CAAC;IAqEzF,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAItC,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IA2BvC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IASzB,aAAa,IAAI,OAAO,CAAC,UAAU,CAAC;IAI1C,OAAO,CAAC,iBAAiB;IAazB,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,MAAM,CAAC,UAAU;IAMzB,OAAO,CAAC,eAAe;IAOvB,OAAO,CAAC,aAAa;IAerB,OAAO,CAAC,kBAAkB;IAS1B,OAAO,CAAC,gBAAgB;IAIxB,OAAO,CAAC,qBAAqB;IAY7B,OAAO,CAAC,qBAAqB;IAS7B,OAAO,CAAC,YAAY;IAoBpB,OAAO,CAAC,+BAA+B;YAUzB,YAAY;YA+JZ,MAAM;CAyerB"}