v1.6.4

Author: igorls

README.md

@@ -136,6 +136,56 @@ Hash-based routing intelligently handles failures at the workflow level (not cli
 
 See [Hash-Based Routing Guide](./docs/hash-routing-guide.md) for details and demos.
 
+### Advanced: `MultiWorkflowPool` for Heterogeneous Clusters
+
+For complex use cases involving a heterogeneous cluster of workers (e.g., some with SDXL models, others for video generation), `MultiWorkflowPool` provides fine-grained control over job routing based on workflow requirements.
+
+It uses an event-driven architecture to manage clients with specific **workflow affinities**, ensuring that jobs are only sent to nodes capable of processing them.
+
+- **Workflow Affinity:** Assign clients to specific workflows. Jobs are automatically routed to the correct client.
+- **Dynamic Job Queues:** A separate job queue is created for each workflow type, preventing head-of-line blocking.
+- **Event-Driven Architecture:** Zero polling for maximum efficiency and responsiveness.
+- **Built-in Monitoring:** Optional real-time monitoring of client and queue states.
+
+**Example:**
+```ts
+import { MultiWorkflowPool, Workflow } from "comfyui-node";
+import SdxlWorkflow from './sdxl-workflow.json';
+import VideoWorkflow from './video-workflow.json';
+
+// 1. Define workflows and generate their hash for affinity mapping
+const sdxlWF = Workflow.from(SdxlWorkflow).updateHash();
+const videoWF = Workflow.from(VideoWorkflow).updateHash();
+
+// 2. Create a new pool
+const pool = new MultiWorkflowPool({
+  logLevel: "info",
+  enableMonitoring: true,
+});
+
+// 3. Add clients with workflow affinity
+// This client is specialized for SDXL workflows
+pool.addClient("http://localhost:8188", { workflowAffinity: [sdxlWF] });
+
+// This client is specialized for Video workflows
+pool.addClient("http://localhost:8189", { workflowAffinity: [videoWF] });
+
+// This client is a general-purpose worker
+pool.addClient("http://localhost:8190");
+
+// 4. Initialize the pool (connects to all clients)
+await pool.init();
+
+// 5. Submit jobs
+// The pool automatically routes them to the correct client
+const sdxlJobId = await pool.submitJob(sdxlWF);
+const videoJobId = await pool.submitJob(videoWF);
+
+// 6. Wait for a job to complete
+const results = await pool.waitForJobCompletion(sdxlJobId);
+console.log("SDXL Job completed!", results.images);
+```
+
 ## What's New in v1.5.0
 
 - **WorkflowPool Profiling** – Enable automatic per-node performance tracking with `enableProfiling: true`.

dist/index.d.ts

@@ -14,6 +14,6 @@ export { seed } from "./tools.js";
 export { MultiWorkflowPool } from "./multipool/index.js";
 export { Workflow as MultiWorkflow } from "./multipool/workflow.js";
 export type { PoolEvent, ClientEventPayload, MultiWorkflowPoolOptions } from "./multipool/interfaces.js";
-export type { JobResults, JobState, JobStatus as MultiJobStatus, JobResultStatus } from "./multipool/job-state-registry.js";
-export type { JobProfileStats as MultiJobProfileStats } from "./multipool/job-profiler.js";
+export type { JobResults, JobState, JobStatus as MultiJobStatus, JobResultStatus } from "./multipool/interfaces.js";
+export type { JobProfileStats as MultiJobProfileStats } from "./multipool/interfaces.js";
 //# sourceMappingURL=index.d.ts.map

dist/index.d.ts.map

@@ -1,16 +1,7 @@
-import { MultiWorkflowPool } from "src/multipool/multi-workflow-pool.js";
-import { ComfyApi } from "src/client.js";
+import { MultiWorkflowPool } from "./multi-workflow-pool.js";
 import { Workflow } from "./workflow.js";
 import { Logger } from "./logger.js";
-export type ClientState = "idle" | "busy" | "offline";
-export interface EnhancedClient {
-    url: string;
-    state: ClientState;
-    nodeName: string;
-    priority?: number;
-    api: ComfyApi;
-    workflowAffinity?: Set<string>;
-}
+import { EnhancedClient } from "./interfaces.js";
 export declare class ClientRegistry {
     pool: MultiWorkflowPool;
     private logger;