comfyui-node 1.6.1 → 1.6.3

package/dist/multipool/tests/client-registry.spec.js

@@ -0,0 +1,191 @@
+import { describe, it, expect, beforeEach, jest } from "bun:test";
+import { ClientRegistry } from "../client-registry.js";
+// Mock dependencies
+const createPoolMock = () => ({});
+const createLoggerMock = () => ({
+    debug: jest.fn(),
+    error: jest.fn(),
+});
+const createWorkflowMock = (hash) => {
+    const mock = {
+        structureHash: hash,
+        updateHash: jest.fn(),
+    };
+    mock.updateHash.mockImplementation(() => {
+        mock.structureHash = "mock-hash";
+    });
+    return mock;
+};
+describe("ClientRegistry", () => {
+    let poolMock;
+    let loggerMock;
+    let registry;
+    beforeEach(() => {
+        poolMock = createPoolMock();
+        loggerMock = createLoggerMock();
+        registry = new ClientRegistry(poolMock, loggerMock);
+        jest.clearAllMocks();
+    });
+    describe("addClient", () => {
+        it("should add a client without options", () => {
+            const clientUrl = "http://localhost:8188";
+            registry.addClient(clientUrl);
+            expect(registry.clients.has(clientUrl)).toBe(true);
+            const client = registry.clients.get(clientUrl);
+            expect(client.url).toBe(clientUrl);
+            expect(client.state).toBe("idle");
+            expect(client.nodeName).toBe("localhost");
+            expect(client.priority).toBeUndefined();
+            expect(client.workflowAffinity).toBeUndefined();
+        });
+        it("should add a client with priority", () => {
+            const clientUrl = "http://localhost:8188";
+            registry.addClient(clientUrl, { priority: 5 });
+            const client = registry.clients.get(clientUrl);
+            expect(client.priority).toBe(5);
+        });
+        it("should add a client with workflow affinity", () => {
+            const clientUrl = "http://localhost:8188";
+            const workflow = createWorkflowMock("hash1");
+            registry.addClient(clientUrl, { workflowAffinity: [workflow] });
+            const client = registry.clients.get(clientUrl);
+            expect(client.workflowAffinity).toEqual(new Set(["hash1"]));
+            expect(registry.workflowAffinityMap.get("hash1")).toEqual(new Set([clientUrl]));
+        });
+        it("should update hash if not present in workflow", () => {
+            const clientUrl = "http://localhost:8188";
+            const workflow = createWorkflowMock(); // no hash
+            registry.addClient(clientUrl, { workflowAffinity: [workflow] });
+            expect(workflow.updateHash).toHaveBeenCalled();
+            const client = registry.clients.get(clientUrl);
+            expect(client.workflowAffinity).toEqual(new Set(["mock-hash"]));
+        });
+        it("should throw error if workflow has no hash after update", () => {
+            const clientUrl = "http://localhost:8188";
+            const workflow = createWorkflowMock();
+            workflow.updateHash.mockImplementation(() => {
+                // don't set hash
+            });
+            expect(() => registry.addClient(clientUrl, { workflowAffinity: [workflow] })).toThrow("Workflow must have a valid structure hash for affinity.");
+        });
+    });
+    describe("removeClient", () => {
+        it("should remove a client", () => {
+            const clientUrl = "http://localhost:8188";
+            registry.addClient(clientUrl);
+            expect(registry.clients.has(clientUrl)).toBe(true);
+            registry.removeClient(clientUrl);
+            expect(registry.clients.has(clientUrl)).toBe(false);
+        });
+    });
+    describe("getQueueStatus", () => {
+        it("should return queue status for existing client", async () => {
+            const clientUrl = "http://localhost:8188";
+            registry.addClient(clientUrl);
+            const mockQueue = { queue_running: [], queue_pending: [] };
+            const client = registry.clients.get(clientUrl);
+            jest.spyOn(client.api, 'getQueue').mockResolvedValue(mockQueue);
+            const result = await registry.getQueueStatus(clientUrl);
+            expect(result).toBe(mockQueue);
+        });
+        it("should throw error for non-existing client", async () => {
+            await expect(registry.getQueueStatus("http://nonexistent:8188")).rejects.toThrow("Client http://nonexistent:8188 not found");
+        });
+    });
+    describe("getOptimalClient", () => {
+        it("should return null if no workflow hash", () => {
+            const workflow = createWorkflowMock();
+            workflow.updateHash.mockImplementation(() => {
+                // no hash
+            });
+            expect(() => registry.getOptimalClient(workflow)).toThrow("Workflow must have a valid structure hash.");
+        });
+        it("should return null if no suitable idle clients with affinity", () => {
+            const workflow = createWorkflowMock("hash1");
+            registry.addClient("http://client1:8188", { workflowAffinity: [workflow] });
+            registry.clients.get("http://client1:8188").state = "busy";
+            const result = registry.getOptimalClient(workflow);
+            expect(result).toBeNull();
+            expect(loggerMock.debug).toHaveBeenCalledWith("No suitable clients found for workflow hash1.");
+        });
+        it("should return the highest priority idle client with affinity", () => {
+            const workflow = createWorkflowMock("hash1");
+            registry.addClient("http://client1:8188", { workflowAffinity: [workflow], priority: 1 });
+            registry.addClient("http://client2:8188", { workflowAffinity: [workflow], priority: 3 });
+            registry.addClient("http://client3:8188", { workflowAffinity: [workflow], priority: 2 });
+            const result = registry.getOptimalClient(workflow);
+            expect(result.url).toBe("http://client2:8188"); // highest priority
+        });
+        it("should return idle client even if no affinity specified", () => {
+            const workflow = createWorkflowMock("hash1");
+            registry.addClient("http://client1:8188"); // no affinity
+            registry.clients.get("http://client1:8188").workflowAffinity = undefined; // ensure no affinity
+            // Since no affinity, suitableClients will be empty
+            const result = registry.getOptimalClient(workflow);
+            expect(result).toBeNull();
+        });
+    });
+    describe("hasClientsForWorkflow", () => {
+        it("should return true if clients have affinity for workflow", () => {
+            const workflow = createWorkflowMock("hash1");
+            registry.addClient("http://client1:8188", { workflowAffinity: [workflow] });
+            expect(registry.hasClientsForWorkflow("hash1")).toBe(true);
+        });
+        it("should return false if no clients have affinity", () => {
+            expect(registry.hasClientsForWorkflow("hash1")).toBe(false);
+        });
+    });
+    describe("getOptimalIdleClient", () => {
+        it("should return the highest priority idle client", async () => {
+            registry.addClient("http://client1:8188", { priority: 1 });
+            registry.addClient("http://client2:8188", { priority: 3 });
+            registry.addClient("http://client3:8188", { priority: 2 });
+            // Mock getQueue to return empty for all clients
+            for (const client of registry.clients.values()) {
+                jest.spyOn(client.api, 'getQueue').mockResolvedValue({ queue_running: [], queue_pending: [] });
+            }
+            const result = await registry.getOptimalIdleClient(createWorkflowMock());
+            expect(result.url).toBe("http://client2:8188");
+        });
+        it("should return null if no idle clients", async () => {
+            registry.addClient("http://client1:8188");
+            registry.clients.get("http://client1:8188").state = "busy";
+            const result = await registry.getOptimalIdleClient(createWorkflowMock());
+            expect(result).toBeNull();
+        });
+        it("should check queue state and update client state", async () => {
+            registry.addClient("http://client1:8188");
+            const client = registry.clients.get("http://client1:8188");
+            jest.spyOn(client.api, 'getQueue').mockResolvedValue({ queue_running: [{}], queue_pending: [] });
+            const result = await registry.getOptimalIdleClient(createWorkflowMock());
+            expect(client.state).toBe("busy");
+            expect(result).toBeNull();
+        });
+    });
+    describe("markClientIncompatibleWithWorkflow", () => {
+        it("should remove affinity for the workflow", () => {
+            const workflow = createWorkflowMock("hash1");
+            registry.addClient("http://client1:8188", { workflowAffinity: [workflow] });
+            registry.markClientIncompatibleWithWorkflow("http://client1:8188", "hash1");
+            const client = registry.clients.get("http://client1:8188");
+            expect(client.workflowAffinity.has("hash1")).toBe(false);
+            expect(registry.workflowAffinityMap.has("hash1")).toBe(false);
+        });
+    });
+    describe("getAllEligibleClientsForWorkflow", () => {
+        it("should return clients with affinity for the workflow", () => {
+            const workflow = createWorkflowMock("hash1");
+            registry.addClient("http://client1:8188", { workflowAffinity: [workflow] });
+            registry.addClient("http://client2:8188");
+            const result = registry.getAllEligibleClientsForWorkflow(workflow);
+            expect(result).toHaveLength(1);
+            expect(result[0].url).toBe("http://client1:8188");
+        });
+        it("should throw if workflow has no hash", () => {
+            const workflow = createWorkflowMock();
+            workflow.updateHash.mockImplementation(() => { });
+            expect(() => registry.getAllEligibleClientsForWorkflow(workflow)).toThrow("Workflow must have a valid structure hash.");
+        });
+    });
+});
+//# sourceMappingURL=client-registry.spec.js.map

package/dist/multipool/tests/client-registry.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client-registry.spec.js","sourceRoot":"","sources":["../../../src/multipool/tests/client-registry.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,UAAU,CAAC;AAClE,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAIvD,oBAAoB;AACpB,MAAM,cAAc,GAAG,GAAG,EAAE,CAAC,CAAC,EAAE,CAAQ,CAAC;AAEzC,MAAM,gBAAgB,GAAG,GAAG,EAAE,CAAC,CAAC;IAC9B,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;CACjB,CAAQ,CAAC;AAEV,MAAM,kBAAkB,GAAG,CAAC,IAAa,EAAE,EAAE;IAC3C,MAAM,IAAI,GAAG;QACX,aAAa,EAAE,IAAI;QACnB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;KACtB,CAAC;IACF,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,GAAG,EAAE;QACtC,IAAI,CAAC,aAAa,GAAG,WAAW,CAAC;IACnC,CAAC,CAAC,CAAC;IACH,OAAO,IAAW,CAAC;AACrB,CAAC,CAAC;AAEF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,QAA2B,CAAC;IAChC,IAAI,UAAkB,CAAC;IACvB,IAAI,QAAwB,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,cAAc,EAAE,CAAC;QAC5B,UAAU,GAAG,gBAAgB,EAAE,CAAC;QAChC,QAAQ,GAAG,IAAI,cAAc,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QACpD,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAE9B,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,aAAa,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAS,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAS,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAClF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC,CAAC,UAAU;YACjD,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAEhE,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC/C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;YACtC,QAAQ,CAAC,UAAU,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC1C,iBAAiB;YACnB,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,yDAAyD,CAAC,CAAC;QACnJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC9B,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEnD,QAAQ,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;YACjC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC9B,MAAM,SAAS,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC;YAC3D,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;YAChD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,0CAA0C,CAAC,CAAC;QAC/H,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;YACtC,QAAQ,CAAC,UAAU,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC1C,UAAU;YACZ,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAC;QAC1G,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC5E,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAE,CAAC,KAAK,GAAG,MAAM,CAAC;YAE5D,MAAM,MAAM,GAAG,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC1B,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,+CAA+C,CAAC,CAAC;QACjG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YACzF,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YACzF,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAEzF,MAAM,MAAM,GAAG,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,CAAC,MAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,mBAAmB;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC,CAAC,cAAc;YACzD,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAE,CAAC,gBAAgB,GAAG,SAAS,CAAC,CAAC,qBAAqB;YAEhG,mDAAmD;YACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAE5E,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAS,CAAC,CAAC;YAClE,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAS,CAAC,CAAC;YAClE,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAS,CAAC,CAAC;YAElE,gDAAgD;YAChD,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC/C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;YACjG,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,oBAAoB,CAAC,kBAAkB,EAAE,CAAC,CAAC;YACzE,MAAM,CAAC,MAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,QAAQ,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YAC1C,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAE,CAAC,KAAK,GAAG,MAAM,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,oBAAoB,CAAC,kBAAkB,EAAE,CAAC,CAAC;YACzE,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,QAAQ,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YAC1C,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAE,CAAC;YAC5D,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,EAAE,aAAa,EAAE,CAAC,EAAS,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;YAExG,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,oBAAoB,CAAC,kBAAkB,EAAE,CAAC,CAAC;YACzE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAClD,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAS,CAAC,CAAC;YAEnF,QAAQ,CAAC,kCAAkC,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;YAE5E,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAE,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,gBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1D,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,QAAQ,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC7C,QAAQ,CAAC,SAAS,CAAC,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,QAAQ,CAAC,EAAS,CAAC,CAAC;YACnF,QAAQ,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YAE1C,MAAM,MAAM,GAAG,QAAQ,CAAC,gCAAgC,CAAC,QAAQ,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;YACtC,QAAQ,CAAC,UAAU,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEjD,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAC;QAC1H,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/multipool/tests/error-classification-tests.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=error-classification-tests.d.ts.map

package/dist/multipool/tests/error-classification-tests.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"error-classification-tests.d.ts","sourceRoot":"","sources":["../../../src/multipool/tests/error-classification-tests.ts"],"names":[],"mappings":""}

package/dist/multipool/tests/error-classification-tests.js

@@ -0,0 +1,374 @@
+import { MultiWorkflowPool } from "src/multipool/multi-workflow-pool.js";
+import { Workflow } from "src/multipool/workflow.js";
+/**
+ * Error Classification Validation Tests
+ *
+ * This script validates the error classification logic by intentionally triggering
+ * different types of errors from real ComfyUI servers.
+ *
+ * Test Cases:
+ * 1. Missing Model Error (workflow_incompatibility)
+ * 2. Missing Custom Node Error (workflow_incompatibility)
+ * 3. Invalid Input Error (transient)
+ * 4. Connection Error (connection)
+ */
+const pool = new MultiWorkflowPool();
+// Add your real ComfyUI servers here
+pool.addClient("http://server1:8188", {
+    workflowAffinity: [],
+    priority: 1
+});
+pool.addClient("http://server2:8188", {
+    workflowAffinity: [],
+    priority: 1
+});
+pool.addClient("http://server3:8188", {
+    workflowAffinity: [],
+    priority: 1
+});
+await pool.init();
+console.log("\n" + "=".repeat(80));
+console.log("ERROR CLASSIFICATION VALIDATION TESTS");
+console.log("=".repeat(80) + "\n");
+// ============================================================================
+// TEST 1: Missing Model Error (should be classified as workflow_incompatibility)
+// ============================================================================
+async function testMissingModel() {
+    console.log("\n📋 TEST 1: Missing Model Error");
+    console.log("-".repeat(80));
+    const workflow = new Workflow({
+        "1": {
+            "inputs": {
+                "text": "beautiful landscape"
+            },
+            "class_type": "CLIPTextEncode",
+            "_meta": {
+                "title": "CLIP Text Encode (Prompt)"
+            }
+        },
+        "2": {
+            "inputs": {
+                "ckpt_name": "this_model_definitely_does_not_exist_anywhere_12345.safetensors"
+            },
+            "class_type": "CheckpointLoaderSimple",
+            "_meta": {
+                "title": "Load Checkpoint"
+            }
+        },
+        "3": {
+            "inputs": {
+                "seed": 12345,
+                "steps": 20,
+                "cfg": 7.0,
+                "sampler_name": "euler",
+                "scheduler": "normal",
+                "denoise": 1.0,
+                "model": ["2", 0],
+                "positive": ["1", 0],
+                "negative": ["1", 0],
+                "latent_image": ["4", 0]
+            },
+            "class_type": "KSampler",
+            "_meta": {
+                "title": "KSampler"
+            }
+        },
+        "4": {
+            "inputs": {
+                "width": 512,
+                "height": 512,
+                "batch_size": 1
+            },
+            "class_type": "EmptyLatentImage",
+            "_meta": {
+                "title": "Empty Latent Image"
+            }
+        },
+        "5": {
+            "inputs": {
+                "samples": ["3", 0],
+                "vae": ["2", 2]
+            },
+            "class_type": "VAEDecode",
+            "_meta": {
+                "title": "VAE Decode"
+            }
+        },
+        "6": {
+            "inputs": {
+                "filename_prefix": "ComfyUI",
+                "images": ["5", 0]
+            },
+            "class_type": "SaveImage",
+            "_meta": {
+                "title": "Save Image"
+            }
+        }
+    });
+    try {
+        const jobId = await pool.submitJob(workflow);
+        console.log(`Submitted job: ${jobId}`);
+        const result = await pool.waitForJobCompletion(jobId);
+        if (result.status === "failed") {
+            console.log("✅ Test PASSED: Job failed as expected");
+            console.log(`Error classification should be: workflow_incompatibility`);
+            console.log(`Actual error:`, result.error);
+        }
+        else {
+            console.log("❌ Test FAILED: Job should have failed but didn't");
+        }
+    }
+    catch (error) {
+        console.log("❌ Test ERROR:", error);
+    }
+}
+// ============================================================================
+// TEST 2: Missing Custom Node Error (should be classified as workflow_incompatibility)
+// ============================================================================
+async function testMissingCustomNode() {
+    console.log("\n📋 TEST 2: Missing Custom Node Error");
+    console.log("-".repeat(80));
+    const workflow = new Workflow({
+        "1": {
+            "inputs": {
+                "some_input": "test"
+            },
+            "class_type": "NonExistentCustomNode_XYZ_12345",
+            "_meta": {
+                "title": "This Node Does Not Exist"
+            }
+        },
+        "2": {
+            "inputs": {
+                "filename_prefix": "test",
+                "images": ["1", 0]
+            },
+            "class_type": "SaveImage",
+            "_meta": {
+                "title": "Save Image"
+            }
+        }
+    });
+    try {
+        const jobId = await pool.submitJob(workflow);
+        console.log(`Submitted job: ${jobId}`);
+        const result = await pool.waitForJobCompletion(jobId);
+        if (result.status === "failed") {
+            console.log("✅ Test PASSED: Job failed as expected");
+            console.log(`Error classification should be: workflow_incompatibility`);
+            console.log(`Actual error:`, result.error);
+        }
+        else {
+            console.log("❌ Test FAILED: Job should have failed but didn't");
+        }
+    }
+    catch (error) {
+        console.log("❌ Test ERROR:", error);
+    }
+}
+// ============================================================================
+// TEST 3: Invalid Input Error (should be classified as transient)
+// ============================================================================
+async function testInvalidInput() {
+    console.log("\n📋 TEST 3: Invalid Input Error");
+    console.log("-".repeat(80));
+    const workflow = new Workflow({
+        "1": {
+            "inputs": {
+                "width": -999999, // Negative width should cause validation error
+                "height": -999999, // Negative height should cause validation error
+                "batch_size": 1
+            },
+            "class_type": "EmptyLatentImage",
+            "_meta": {
+                "title": "Empty Latent Image"
+            }
+        },
+        "2": {
+            "inputs": {
+                "filename_prefix": "test",
+                "images": ["1", 0]
+            },
+            "class_type": "SaveImage",
+            "_meta": {
+                "title": "Save Image"
+            }
+        }
+    });
+    try {
+        const jobId = await pool.submitJob(workflow);
+        console.log(`Submitted job: ${jobId}`);
+        const result = await pool.waitForJobCompletion(jobId);
+        if (result.status === "failed") {
+            console.log("✅ Test PASSED: Job failed as expected");
+            console.log(`Error classification should be: transient`);
+            console.log(`Actual error:`, result.error);
+        }
+        else {
+            console.log("❌ Test FAILED: Job should have failed but didn't");
+        }
+    }
+    catch (error) {
+        console.log("❌ Test ERROR:", error);
+    }
+}
+// ============================================================================
+// TEST 4: Connection Error (should be classified as connection)
+// ============================================================================
+async function testConnectionError() {
+    console.log("\n📋 TEST 4: Connection Error");
+    console.log("-".repeat(80));
+    const isolatedPool = new MultiWorkflowPool();
+    // Add a client that definitely doesn't exist
+    isolatedPool.addClient("http://this-server-does-not-exist-12345:8188", {
+        workflowAffinity: [],
+        priority: 1
+    });
+    try {
+        console.log("Attempting to connect to non-existent server...");
+        await isolatedPool.init();
+        console.log("❌ Test FAILED: Should have failed to connect");
+    }
+    catch (error) {
+        console.log("✅ Test PASSED: Connection failed as expected");
+        console.log(`Error classification should be: connection`);
+        console.log(`Actual error:`, error);
+    }
+    finally {
+        await isolatedPool.shutdown();
+    }
+}
+// ============================================================================
+// TEST 5: Missing LoRA Error (should be classified as workflow_incompatibility)
+// ============================================================================
+async function testMissingLoRA() {
+    console.log("\n📋 TEST 5: Missing LoRA Error");
+    console.log("-".repeat(80));
+    const workflow = new Workflow({
+        "1": {
+            "inputs": {
+                "text": "beautiful landscape"
+            },
+            "class_type": "CLIPTextEncode",
+            "_meta": {
+                "title": "CLIP Text Encode"
+            }
+        },
+        "2": {
+            "inputs": {
+                "ckpt_name": "novaAnimeXL_ilV125.safetensors" // Use a model that exists on the servers
+            },
+            "class_type": "CheckpointLoaderSimple",
+            "_meta": {
+                "title": "Load Checkpoint"
+            }
+        },
+        "3": {
+            "inputs": {
+                "lora_name": "this_lora_does_not_exist_12345.safetensors",
+                "strength_model": 1.0,
+                "strength_clip": 1.0,
+                "model": ["2", 0],
+                "clip": ["2", 1]
+            },
+            "class_type": "LoraLoader",
+            "_meta": {
+                "title": "Load LoRA"
+            }
+        },
+        "4": {
+            "inputs": {
+                "width": 512,
+                "height": 512,
+                "batch_size": 1
+            },
+            "class_type": "EmptyLatentImage",
+            "_meta": {
+                "title": "Empty Latent Image"
+            }
+        },
+        "5": {
+            "inputs": {
+                "seed": 12345,
+                "steps": 20,
+                "cfg": 7.0,
+                "sampler_name": "euler",
+                "scheduler": "normal",
+                "denoise": 1.0,
+                "model": ["3", 0],
+                "positive": ["1", 0],
+                "negative": ["1", 0],
+                "latent_image": ["4", 0]
+            },
+            "class_type": "KSampler",
+            "_meta": {
+                "title": "KSampler"
+            }
+        },
+        "6": {
+            "inputs": {
+                "samples": ["5", 0],
+                "vae": ["2", 2]
+            },
+            "class_type": "VAEDecode",
+            "_meta": {
+                "title": "VAE Decode"
+            }
+        },
+        "7": {
+            "inputs": {
+                "filename_prefix": "ComfyUI",
+                "images": ["6", 0]
+            },
+            "class_type": "SaveImage",
+            "_meta": {
+                "title": "Save Image"
+            }
+        }
+    });
+    try {
+        const jobId = await pool.submitJob(workflow);
+        console.log(`Submitted job: ${jobId}`);
+        const result = await pool.waitForJobCompletion(jobId);
+        if (result.status === "failed") {
+            console.log("✅ Test PASSED: Job failed as expected");
+            console.log(`Error classification should be: workflow_incompatibility`);
+            console.log(`Actual error:`, result.error);
+        }
+        else {
+            console.log("❌ Test FAILED: Job should have failed but didn't");
+        }
+    }
+    catch (error) {
+        console.log("❌ Test ERROR:", error);
+    }
+}
+// ============================================================================
+// Run all tests
+// ============================================================================
+async function runAllTests() {
+    try {
+        await testMissingModel();
+        await new Promise(resolve => setTimeout(resolve, 2000));
+        await testMissingCustomNode();
+        await new Promise(resolve => setTimeout(resolve, 2000));
+        await testInvalidInput();
+        await new Promise(resolve => setTimeout(resolve, 2000));
+        await testMissingLoRA();
+        await new Promise(resolve => setTimeout(resolve, 2000));
+        await testConnectionError();
+        console.log("\n" + "=".repeat(80));
+        console.log("ALL TESTS COMPLETED");
+        console.log("=".repeat(80) + "\n");
+    }
+    catch (error) {
+        console.error("Fatal error during tests:", error);
+    }
+    finally {
+        await pool.shutdown();
+        process.exit(0);
+    }
+}
+// Run the tests
+runAllTests();
+//# sourceMappingURL=error-classification-tests.js.map

package/dist/multipool/tests/error-classification-tests.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"error-classification-tests.js","sourceRoot":"","sources":["../../../src/multipool/tests/error-classification-tests.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAErD;;;;;;;;;;;GAWG;AAEH,MAAM,IAAI,GAAG,IAAI,iBAAiB,EAAE,CAAC;AAErC,qCAAqC;AACrC,IAAI,CAAC,SAAS,CAAC,qBAAqB,EAAE;IACpC,gBAAgB,EAAE,EAAE;IACpB,QAAQ,EAAE,CAAC;CACZ,CAAC,CAAC;AAEH,IAAI,CAAC,SAAS,CAAC,qBAAqB,EAAE;IACpC,gBAAgB,EAAE,EAAE;IACpB,QAAQ,EAAE,CAAC;CACZ,CAAC,CAAC;AAEH,IAAI,CAAC,SAAS,CAAC,qBAAqB,EAAE;IACpC,gBAAgB,EAAE,EAAE;IACpB,QAAQ,EAAE,CAAC;CACZ,CAAC,CAAC;AAEH,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;AAElB,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;AACnC,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;AACrD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;AAEnC,+EAA+E;AAC/E,iFAAiF;AACjF,+EAA+E;AAC/E,KAAK,UAAU,gBAAgB;IAC7B,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;QAC5B,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,MAAM,EAAE,qBAAqB;aAC9B;YACD,YAAY,EAAE,gBAAgB;YAC9B,OAAO,EAAE;gBACP,OAAO,EAAE,2BAA2B;aACrC;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,WAAW,EAAE,iEAAiE;aAC/E;YACD,YAAY,EAAE,wBAAwB;YACtC,OAAO,EAAE;gBACP,OAAO,EAAE,iBAAiB;aAC3B;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,GAAG;gBACV,cAAc,EAAE,OAAO;gBACvB,WAAW,EAAE,QAAQ;gBACrB,SAAS,EAAE,GAAG;gBACd,OAAO,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACjB,UAAU,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACpB,UAAU,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACpB,cAAc,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aACzB;YACD,YAAY,EAAE,UAAU;YACxB,OAAO,EAAE;gBACP,OAAO,EAAE,UAAU;aACpB;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE,GAAG;gBACb,YAAY,EAAE,CAAC;aAChB;YACD,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE;gBACP,OAAO,EAAE,oBAAoB;aAC9B;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,SAAS,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACnB,KAAK,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aAChB;YACD,YAAY,EAAE,WAAW;YACzB,OAAO,EAAE;gBACP,OAAO,EAAE,YAAY;aACtB;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,iBAAiB,EAAE,SAAS;gBAC5B,QAAQ,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aACnB;YACD,YAAY,EAAE,WAAW;YACzB,OAAO,EAAE;gBACP,OAAO,EAAE,YAAY;aACtB;SACF;KACF,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEtD,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;YACrD,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,uFAAuF;AACvF,+EAA+E;AAC/E,KAAK,UAAU,qBAAqB;IAClC,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;QAC5B,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,YAAY,EAAE,MAAM;aACrB;YACD,YAAY,EAAE,iCAAiC;YAC/C,OAAO,EAAE;gBACP,OAAO,EAAE,0BAA0B;aACpC;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,iBAAiB,EAAE,MAAM;gBACzB,QAAQ,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aACnB;YACD,YAAY,EAAE,WAAW;YACzB,OAAO,EAAE;gBACP,OAAO,EAAE,YAAY;aACtB;SACF;KACF,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEtD,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;YACrD,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,kEAAkE;AAClE,+EAA+E;AAC/E,KAAK,UAAU,gBAAgB;IAC7B,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;QAC5B,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,OAAO,EAAE,CAAC,MAAM,EAAG,+CAA+C;gBAClE,QAAQ,EAAE,CAAC,MAAM,EAAG,gDAAgD;gBACpE,YAAY,EAAE,CAAC;aAChB;YACD,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE;gBACP,OAAO,EAAE,oBAAoB;aAC9B;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,iBAAiB,EAAE,MAAM;gBACzB,QAAQ,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aACnB;YACD,YAAY,EAAE,WAAW;YACzB,OAAO,EAAE;gBACP,OAAO,EAAE,YAAY;aACtB;SACF;KACF,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEtD,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;YACrD,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;YACzD,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,gEAAgE;AAChE,+EAA+E;AAC/E,KAAK,UAAU,mBAAmB;IAChC,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;IAE7C,6CAA6C;IAC7C,YAAY,CAAC,SAAS,CAAC,8CAA8C,EAAE;QACrE,gBAAgB,EAAE,EAAE;QACpB,QAAQ,EAAE,CAAC;KACZ,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;QAC/D,MAAM,YAAY,CAAC,IAAI,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;IAC9D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;QAC5D,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;QAC1D,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC;YAAS,CAAC;QACT,MAAM,YAAY,CAAC,QAAQ,EAAE,CAAC;IAChC,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,gFAAgF;AAChF,+EAA+E;AAC/E,KAAK,UAAU,eAAe;IAC5B,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IAC/C,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;QAC5B,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,MAAM,EAAE,qBAAqB;aAC9B;YACD,YAAY,EAAE,gBAAgB;YAC9B,OAAO,EAAE;gBACP,OAAO,EAAE,kBAAkB;aAC5B;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,WAAW,EAAE,gCAAgC,CAAE,yCAAyC;aACzF;YACD,YAAY,EAAE,wBAAwB;YACtC,OAAO,EAAE;gBACP,OAAO,EAAE,iBAAiB;aAC3B;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,WAAW,EAAE,4CAA4C;gBACzD,gBAAgB,EAAE,GAAG;gBACrB,eAAe,EAAE,GAAG;gBACpB,OAAO,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACjB,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aACjB;YACD,YAAY,EAAE,YAAY;YAC1B,OAAO,EAAE;gBACP,OAAO,EAAE,WAAW;aACrB;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE,GAAG;gBACb,YAAY,EAAE,CAAC;aAChB;YACD,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE;gBACP,OAAO,EAAE,oBAAoB;aAC9B;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,GAAG;gBACV,cAAc,EAAE,OAAO;gBACvB,WAAW,EAAE,QAAQ;gBACrB,SAAS,EAAE,GAAG;gBACd,OAAO,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACjB,UAAU,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACpB,UAAU,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACpB,cAAc,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aACzB;YACD,YAAY,EAAE,UAAU;YACxB,OAAO,EAAE;gBACP,OAAO,EAAE,UAAU;aACpB;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,SAAS,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACnB,KAAK,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aAChB;YACD,YAAY,EAAE,WAAW;YACzB,OAAO,EAAE;gBACP,OAAO,EAAE,YAAY;aACtB;SACF;QACD,GAAG,EAAE;YACH,QAAQ,EAAE;gBACR,iBAAiB,EAAE,SAAS;gBAC5B,QAAQ,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;aACnB;YACD,YAAY,EAAE,WAAW;YACzB,OAAO,EAAE;gBACP,OAAO,EAAE,YAAY;aACtB;SACF;KACF,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEtD,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;YACrD,OAAO,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,gBAAgB;AAChB,+EAA+E;AAC/E,KAAK,UAAU,WAAW;IACxB,IAAI,CAAC;QACH,MAAM,gBAAgB,EAAE,CAAC;QACzB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QAExD,MAAM,qBAAqB,EAAE,CAAC;QAC9B,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QAExD,MAAM,gBAAgB,EAAE,CAAC;QACzB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QAExD,MAAM,eAAe,EAAE,CAAC;QACxB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QAExD,MAAM,mBAAmB,EAAE,CAAC;QAE5B,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IAErC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC;YAAS,CAAC;QACT,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACtB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,gBAAgB;AAChB,WAAW,EAAE,CAAC"}

package/dist/multipool/tests/event-forwarding-demo.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Event Forwarding Demo for MultiWorkflowPool
+ *
+ * Demonstrates that all ComfyUI client events are forwarded through PoolEventManager
+ */
+export {};
+//# sourceMappingURL=event-forwarding-demo.d.ts.map

package/dist/multipool/tests/event-forwarding-demo.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"event-forwarding-demo.d.ts","sourceRoot":"","sources":["../../../src/multipool/tests/event-forwarding-demo.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}