comfyui-node 1.6.1 → 1.6.3

package/dist/multipool/tests/event-forwarding-demo.js

@@ -0,0 +1,88 @@
+/**
+ * Event Forwarding Demo for MultiWorkflowPool
+ *
+ * Demonstrates that all ComfyUI client events are forwarded through PoolEventManager
+ */
+import { MultiWorkflowPool } from "../multi-workflow-pool.js";
+import { Workflow } from "../workflow.js";
+import GenerationGraph from "../../../scripts/workflows/T2I-one-obsession.json" with { type: "json" };
+const GEN_HOST = "http://localhost:8188";
+// Create pool
+const pool = new MultiWorkflowPool({
+    logLevel: "warn" // Keep logs minimal to see events clearly
+});
+const genWorkflow = Workflow.fromAugmented(GenerationGraph);
+pool.addClient(GEN_HOST, {
+    workflowAffinity: [genWorkflow],
+    priority: 1
+});
+console.log("\n" + "=".repeat(80));
+console.log("EVENT FORWARDING DEMO - MultiWorkflowPool");
+console.log("=".repeat(80));
+console.log(`Generation Host: ${GEN_HOST}`);
+console.log("Testing: All client events forwarded through PoolEventManager");
+console.log("=".repeat(80) + "\n");
+// Track all events received
+const eventLog = [];
+// Attach event hooks for various client events
+const trackedEvents = [
+    "client:status",
+    "client:progress",
+    "client:executing",
+    "client:execution_cached",
+    "client:executed",
+    "client:execution_success",
+    "client:b_preview_meta"
+];
+trackedEvents.forEach(eventType => {
+    pool.attachEventHook(eventType, (event) => {
+        eventLog.push({
+            type: event.type,
+            clientName: event.payload.clientName,
+            eventType: event.payload.eventType
+        });
+        console.log(`[Event Hook] ${event.type} from ${event.payload.clientName}`);
+    });
+});
+console.log(`✅ Attached event hooks for: ${trackedEvents.join(", ")}\n`);
+await pool.init();
+console.log("\n📤 Submitting job to trigger events...\n");
+// Run a simple generation job
+const workflow = Workflow.fromAugmented(GenerationGraph)
+    .input("1", "value", "test image, simple")
+    .input("10", "steps", 5) // Low steps for faster execution
+    .input("10", "seed", 42);
+const jobId = await pool.submitJob(workflow);
+await pool.waitForJobCompletion(jobId);
+console.log("\n" + "=".repeat(80));
+console.log("JOB COMPLETED - EVENT LOG SUMMARY");
+console.log("=".repeat(80));
+console.log(`\n📊 Total Events Captured: ${eventLog.length}\n`);
+// Group events by type
+const eventsByType = new Map();
+eventLog.forEach(log => {
+    eventsByType.set(log.type, (eventsByType.get(log.type) || 0) + 1);
+});
+console.log("Event Type Breakdown:");
+eventsByType.forEach((count, type) => {
+    console.log(`  ${type}: ${count} events`);
+});
+// Verify critical events were captured
+const criticalEvents = ["client:status", "client:execution_success"];
+const missingEvents = criticalEvents.filter(e => !eventsByType.has(e));
+if (missingEvents.length > 0) {
+    console.log(`\n⚠️  WARNING: Missing critical events: ${missingEvents.join(", ")}`);
+}
+else {
+    console.log("\n✅ All critical events were forwarded successfully!");
+}
+// Show detailed event timeline
+console.log("\n📅 Event Timeline:");
+eventLog.forEach((log, i) => {
+    console.log(`  ${i + 1}. ${log.type} (${log.eventType}) from ${log.clientName}`);
+});
+console.log("\n" + "=".repeat(80));
+await pool.shutdown();
+console.log("\n✅ Event forwarding demo completed, exiting...");
+process.exit(0);
+//# sourceMappingURL=event-forwarding-demo.js.map

package/dist/multipool/tests/event-forwarding-demo.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"event-forwarding-demo.js","sourceRoot":"","sources":["../../../src/multipool/tests/event-forwarding-demo.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,eAAe,MAAM,mDAAmD,CAAC,OAAO,IAAI,EAAE,MAAM,EAAE,CAAC;AAEtG,MAAM,QAAQ,GAAG,uBAAuB,CAAC;AAEzC,cAAc;AACd,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAAC;IACjC,QAAQ,EAAE,MAAM,CAAE,0CAA0C;CAC7D,CAAC,CAAC;AAEH,MAAM,WAAW,GAAG,QAAQ,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;AAE5D,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE;IACvB,gBAAgB,EAAE,CAAC,WAAW,CAAC;IAC/B,QAAQ,EAAE,CAAC;CACZ,CAAC,CAAC;AAEH,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;AACnC,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;AACzD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;AAC5B,OAAO,CAAC,GAAG,CAAC,oBAAoB,QAAQ,EAAE,CAAC,CAAC;AAC5C,OAAO,CAAC,GAAG,CAAC,+DAA+D,CAAC,CAAC;AAC7E,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;AAEnC,4BAA4B;AAC5B,MAAM,QAAQ,GAAmE,EAAE,CAAC;AAEpF,+CAA+C;AAC/C,MAAM,aAAa,GAAG;IACpB,eAAe;IACf,iBAAiB;IACjB,kBAAkB;IAClB,yBAAyB;IACzB,iBAAiB;IACjB,0BAA0B;IAC1B,uBAAuB;CACxB,CAAC;AAEF,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;IAChC,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;QACxC,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,UAAU,EAAE,KAAK,CAAC,OAAO,CAAC,UAAU;YACpC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,SAAS;SACnC,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,gBAAgB,KAAK,CAAC,IAAI,SAAS,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,GAAG,CAAC,+BAA+B,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEzE,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;AAElB,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;AAE1D,8BAA8B;AAC9B,MAAM,QAAQ,GAAG,QAAQ,CAAC,aAAa,CAAC,eAAe,CAAC;KACrD,KAAK,CAAC,GAAG,EAAE,OAAO,EAAE,oBAAoB,CAAC;KACzC,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAE,iCAAiC;KAC1D,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;AAE3B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;AAC7C,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;AAEvC,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;AACnC,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;AACjD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;AAE5B,OAAO,CAAC,GAAG,CAAC,+BAA+B,QAAQ,CAAC,MAAM,IAAI,CAAC,CAAC;AAEhE,uBAAuB;AACvB,MAAM,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;AAC/C,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;IACrB,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AACpE,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;AACrC,YAAY,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;IACnC,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,KAAK,KAAK,SAAS,CAAC,CAAC;AAC5C,CAAC,CAAC,CAAC;AAEH,uCAAuC;AACvC,MAAM,cAAc,GAAG,CAAC,eAAe,EAAE,0BAA0B,CAAC,CAAC;AACrE,MAAM,aAAa,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AAEvE,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;IAC7B,OAAO,CAAC,GAAG,CAAC,2CAA2C,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACrF,CAAC;KAAM,CAAC;IACN,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;AACtE,CAAC;AAED,+BAA+B;AAC/B,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;AACpC,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;IAC1B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,SAAS,UAAU,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;AACnF,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;AAEnC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;AAEtB,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;AAC/D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC"}

package/dist/multipool/tests/helpers.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=helpers.spec.d.ts.map

package/dist/multipool/tests/helpers.spec.d.ts.map

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

package/dist/multipool/tests/helpers.spec.js

@@ -0,0 +1,100 @@
+import { describe, it, expect } from "bun:test";
+import { classifyFailure } from "../helpers.js";
+describe("classifyFailure", () => {
+    // Test for Connection Failure
+    it("should classify connection errors", () => {
+        const connectionError = new Error("Connection timed out");
+        expect(classifyFailure(connectionError)).toEqual({
+            type: "connection",
+            message: "Connection timed out",
+        });
+        const anotherConnectionError = { message: "connect ECONNREFUSED 127.0.0.1:8188" };
+        expect(classifyFailure(anotherConnectionError)).toEqual({
+            type: "connection",
+            message: "connect ECONNREFUSED 127.0.0.1:8188",
+        });
+    });
+    // Test for Workflow Incompatibility - Missing Model
+    it("should classify missing model errors as workflow_incompatibility", () => {
+        const missingModelError = {
+            bodyJSON: {
+                node_errors: {
+                    "2": {
+                        errors: [
+                            {
+                                type: "value_not_in_list",
+                                details: "Value not in list: ckpt_name 'this_model_does_not_exist.safetensors' not in [...]",
+                            },
+                        ],
+                    },
+                },
+            },
+        };
+        const classification = classifyFailure(missingModelError);
+        expect(classification.type).toBe("workflow_incompatibility");
+        expect(classification.message).toContain("Missing model file");
+    });
+    // Test for Workflow Incompatibility - Missing Custom Node
+    it("should classify missing custom node errors as workflow_incompatibility", () => {
+        const missingNodeError = {
+            bodyJSON: {
+                error: {
+                    type: "invalid_prompt",
+                    message: "Node type not found: NonExistentNode",
+                },
+                node_errors: {},
+            },
+        };
+        const classification = classifyFailure(missingNodeError);
+        expect(classification.type).toBe("workflow_incompatibility");
+    });
+    it('should classify python exception for missing nodes as workflow_incompatibility', () => {
+        const pythonError = {
+            bodyJSON: {
+                "exception_type": "Exception",
+                "exception_message": "Node type not found: ImpactWildcardEncode",
+                "traceback": "..."
+            }
+        };
+        const classification = classifyFailure(pythonError);
+        expect(classification.type).toBe('workflow_incompatibility');
+        expect(classification.message).toBe('Node type not found: ImpactWildcardEncode');
+    });
+    // Test for Transient Failure - Out of Memory
+    it("should classify OOM errors as transient", () => {
+        const oomError = {
+            bodyJSON: {
+                exception_message: "CUDA out of memory. Tried to allocate...",
+            },
+        };
+        expect(classifyFailure(oomError)).toEqual({
+            type: "transient",
+            message: "CUDA out of memory",
+        });
+    });
+    // Test for Transient Failure - Other exceptions
+    it("should classify other exceptions as transient", () => {
+        const genericError = {
+            bodyJSON: {
+                exception_message: "Something else went wrong",
+            },
+        };
+        expect(classifyFailure(genericError)).toEqual({
+            type: "transient",
+            message: "Something else went wrong",
+        });
+    });
+    // Test for Default Case
+    it("should default to transient for unknown errors with a response body", () => {
+        const unknownError = {
+            bodyJSON: {
+                some_unusual_error: "details here",
+            },
+        };
+        expect(classifyFailure(unknownError)).toEqual({
+            type: "transient",
+            message: '{"some_unusual_error":"details here"}',
+        });
+    });
+});
+//# sourceMappingURL=helpers.spec.js.map

package/dist/multipool/tests/helpers.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"helpers.spec.js","sourceRoot":"","sources":["../../../src/multipool/tests/helpers.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAEhD,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,8BAA8B;IAC9B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,eAAe,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1D,MAAM,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,sBAAsB;SAChC,CAAC,CAAC;QAEH,MAAM,sBAAsB,GAAG,EAAE,OAAO,EAAE,qCAAqC,EAAE,CAAC;QAClF,MAAM,CAAC,eAAe,CAAC,sBAAsB,CAAC,CAAC,CAAC,OAAO,CAAC;YACtD,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,qCAAqC;SAC/C,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,oDAAoD;IACpD,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,iBAAiB,GAAG;YACxB,QAAQ,EAAE;gBACR,WAAW,EAAE;oBACX,GAAG,EAAE;wBACH,MAAM,EAAE;4BACN;gCACE,IAAI,EAAE,mBAAmB;gCACzB,OAAO,EAAE,mFAAmF;6BAC7F;yBACF;qBACF;iBACF;aACF;SACF,CAAC;QACF,MAAM,cAAc,GAAG,eAAe,CAAC,iBAAiB,CAAC,CAAC;QAC1D,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAC7D,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,0DAA0D;IAC1D,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;QAChF,MAAM,gBAAgB,GAAG;YACvB,QAAQ,EAAE;gBACR,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAgB;oBACtB,OAAO,EAAE,sCAAsC;iBAChD;gBACD,WAAW,EAAE,EAAE;aAChB;SACF,CAAC;QACF,MAAM,cAAc,GAAG,eAAe,CAAC,gBAAgB,CAAC,CAAC;QACzD,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,WAAW,GAAG;YAChB,QAAQ,EAAE;gBACN,gBAAgB,EAAE,WAAW;gBAC7B,mBAAmB,EAAE,2CAA2C;gBAChE,WAAW,EAAE,KAAK;aACrB;SACJ,CAAC;QACF,MAAM,cAAc,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;QACpD,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAC7D,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;IAGH,6CAA6C;IAC7C,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE;gBACR,iBAAiB,EAAE,0CAA0C;aAC9D;SACF,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YACxC,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,oBAAoB;SAC9B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,gDAAgD;IAChD,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,YAAY,GAAG;YACnB,QAAQ,EAAE;gBACR,iBAAiB,EAAE,2BAA2B;aAC/C;SACF,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC5C,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,2BAA2B;SACrC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,wBAAwB;IACxB,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,YAAY,GAAG;YACnB,QAAQ,EAAE;gBACR,kBAAkB,EAAE,cAAc;aACnC;SACF,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC5C,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,uCAAuC;SACjD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/multipool/tests/job-queue-processor.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=job-queue-processor.spec.d.ts.map

package/dist/multipool/tests/job-queue-processor.spec.d.ts.map

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

package/dist/multipool/tests/job-queue-processor.spec.js

@@ -0,0 +1,89 @@
+import { describe, it, expect, beforeEach, jest } from "bun:test";
+import { JobQueueProcessor } from "../job-queue-processor.js";
+import { Workflow } from "../workflow.js";
+import { Logger } from "../logger.js";
+// Mock dependencies
+const createJobStateRegistryMock = () => ({
+    getJobStatus: jest.fn().mockReturnValue("pending"),
+    setJobStatus: jest.fn(),
+    setPromptId: jest.fn(),
+    updateJobAutoSeeds: jest.fn(),
+    setJobFailure: jest.fn(),
+});
+const createClientRegistryMock = () => ({
+    getOptimalClient: jest.fn(),
+    getOptimalIdleClient: jest.fn(),
+    clients: new Map(),
+    markClientIncompatibleWithWorkflow: jest.fn(),
+    getAllEligibleClientsForWorkflow: jest.fn().mockReturnValue([]),
+});
+const createLoggerMock = () => new Logger("test", "silent");
+describe("JobQueueProcessor", () => {
+    let jobStateRegistryMock;
+    let clientRegistryMock;
+    let loggerMock;
+    beforeEach(() => {
+        jobStateRegistryMock = createJobStateRegistryMock();
+        clientRegistryMock = createClientRegistryMock();
+        loggerMock = createLoggerMock();
+    });
+    const createProcessor = (hash = "test-hash") => new JobQueueProcessor(jobStateRegistryMock, clientRegistryMock, hash, loggerMock);
+    it("should enqueue a job and trigger processing", async () => {
+        const processor = createProcessor();
+        const processQueueSpy = jest.spyOn(processor, "processQueue").mockImplementation(async () => { });
+        await processor.enqueueJob("job-1", new Workflow({}));
+        expect(processor.queue).toHaveLength(1);
+        expect(processor.queue[0].jobId).toBe("job-1");
+        expect(processQueueSpy).toHaveBeenCalled();
+    });
+    it("should not process queue if already processing", async () => {
+        const processor = createProcessor();
+        processor.isProcessing = true;
+        const loggerSpy = jest.spyOn(loggerMock, 'debug');
+        await processor.processQueue();
+        expect(loggerSpy).toHaveBeenCalledWith(`Job queue for workflow hash test-hash is already being processed, skipping.`);
+    });
+    it("should assign a job to an optimal client and run it successfully", async () => {
+        const processor = createProcessor();
+        const workflow = new Workflow({});
+        workflow.uploadAssets = jest.fn().mockResolvedValue(undefined);
+        const jobId = "job-1";
+        const clientMock = {
+            url: "http://localhost:8188",
+            nodeName: "test-client",
+            state: "idle",
+            api: {
+                getQueue: jest.fn().mockResolvedValue({ queue_running: [], queue_pending: [] }),
+                ext: { queue: { queuePrompt: jest.fn().mockResolvedValue({ prompt_id: "prompt-1" }) } },
+            },
+        };
+        clientRegistryMock.getOptimalClient.mockReturnValue(clientMock);
+        processor.queue.push({ jobId, workflow, attempts: 1 });
+        await processor.processQueue();
+        expect(clientRegistryMock.getOptimalClient).toHaveBeenCalledWith(workflow);
+        expect(jobStateRegistryMock.setJobStatus).toHaveBeenCalledWith(jobId, "assigned", clientMock.url);
+        expect(workflow.uploadAssets).toHaveBeenCalledWith(clientMock.api);
+        expect(clientMock.api.ext.queue.queuePrompt).toHaveBeenCalled();
+        expect(jobStateRegistryMock.setPromptId).toHaveBeenCalledWith(jobId, "prompt-1");
+        expect(jobStateRegistryMock.setJobStatus).toHaveBeenCalledWith(jobId, "running");
+        expect(clientMock.state).toBe("busy");
+    });
+    it("should re-queue a job if no idle clients are available", async () => {
+        const processor = createProcessor();
+        const workflow = new Workflow({});
+        const jobId = "job-1";
+        clientRegistryMock.getOptimalClient.mockReturnValue(null);
+        processor.queue.push({ jobId, workflow, attempts: 1 });
+        await processor.processQueue();
+        expect(jobStateRegistryMock.setJobStatus).toHaveBeenCalledWith(jobId, "pending");
+        expect(processor.queue).toHaveLength(1);
+        expect(processor.isProcessing).toBe(false);
+    });
+    it("should dequeue a job", () => {
+        const processor = createProcessor();
+        processor.queue = [{ jobId: "job-1", workflow: new Workflow({}), attempts: 1 }];
+        processor.dequeueJob("job-1");
+        expect(processor.queue).toHaveLength(0);
+    });
+});
+//# sourceMappingURL=job-queue-processor.spec.js.map

package/dist/multipool/tests/job-queue-processor.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"job-queue-processor.spec.js","sourceRoot":"","sources":["../../../src/multipool/tests/job-queue-processor.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,UAAU,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAE9D,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,oBAAoB;AACpB,MAAM,0BAA0B,GAAG,GAAG,EAAE,CAAC,CAAC;IACxC,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;IAClD,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;IACvB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;IACtB,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC7B,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;CACzB,CAAQ,CAAC;AAEV,MAAM,wBAAwB,GAAG,GAAG,EAAE,CAAC,CAAC;IACtC,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC/B,OAAO,EAAE,IAAI,GAAG,EAAE;IAClB,kCAAkC,EAAE,IAAI,CAAC,EAAE,EAAE;IAC7C,gCAAgC,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;CAChE,CAAQ,CAAC;AAEV,MAAM,gBAAgB,GAAG,GAAG,EAAE,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;AAE5D,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,oBAAsC,CAAC;IAC3C,IAAI,kBAAuB,CAAC;IAC5B,IAAI,UAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,oBAAoB,GAAG,0BAA0B,EAAE,CAAC;QACpD,kBAAkB,GAAG,wBAAwB,EAAE,CAAC;QAChD,UAAU,GAAG,gBAAgB,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,MAAM,eAAe,GAAG,CAAC,IAAI,GAAG,WAAW,EAAE,EAAE,CAC7C,IAAI,iBAAiB,CAAC,oBAAoB,EAAE,kBAAkB,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;IAEpF,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;QACpC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC,CAAC;QAEjG,MAAM,SAAS,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QAEtD,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/C,MAAM,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;QACpC,SAAS,CAAC,YAAY,GAAG,IAAI,CAAC;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAElD,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC;QAE/B,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,6EAA6E,CAAC,CAAC;IACxH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClC,QAAQ,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC/D,MAAM,KAAK,GAAG,OAAO,CAAC;QACtB,MAAM,UAAU,GAAG;YACjB,GAAG,EAAE,uBAAuB;YAC5B,QAAQ,EAAE,aAAa;YACvB,KAAK,EAAE,MAAM;YACb,GAAG,EAAE;gBACH,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC;gBAC/E,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE;aACxF;SACF,CAAC;QAEF,kBAAkB,CAAC,gBAAgB,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QAChE,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAEvD,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC;QAE/B,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAC3E,MAAM,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;QAClG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACnE,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAChE,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACjF,MAAM,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACjF,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,OAAO,CAAC;QAEtB,kBAAkB,CAAC,gBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC1D,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAEvD,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC;QAE/B,MAAM,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACjF,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;QACpC,SAAS,CAAC,KAAK,GAAG,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,QAAQ,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAChF,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC9B,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/multipool/tests/job-state-registry.d.ts

@@ -1,17 +1,17 @@
-import { MultiWorkflowPool } from "src/multipool/multi-workflow-pool.js";
-import { Workflow } from "src/multipool/workflow.js";
-interface JobState {
-    jobId: string;
-    workflow: Workflow;
-    status: 'pending' | 'running' | 'completed' | 'failed' | 'canceled';
-}
-export declare class JobStateRegistry {
-    pool: MultiWorkflowPool;
-    jobs: Map<string, JobState>;
-    constructor(pool: MultiWorkflowPool);
-    addJob(workflow: Workflow): `${string}-${string}-${string}-${string}-${string}`;
-    getJobStatus(jobId: string): void;
-    cancelJob(jobId: string): void;
-}
-export {};
+import { MultiWorkflowPool } from "src/multipool/multi-workflow-pool.js";
+import { Workflow } from "src/multipool/workflow.js";
+interface JobState {
+    jobId: string;
+    workflow: Workflow;
+    status: 'pending' | 'running' | 'completed' | 'failed' | 'canceled';
+}
+export declare class JobStateRegistry {
+    pool: MultiWorkflowPool;
+    jobs: Map<string, JobState>;
+    constructor(pool: MultiWorkflowPool);
+    addJob(workflow: Workflow): `${string}-${string}-${string}-${string}-${string}`;
+    getJobStatus(jobId: string): void;
+    cancelJob(jobId: string): void;
+}
+export {};
 //# sourceMappingURL=job-state-registry.d.ts.map

package/dist/multipool/tests/job-state-registry.js

@@ -1,24 +1,24 @@
-import { randomUUID } from "node:crypto";
-export class JobStateRegistry {
-    pool;
-    jobs = new Map();
-    constructor(pool) {
-        this.pool = pool;
-    }
-    addJob(workflow) {
-        // Create new job id
-        const jobId = randomUUID();
-        const jobState = {
-            jobId,
-            workflow,
-            status: 'pending',
-        };
-        this.jobs.set(jobId, jobState);
-        return jobId;
-    }
-    getJobStatus(jobId) {
-    }
-    cancelJob(jobId) {
-    }
-}
+import { randomUUID } from "node:crypto";
+export class JobStateRegistry {
+    pool;
+    jobs = new Map();
+    constructor(pool) {
+        this.pool = pool;
+    }
+    addJob(workflow) {
+        // Create new job id
+        const jobId = randomUUID();
+        const jobState = {
+            jobId,
+            workflow,
+            status: 'pending',
+        };
+        this.jobs.set(jobId, jobState);
+        return jobId;
+    }
+    getJobStatus(jobId) {
+    }
+    cancelJob(jobId) {
+    }
+}
 //# sourceMappingURL=job-state-registry.js.map

package/dist/multipool/tests/job-state-registry.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=job-state-registry.spec.d.ts.map

package/dist/multipool/tests/job-state-registry.spec.d.ts.map

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

package/dist/multipool/tests/job-state-registry.spec.js

@@ -0,0 +1,143 @@
+import { describe, it, expect, beforeEach, jest } from "bun:test";
+import { JobStateRegistry } from "../job-state-registry.js";
+import { ClientRegistry } from "../client-registry.js";
+import { Workflow } from "../workflow.js";
+import { Logger } from "../logger.js";
+describe("JobStateRegistry", () => {
+    let jobRegistry;
+    let poolMock;
+    let clientRegistryMock;
+    let loggerMock;
+    beforeEach(() => {
+        // Mock MultiWorkflowPool
+        poolMock = {
+            options: { enableProfiling: false },
+            queues: new Map(),
+        };
+        loggerMock = new Logger("test", "silent");
+        // Mock ClientRegistry
+        clientRegistryMock = new ClientRegistry(poolMock, loggerMock);
+        // Initialize JobStateRegistry
+        jobRegistry = new JobStateRegistry(poolMock, clientRegistryMock);
+    });
+    it("should add a new job with a pending status", () => {
+        const workflow = new Workflow({});
+        const jobId = jobRegistry.addJob(workflow);
+        expect(jobId).toBeString();
+        expect(jobRegistry.getJobStatus(jobId)).toBe("pending");
+        const jobState = jobRegistry.jobs.get(jobId);
+        expect(jobState).toBeDefined();
+        expect(jobState?.workflow).toBe(workflow);
+        expect(jobState?.resolver).toBeFunction();
+        expect(jobState?.resultsPromise).toBeInstanceOf(Promise);
+    });
+    it("should throw an error when getting the status of a non-existent job", () => {
+        expect(() => jobRegistry.getJobStatus("non-existent-job-id")).toThrow("Job with ID non-existent-job-id not found.");
+    });
+    it("should set job status", () => {
+        const workflow = new Workflow({});
+        const jobId = jobRegistry.addJob(workflow);
+        jobRegistry.setJobStatus(jobId, "assigned", "http://localhost:8188");
+        expect(jobRegistry.getJobStatus(jobId)).toBe("assigned");
+        const jobState = jobRegistry.jobs.get(jobId);
+        expect(jobState?.assignedClientUrl).toBe("http://localhost:8188");
+    });
+    it("should set prompt ID and map it to job ID", () => {
+        const workflow = new Workflow({});
+        const jobId = jobRegistry.addJob(workflow);
+        const promptId = "prompt-123";
+        jobRegistry.setPromptId(jobId, promptId);
+        const jobState = jobRegistry.jobs.get(jobId);
+        expect(jobState?.prompt_id).toBe(promptId);
+        expect(jobRegistry.promptIdToJobId.get(promptId)).toBe(jobId);
+    });
+    it("should complete a job and resolve the promise", async () => {
+        const workflow = new Workflow({});
+        const jobId = jobRegistry.addJob(workflow);
+        const promptId = "prompt-123";
+        jobRegistry.setPromptId(jobId, promptId);
+        jobRegistry.setJobStatus(jobId, "running", "http://localhost:8188");
+        const resultsPromise = jobRegistry.waitForResults(jobId);
+        // Simulate adding images
+        jobRegistry.addJobImages(promptId, [{ filename: "test.png", subfolder: "", type: "output" }]);
+        // Mock client for image URL generation
+        clientRegistryMock.clients.set("http://localhost:8188", {
+            api: { ext: { file: { getPathImage: (image) => `http://localhost:8188/view?filename=${image.filename}` } } }
+        });
+        jobRegistry.completeJob(promptId);
+        const results = await resultsPromise;
+        expect(jobRegistry.getJobStatus(jobId)).toBe("completed");
+        expect(results.status).toBe("completed");
+        expect(results.jobId).toBe(jobId);
+        expect(results.prompt_id).toBe(promptId);
+        expect(results.images).toEqual(["http://localhost:8188/view?filename=test.png"]);
+    });
+    it("should fail a job and resolve the promise with an error", async () => {
+        const workflow = new Workflow({});
+        const jobId = jobRegistry.addJob(workflow);
+        const promptId = "prompt-456";
+        jobRegistry.setPromptId(jobId, promptId);
+        jobRegistry.setJobStatus(jobId, "running");
+        const resultsPromise = jobRegistry.waitForResults(jobId);
+        const errorDetails = { error: "Test error" };
+        jobRegistry.setJobFailure(jobId, errorDetails);
+        const results = await resultsPromise;
+        expect(jobRegistry.getJobStatus(jobId)).toBe("failed");
+        expect(results.status).toBe("failed");
+        expect(results.jobId).toBe(jobId);
+        expect(results.prompt_id).toBe(promptId);
+        expect(results.error).toEqual(errorDetails);
+    });
+    it("should enable profiling when the pool option is set", () => {
+        poolMock.options.enableProfiling = true;
+        jobRegistry = new JobStateRegistry(poolMock, clientRegistryMock);
+        const workflow = new Workflow({});
+        const jobId = jobRegistry.addJob(workflow);
+        const jobState = jobRegistry.jobs.get(jobId);
+        expect(jobState?.profiler).toBeDefined();
+    });
+    describe("cancelJob", () => {
+        it("should cancel a pending job", async () => {
+            const workflow = new Workflow({});
+            const jobId = jobRegistry.addJob(workflow);
+            const resultsPromise = jobRegistry.waitForResults(jobId);
+            // Mock queue to check if dequeue is called
+            const queueMock = { dequeueJob: jest.fn() };
+            jobRegistry.pool.queues.set(workflow.structureHash || "general", queueMock);
+            await jobRegistry.cancelJob(jobId);
+            expect(jobRegistry.getJobStatus(jobId)).toBe("canceled");
+            expect(queueMock.dequeueJob).toHaveBeenCalledWith(jobId);
+            const result = await resultsPromise;
+            expect(result.status).toBe("canceled");
+        });
+        it("should cancel a running job", async () => {
+            const workflow = new Workflow({});
+            const jobId = jobRegistry.addJob(workflow);
+            const promptId = "prompt-789";
+            const clientUrl = "http://localhost:8188";
+            jobRegistry.setJobStatus(jobId, "running", clientUrl);
+            jobRegistry.setPromptId(jobId, promptId);
+            const resultsPromise = jobRegistry.waitForResults(jobId);
+            // Mock client and its API
+            const clientApiMock = { ext: { queue: { interrupt: jest.fn() } } };
+            clientRegistryMock.clients.set(clientUrl, { api: clientApiMock, state: "busy" });
+            // Mock queue processing
+            const queueMock = { processQueue: jest.fn().mockResolvedValue(undefined) };
+            jobRegistry.pool.queues.set(workflow.structureHash || "general", queueMock);
+            await jobRegistry.cancelJob(jobId);
+            expect(jobRegistry.getJobStatus(jobId)).toBe("canceled");
+            expect(clientApiMock.ext.queue.interrupt).toHaveBeenCalledWith(promptId);
+            expect(clientRegistryMock.clients.get(clientUrl)?.state).toBe("idle");
+            expect(queueMock.processQueue).toHaveBeenCalled();
+            const result = await resultsPromise;
+            expect(result.status).toBe("canceled");
+        });
+        it("should throw an error when trying to cancel a completed job", async () => {
+            const workflow = new Workflow({});
+            const jobId = jobRegistry.addJob(workflow);
+            jobRegistry.setJobStatus(jobId, "completed");
+            await expect(jobRegistry.cancelJob(jobId)).rejects.toThrow(`Cannot cancel job ${jobId} with status completed.`);
+        });
+    });
+});
+//# sourceMappingURL=job-state-registry.spec.js.map

package/dist/multipool/tests/job-state-registry.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"job-state-registry.spec.js","sourceRoot":"","sources":["../../../src/multipool/tests/job-state-registry.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,UAAU,CAAC;AAClE,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAE5D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,WAA6B,CAAC;IAClC,IAAI,QAA2B,CAAC;IAChC,IAAI,kBAAkC,CAAC;IACvC,IAAI,UAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,yBAAyB;QACzB,QAAQ,GAAG;YACT,OAAO,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE;YACnC,MAAM,EAAE,IAAI,GAAG,EAAE;SACX,CAAC;QAET,UAAU,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAE1C,sBAAsB;QACtB,kBAAkB,GAAG,IAAI,cAAc,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAE9D,8BAA8B;QAC9B,WAAW,GAAG,IAAI,gBAAgB,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAE3C,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAC;QAC3B,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/B,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,YAAY,EAAE,CAAC;QAC1C,MAAM,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC,CAAC,OAAO,CACnE,4CAA4C,CAC7C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,WAAW,CAAC,YAAY,CAAC,KAAK,EAAE,UAAU,EAAE,uBAAuB,CAAC,CAAC;QAErE,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,MAAM,QAAQ,GAAG,YAAY,CAAC;QAE9B,WAAW,CAAC,WAAW,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAEzC,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3C,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,MAAM,QAAQ,GAAG,YAAY,CAAC;QAC9B,WAAW,CAAC,WAAW,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACzC,WAAW,CAAC,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,uBAAuB,CAAC,CAAC;QAEpE,MAAM,cAAc,GAAG,WAAW,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAEzD,yBAAyB;QACzB,WAAW,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAE9F,uCAAuC;QACvC,kBAAkB,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE;YACpD,GAAG,EAAE,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,EAAE,YAAY,EAAE,CAAC,KAAU,EAAE,EAAE,CAAC,uCAAuC,KAAK,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE;SAC7G,CAAC,CAAC;QAEV,WAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAElC,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC;QAErC,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,8CAA8C,CAAC,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3C,MAAM,QAAQ,GAAG,YAAY,CAAC;QAC9B,WAAW,CAAC,WAAW,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACzC,WAAW,CAAC,YAAY,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAE3C,MAAM,cAAc,GAAG,WAAW,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,YAAY,GAAG,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;QAE7C,WAAW,CAAC,aAAa,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;QAE/C,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC;QAErC,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,QAAQ,CAAC,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC;QACxC,WAAW,GAAG,IAAI,gBAAgB,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;QAEjE,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC3C,MAAM,cAAc,GAAG,WAAW,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAEzD,2CAA2C;YAC3C,MAAM,SAAS,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC;YAC5C,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,IAAI,SAAS,EAAE,SAAgB,CAAC,CAAC;YAEnF,MAAM,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAEnC,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzD,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC3C,MAAM,QAAQ,GAAG,YAAY,CAAC;YAC9B,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,WAAW,CAAC,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YACtD,WAAW,CAAC,WAAW,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;YAEzC,MAAM,cAAc,GAAG,WAAW,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAEzD,0BAA0B;YAC1B,MAAM,aAAa,GAAG,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;YACnE,kBAAkB,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAS,CAAC,CAAC;YAExF,wBAAwB;YACxB,MAAM,SAAS,GAAG,EAAE,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3E,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,IAAI,SAAS,EAAE,SAAgB,CAAC,CAAC;YAEnF,MAAM,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAEnC,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzD,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YACzE,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtE,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC3C,WAAW,CAAC,YAAY,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YAE7C,MAAM,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,qBAAqB,KAAK,yBAAyB,CACpD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/multipool/tests/multipool-basic.d.ts

@@ -1,2 +1,12 @@
-export {};
+export declare class SimulatedUser {
+    promptGenerator: (() => string) | null;
+    modelFunction: (prompt: string) => Promise<string[]>;
+    shouldGenerate: boolean;
+    totalImages: number;
+    collectedImages: number;
+    constructor(generator?: (() => string) | null, modelFunction?: (prompt: string) => Promise<string[]>, totalImages?: number);
+    stop(): void;
+    start(): void;
+    generateImages(count: number): Promise<void>;
+}
 //# sourceMappingURL=multipool-basic.d.ts.map

package/dist/multipool/tests/multipool-basic.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"multipool-basic.d.ts","sourceRoot":"","sources":["../../../src/multipool/tests/multipool-basic.ts"],"names":[],"mappings":""}
+{"version":3,"file":"multipool-basic.d.ts","sourceRoot":"","sources":["../../../src/multipool/tests/multipool-basic.ts"],"names":[],"mappings":"AA8GA,qBAAa,aAAa;IAExB,eAAe,EAAE,CAAC,MAAM,MAAM,CAAC,GAAG,IAAI,CAAC;IACvC,aAAa,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAkB;IACtE,cAAc,EAAE,OAAO,CAAQ;IAC/B,WAAW,EAAE,MAAM,CAAM;IACzB,eAAe,EAAE,MAAM,CAAK;gBAG1B,SAAS,GAAE,CAAC,MAAM,MAAM,CAAC,GAAG,IAAW,EACvC,aAAa,GAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,EAAE,CAAkB,EACrE,WAAW,GAAE,MAAW;IAO1B,IAAI;IAIJ,KAAK;IAOC,cAAc,CAAC,KAAK,EAAE,MAAM;CAkBnC"}