comfyui-node 1.6.5 → 1.6.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +20 -20
- package/README.md +342 -341
- package/dist/.tsbuildinfo +1 -1
- package/dist/multipool/client-registry.d.ts +3 -3
- package/dist/multipool/client-registry.d.ts.map +1 -1
- package/dist/multipool/client-registry.js +9 -9
- package/dist/multipool/client-registry.js.map +1 -1
- package/dist/multipool/helpers.d.ts +4 -4
- package/dist/multipool/index.d.ts +2 -2
- package/dist/multipool/interfaces.d.ts +0 -2
- package/dist/multipool/interfaces.d.ts.map +1 -1
- package/dist/multipool/job-queue-processor.d.ts +3 -3
- package/dist/multipool/job-queue-processor.d.ts.map +1 -1
- package/dist/multipool/job-queue-processor.js +28 -27
- package/dist/multipool/job-queue-processor.js.map +1 -1
- package/dist/multipool/logger.d.ts +29 -29
- package/dist/multipool/logger.js +74 -74
- package/dist/multipool/multi-workflow-pool.d.ts +0 -1
- package/dist/multipool/multi-workflow-pool.d.ts.map +1 -1
- package/dist/multipool/multi-workflow-pool.js +36 -37
- package/dist/multipool/multi-workflow-pool.js.map +1 -1
- package/dist/multipool/tests/client-registry-api-demo.js +1 -3
- package/dist/multipool/tests/client-registry-api-demo.js.map +1 -1
- package/dist/multipool/tests/client-registry.spec.js +6 -7
- package/dist/multipool/tests/client-registry.spec.js.map +1 -1
- package/dist/multipool/tests/error-classification-tests.d.ts +1 -1
- package/dist/multipool/tests/event-forwarding-demo.js +1 -3
- package/dist/multipool/tests/event-forwarding-demo.js.map +1 -1
- package/dist/multipool/tests/job-queue-processor.spec.js +7 -7
- package/dist/multipool/tests/job-queue-processor.spec.js.map +1 -1
- package/dist/multipool/tests/job-state-registry.d.ts +16 -16
- package/dist/multipool/tests/job-state-registry.js +23 -23
- package/dist/multipool/tests/job-state-registry.spec.js +5 -4
- package/dist/multipool/tests/job-state-registry.spec.js.map +1 -1
- package/dist/multipool/tests/multipool-basic.d.ts +11 -11
- package/dist/multipool/tests/profiling-demo.d.ts +6 -6
- package/dist/multipool/tests/profiling-demo.js +1 -2
- package/dist/multipool/tests/profiling-demo.js.map +1 -1
- package/dist/multipool/tests/prompt-generator.d.ts +9 -9
- package/dist/multipool/tests/prompt-generator.js +25 -25
- package/dist/multipool/tests/test-helpers.d.ts +3 -3
- package/dist/multipool/tests/test-helpers.js +9 -9
- package/dist/multipool/tests/two-stage-edit-simulation.d.ts +31 -31
- package/dist/multipool/tests/two-stage-edit-simulation.d.ts.map +1 -1
- package/dist/multipool/tests/two-stage-edit-simulation.js +1 -2
- package/dist/multipool/tests/two-stage-edit-simulation.js.map +1 -1
- package/dist/pool/SmartPool.d.ts +143 -143
- package/dist/pool/SmartPool.js +676 -676
- package/dist/pool/SmartPoolV2.d.ts +119 -119
- package/dist/pool/SmartPoolV2.js +586 -586
- package/dist/pool/WorkflowPool.d.ts +202 -202
- package/dist/pool/client/ClientManager.d.ts +86 -86
- package/dist/pool/index.d.ts +9 -9
- package/dist/types/api.d.ts +2 -0
- package/dist/types/api.d.ts.map +1 -1
- package/package.json +4 -4
|
@@ -1 +1 @@
|
|
|
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,
|
|
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,sBAAsB,GAAG,GAAG,EAAE,CAAC,CAAC;IACpC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;CACrB,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,UAA4B,CAAC;IACjC,IAAI,QAAwB,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,cAAc,EAAE,CAAC;QAC5B,UAAU,GAAG,sBAAsB,EAAE,CAAC;QACtC,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,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,+CAA+C,EAAE,CAAC,CAAC;QACjI,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"}
|
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export {};
|
|
1
|
+
export {};
|
|
2
2
|
//# sourceMappingURL=error-classification-tests.d.ts.map
|
|
@@ -8,9 +8,7 @@ import { Workflow } from "../workflow.js";
|
|
|
8
8
|
import GenerationGraph from "../../../scripts/workflows/T2I-one-obsession.json" with { type: "json" };
|
|
9
9
|
const GEN_HOST = "http://localhost:8188";
|
|
10
10
|
// Create pool
|
|
11
|
-
const pool = new MultiWorkflowPool({
|
|
12
|
-
logLevel: "warn" // Keep logs minimal to see events clearly
|
|
13
|
-
});
|
|
11
|
+
const pool = new MultiWorkflowPool({});
|
|
14
12
|
const genWorkflow = Workflow.fromAugmented(GenerationGraph);
|
|
15
13
|
pool.addClient(GEN_HOST, {
|
|
16
14
|
workflowAffinity: [genWorkflow],
|
|
@@ -1 +1 @@
|
|
|
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
|
|
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,EAAE,CAAC,CAAC;AAEvC,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"}
|
|
@@ -1,7 +1,6 @@
|
|
|
1
1
|
import { describe, it, expect, beforeEach, jest } from "bun:test";
|
|
2
2
|
import { JobQueueProcessor } from "../job-queue-processor.js";
|
|
3
3
|
import { Workflow } from "../workflow.js";
|
|
4
|
-
import { Logger } from "../logger.js";
|
|
5
4
|
// Mock dependencies
|
|
6
5
|
const createJobStateRegistryMock = () => ({
|
|
7
6
|
getJobStatus: jest.fn().mockReturnValue("pending"),
|
|
@@ -17,17 +16,19 @@ const createClientRegistryMock = () => ({
|
|
|
17
16
|
markClientIncompatibleWithWorkflow: jest.fn(),
|
|
18
17
|
getAllEligibleClientsForWorkflow: jest.fn().mockReturnValue([]),
|
|
19
18
|
});
|
|
20
|
-
const
|
|
19
|
+
const createEventManagerMock = () => ({
|
|
20
|
+
emitEvent: jest.fn(),
|
|
21
|
+
});
|
|
21
22
|
describe("JobQueueProcessor", () => {
|
|
22
23
|
let jobStateRegistryMock;
|
|
23
24
|
let clientRegistryMock;
|
|
24
|
-
let
|
|
25
|
+
let eventsMock;
|
|
25
26
|
beforeEach(() => {
|
|
26
27
|
jobStateRegistryMock = createJobStateRegistryMock();
|
|
27
28
|
clientRegistryMock = createClientRegistryMock();
|
|
28
|
-
|
|
29
|
+
eventsMock = createEventManagerMock();
|
|
29
30
|
});
|
|
30
|
-
const createProcessor = (hash = "test-hash") => new JobQueueProcessor(jobStateRegistryMock, clientRegistryMock, hash,
|
|
31
|
+
const createProcessor = (hash = "test-hash") => new JobQueueProcessor(jobStateRegistryMock, clientRegistryMock, hash, eventsMock);
|
|
31
32
|
it("should enqueue a job and trigger processing", async () => {
|
|
32
33
|
const processor = createProcessor();
|
|
33
34
|
const processQueueSpy = jest.spyOn(processor, "processQueue").mockImplementation(async () => { });
|
|
@@ -39,9 +40,8 @@ describe("JobQueueProcessor", () => {
|
|
|
39
40
|
it("should not process queue if already processing", async () => {
|
|
40
41
|
const processor = createProcessor();
|
|
41
42
|
processor.isProcessing = true;
|
|
42
|
-
const loggerSpy = jest.spyOn(loggerMock, 'debug');
|
|
43
43
|
await processor.processQueue();
|
|
44
|
-
expect(
|
|
44
|
+
expect(eventsMock.emitEvent).toHaveBeenCalledWith({ type: "debug", payload: `Job queue for workflow hash test-hash is already being processed, skipping.` });
|
|
45
45
|
});
|
|
46
46
|
it("should assign a job to an optimal client and run it successfully", async () => {
|
|
47
47
|
const processor = createProcessor();
|
|
@@ -1 +1 @@
|
|
|
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;
|
|
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;AAG1C,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,sBAAsB,GAAG,GAAG,EAAE,CAAC,CAAC;IACpC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;CACrB,CAAQ,CAAC;AAEV,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,oBAAsC,CAAC;IAC3C,IAAI,kBAAuB,CAAC;IAC5B,IAAI,UAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,oBAAoB,GAAG,0BAA0B,EAAE,CAAC;QACpD,kBAAkB,GAAG,wBAAwB,EAAE,CAAC;QAChD,UAAU,GAAG,sBAAsB,EAAE,CAAC;IACxC,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;QAE9B,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC;QAE/B,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,6EAA6E,EAAE,CAAC,CAAC;IAC/J,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"}
|
|
@@ -1,17 +1,17 @@
|
|
|
1
|
-
import { MultiWorkflowPool } from "src/multipool/multi-workflow-pool.js";
|
|
2
|
-
import { Workflow } from "src/multipool/workflow.js";
|
|
3
|
-
interface JobState {
|
|
4
|
-
jobId: string;
|
|
5
|
-
workflow: Workflow;
|
|
6
|
-
status: 'pending' | 'running' | 'completed' | 'failed' | 'canceled';
|
|
7
|
-
}
|
|
8
|
-
export declare class JobStateRegistry {
|
|
9
|
-
pool: MultiWorkflowPool;
|
|
10
|
-
jobs: Map<string, JobState>;
|
|
11
|
-
constructor(pool: MultiWorkflowPool);
|
|
12
|
-
addJob(workflow: Workflow): `${string}-${string}-${string}-${string}-${string}`;
|
|
13
|
-
getJobStatus(jobId: string): void;
|
|
14
|
-
cancelJob(jobId: string): void;
|
|
15
|
-
}
|
|
16
|
-
export {};
|
|
1
|
+
import { MultiWorkflowPool } from "src/multipool/multi-workflow-pool.js";
|
|
2
|
+
import { Workflow } from "src/multipool/workflow.js";
|
|
3
|
+
interface JobState {
|
|
4
|
+
jobId: string;
|
|
5
|
+
workflow: Workflow;
|
|
6
|
+
status: 'pending' | 'running' | 'completed' | 'failed' | 'canceled';
|
|
7
|
+
}
|
|
8
|
+
export declare class JobStateRegistry {
|
|
9
|
+
pool: MultiWorkflowPool;
|
|
10
|
+
jobs: Map<string, JobState>;
|
|
11
|
+
constructor(pool: MultiWorkflowPool);
|
|
12
|
+
addJob(workflow: Workflow): `${string}-${string}-${string}-${string}-${string}`;
|
|
13
|
+
getJobStatus(jobId: string): void;
|
|
14
|
+
cancelJob(jobId: string): void;
|
|
15
|
+
}
|
|
16
|
+
export {};
|
|
17
17
|
//# sourceMappingURL=job-state-registry.d.ts.map
|
|
@@ -1,24 +1,24 @@
|
|
|
1
|
-
import { randomUUID } from "node:crypto";
|
|
2
|
-
export class JobStateRegistry {
|
|
3
|
-
pool;
|
|
4
|
-
jobs = new Map();
|
|
5
|
-
constructor(pool) {
|
|
6
|
-
this.pool = pool;
|
|
7
|
-
}
|
|
8
|
-
addJob(workflow) {
|
|
9
|
-
// Create new job id
|
|
10
|
-
const jobId = randomUUID();
|
|
11
|
-
const jobState = {
|
|
12
|
-
jobId,
|
|
13
|
-
workflow,
|
|
14
|
-
status: 'pending',
|
|
15
|
-
};
|
|
16
|
-
this.jobs.set(jobId, jobState);
|
|
17
|
-
return jobId;
|
|
18
|
-
}
|
|
19
|
-
getJobStatus(jobId) {
|
|
20
|
-
}
|
|
21
|
-
cancelJob(jobId) {
|
|
22
|
-
}
|
|
23
|
-
}
|
|
1
|
+
import { randomUUID } from "node:crypto";
|
|
2
|
+
export class JobStateRegistry {
|
|
3
|
+
pool;
|
|
4
|
+
jobs = new Map();
|
|
5
|
+
constructor(pool) {
|
|
6
|
+
this.pool = pool;
|
|
7
|
+
}
|
|
8
|
+
addJob(workflow) {
|
|
9
|
+
// Create new job id
|
|
10
|
+
const jobId = randomUUID();
|
|
11
|
+
const jobState = {
|
|
12
|
+
jobId,
|
|
13
|
+
workflow,
|
|
14
|
+
status: 'pending',
|
|
15
|
+
};
|
|
16
|
+
this.jobs.set(jobId, jobState);
|
|
17
|
+
return jobId;
|
|
18
|
+
}
|
|
19
|
+
getJobStatus(jobId) {
|
|
20
|
+
}
|
|
21
|
+
cancelJob(jobId) {
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
24
|
//# sourceMappingURL=job-state-registry.js.map
|
|
@@ -2,21 +2,22 @@ import { describe, it, expect, beforeEach, jest } from "bun:test";
|
|
|
2
2
|
import { JobStateRegistry } from "../job-state-registry.js";
|
|
3
3
|
import { ClientRegistry } from "../client-registry.js";
|
|
4
4
|
import { Workflow } from "../workflow.js";
|
|
5
|
-
import { Logger } from "../logger.js";
|
|
6
5
|
describe("JobStateRegistry", () => {
|
|
7
6
|
let jobRegistry;
|
|
8
7
|
let poolMock;
|
|
9
8
|
let clientRegistryMock;
|
|
10
|
-
let
|
|
9
|
+
let eventsMock;
|
|
11
10
|
beforeEach(() => {
|
|
12
11
|
// Mock MultiWorkflowPool
|
|
13
12
|
poolMock = {
|
|
14
13
|
options: { enableProfiling: false },
|
|
15
14
|
queues: new Map(),
|
|
16
15
|
};
|
|
17
|
-
|
|
16
|
+
eventsMock = {
|
|
17
|
+
emitEvent: jest.fn(),
|
|
18
|
+
};
|
|
18
19
|
// Mock ClientRegistry
|
|
19
|
-
clientRegistryMock = new ClientRegistry(poolMock,
|
|
20
|
+
clientRegistryMock = new ClientRegistry(poolMock, eventsMock);
|
|
20
21
|
// Initialize JobStateRegistry
|
|
21
22
|
jobRegistry = new JobStateRegistry(poolMock, clientRegistryMock);
|
|
22
23
|
});
|
|
@@ -1 +1 @@
|
|
|
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;
|
|
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;AAG1C,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,WAA6B,CAAC;IAClC,IAAI,QAA2B,CAAC;IAChC,IAAI,kBAAkC,CAAC;IACvC,IAAI,UAA4B,CAAC;IAEjC,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;YACX,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;SACd,CAAC;QAET,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"}
|
|
@@ -1,12 +1,12 @@
|
|
|
1
|
-
export declare class SimulatedUser {
|
|
2
|
-
promptGenerator: (() => string) | null;
|
|
3
|
-
modelFunction: (prompt: string) => Promise<string[]>;
|
|
4
|
-
shouldGenerate: boolean;
|
|
5
|
-
totalImages: number;
|
|
6
|
-
collectedImages: number;
|
|
7
|
-
constructor(generator?: (() => string) | null, modelFunction?: (prompt: string) => Promise<string[]>, totalImages?: number);
|
|
8
|
-
stop(): void;
|
|
9
|
-
start(): void;
|
|
10
|
-
generateImages(count: number): Promise<void>;
|
|
11
|
-
}
|
|
1
|
+
export declare class SimulatedUser {
|
|
2
|
+
promptGenerator: (() => string) | null;
|
|
3
|
+
modelFunction: (prompt: string) => Promise<string[]>;
|
|
4
|
+
shouldGenerate: boolean;
|
|
5
|
+
totalImages: number;
|
|
6
|
+
collectedImages: number;
|
|
7
|
+
constructor(generator?: (() => string) | null, modelFunction?: (prompt: string) => Promise<string[]>, totalImages?: number);
|
|
8
|
+
stop(): void;
|
|
9
|
+
start(): void;
|
|
10
|
+
generateImages(count: number): Promise<void>;
|
|
11
|
+
}
|
|
12
12
|
//# sourceMappingURL=multipool-basic.d.ts.map
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Profiling Demo for MultiWorkflowPool
|
|
3
|
-
*
|
|
4
|
-
* Demonstrates the integrated JobProfiler functionality
|
|
5
|
-
*/
|
|
6
|
-
export {};
|
|
1
|
+
/**
|
|
2
|
+
* Profiling Demo for MultiWorkflowPool
|
|
3
|
+
*
|
|
4
|
+
* Demonstrates the integrated JobProfiler functionality
|
|
5
|
+
*/
|
|
6
|
+
export {};
|
|
7
7
|
//# sourceMappingURL=profiling-demo.d.ts.map
|
|
@@ -9,8 +9,7 @@ import GenerationGraph from "../../../scripts/workflows/T2I-one-obsession.json"
|
|
|
9
9
|
const GEN_HOST = "http://localhost:8188";
|
|
10
10
|
// Create pool with profiling enabled
|
|
11
11
|
const pool = new MultiWorkflowPool({
|
|
12
|
-
enableProfiling: true
|
|
13
|
-
logLevel: "info"
|
|
12
|
+
enableProfiling: true
|
|
14
13
|
});
|
|
15
14
|
const genWorkflow = Workflow.fromAugmented(GenerationGraph);
|
|
16
15
|
pool.addClient(GEN_HOST, {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"profiling-demo.js","sourceRoot":"","sources":["../../../src/multipool/tests/profiling-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,qCAAqC;AACrC,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAAC;IACjC,eAAe,EAAE,IAAI;
|
|
1
|
+
{"version":3,"file":"profiling-demo.js","sourceRoot":"","sources":["../../../src/multipool/tests/profiling-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,qCAAqC;AACrC,MAAM,IAAI,GAAG,IAAI,iBAAiB,CAAC;IACjC,eAAe,EAAE,IAAI;CACtB,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,oCAAoC,CAAC,CAAC;AAClD,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,oBAAoB,CAAC,CAAC;AAClC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;AAEnC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;AAElB,6CAA6C;AAC7C,MAAM,QAAQ,GAAG,QAAQ,CAAC,aAAa,CAAC,eAAe,CAAC;KACrD,KAAK,CAAC,GAAG,EAAE,OAAO,EAAE,uEAAuE,CAAC;KAC5F,KAAK,CAAC,GAAG,EAAE,OAAO,EAAE,2BAA2B,CAAC;KAChD,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC;KACxB,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC;AAE3B,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;AAE1D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;AAC7C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;AAEtD,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,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;IACxB,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,CAAC;IAElC,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IACvC,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC;IAC1D,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IAEvD,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IACjE,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;IACpE,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;IAClE,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;IAClE,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;IAE3E,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1C,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3F,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;YAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,IAAI,CAAC,MAAM,GAAG,KAAK,KAAK,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;YAC3E,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,IAAI,IAAI,SAAS,EAAE,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;QACnD,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;YAC3C,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;YAC/D,IAAI,WAAW,EAAE,cAAc,EAAE,CAAC;gBAChC,OAAO,CAAC,GAAG,CAAC,UAAU,MAAM,KAAK,WAAW,CAAC,IAAI,IAAI,SAAS,MAAM,WAAW,CAAC,cAAc,CAAC,MAAM,kBAAkB,CAAC,CAAC;YAC3H,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,wCAAwC;IACxC,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;IACpD,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK;SAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,SAAS,CAAC;SACrC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,CAAC;IAE3D,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,YAAY,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,QAAQ,IAAI,CAAC;QAC5D,OAAO,CAAC,GAAG,CAAC,KAAK,MAAM,WAAW,IAAI,CAAC,MAAM,GAAG,KAAK,MAAM,IAAI,CAAC,IAAI,IAAI,SAAS,MAAM,QAAQ,EAAE,CAAC,CAAC;IACrG,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;AACrC,CAAC;KAAM,CAAC;IACN,OAAO,CAAC,GAAG,CAAC,+DAA+D,CAAC,CAAC;AAC/E,CAAC;AAED,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;AAEtB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;AACxD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC"}
|
|
@@ -1,10 +1,10 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Generate a random image generation prompt for testing.
|
|
3
|
-
*/
|
|
4
|
-
export declare function promptGenerator(): string;
|
|
5
|
-
/**
|
|
6
|
-
* Generate a random prompt specifically for the Anime XL model. Using more anime-focused elements and danbooru-style tags.
|
|
7
|
-
*/
|
|
8
|
-
export declare function animeXLPromptGenerator(): string;
|
|
9
|
-
export declare const NEGATIVE_PROMPT = "modern, recent, old, oldest, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured, long body, lowres, bad anatomy, bad hands, missing fingers, extra digits, fewer digits, cropped, very displeasing, (worst quality, bad quality:1.2), bad anatomy, sketch, jpeg artifacts, signature, watermark, username, signature, simple background, conjoined,bad ai-generated";
|
|
1
|
+
/**
|
|
2
|
+
* Generate a random image generation prompt for testing.
|
|
3
|
+
*/
|
|
4
|
+
export declare function promptGenerator(): string;
|
|
5
|
+
/**
|
|
6
|
+
* Generate a random prompt specifically for the Anime XL model. Using more anime-focused elements and danbooru-style tags.
|
|
7
|
+
*/
|
|
8
|
+
export declare function animeXLPromptGenerator(): string;
|
|
9
|
+
export declare const NEGATIVE_PROMPT = "modern, recent, old, oldest, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured, long body, lowres, bad anatomy, bad hands, missing fingers, extra digits, fewer digits, cropped, very displeasing, (worst quality, bad quality:1.2), bad anatomy, sketch, jpeg artifacts, signature, watermark, username, signature, simple background, conjoined,bad ai-generated";
|
|
10
10
|
//# sourceMappingURL=prompt-generator.d.ts.map
|
|
@@ -1,26 +1,26 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Generate a random image generation prompt for testing.
|
|
3
|
-
*/
|
|
4
|
-
export function promptGenerator() {
|
|
5
|
-
const subjects = ["a cat", "a dog", "a futuristic city", "a beautiful landscape", "an astronaut", "a dragon", "a robot", "a magical forest"];
|
|
6
|
-
const styles = ["in the style of Van Gogh", "as a Pixar movie", "like a watercolor painting", "as a cyberpunk scene", "in a surrealist style", "like a comic book illustration"];
|
|
7
|
-
const actions = ["flying through the sky", "exploring a new world", "sitting on a throne", "dancing in the rain", "holding a glowing orb", "standing on a mountain peak"];
|
|
8
|
-
const subject = subjects[Math.floor(Math.random() * subjects.length)];
|
|
9
|
-
const style = styles[Math.floor(Math.random() * styles.length)];
|
|
10
|
-
const action = actions[Math.floor(Math.random() * actions.length)];
|
|
11
|
-
return `A high-resolution image of ${subject} ${action}, ${style}, vibrant colors, detailed, 4k resolution`;
|
|
12
|
-
}
|
|
13
|
-
/**
|
|
14
|
-
* Generate a random prompt specifically for the Anime XL model. Using more anime-focused elements and danbooru-style tags.
|
|
15
|
-
*/
|
|
16
|
-
export function animeXLPromptGenerator() {
|
|
17
|
-
const subjects = ["1girl", "2boys", "catgirl", "mecha", "futuristic city", "magical girl", "samurai", "ninja"];
|
|
18
|
-
const styles = ["anime style", "manga style", "chibi style", "cyberpunk anime", "fantasy anime", "mecha anime"];
|
|
19
|
-
const actions = ["holding a katana", "sitting under cherry blossoms", "flying through the sky", "exploring a futuristic city", "casting a spell", "standing on a rooftop"];
|
|
20
|
-
const subject = subjects[Math.floor(Math.random() * subjects.length)];
|
|
21
|
-
const style = styles[Math.floor(Math.random() * styles.length)];
|
|
22
|
-
const action = actions[Math.floor(Math.random() * actions.length)];
|
|
23
|
-
return `score_9, score_8_up, score_7_up, score_6_up, score_5_up, score_4_up, source_anime, BREAK, ${subject}, ${action}, ${style}`;
|
|
24
|
-
}
|
|
25
|
-
export const NEGATIVE_PROMPT = `modern, recent, old, oldest, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured, long body, lowres, bad anatomy, bad hands, missing fingers, extra digits, fewer digits, cropped, very displeasing, (worst quality, bad quality:1.2), bad anatomy, sketch, jpeg artifacts, signature, watermark, username, signature, simple background, conjoined,bad ai-generated`;
|
|
1
|
+
/**
|
|
2
|
+
* Generate a random image generation prompt for testing.
|
|
3
|
+
*/
|
|
4
|
+
export function promptGenerator() {
|
|
5
|
+
const subjects = ["a cat", "a dog", "a futuristic city", "a beautiful landscape", "an astronaut", "a dragon", "a robot", "a magical forest"];
|
|
6
|
+
const styles = ["in the style of Van Gogh", "as a Pixar movie", "like a watercolor painting", "as a cyberpunk scene", "in a surrealist style", "like a comic book illustration"];
|
|
7
|
+
const actions = ["flying through the sky", "exploring a new world", "sitting on a throne", "dancing in the rain", "holding a glowing orb", "standing on a mountain peak"];
|
|
8
|
+
const subject = subjects[Math.floor(Math.random() * subjects.length)];
|
|
9
|
+
const style = styles[Math.floor(Math.random() * styles.length)];
|
|
10
|
+
const action = actions[Math.floor(Math.random() * actions.length)];
|
|
11
|
+
return `A high-resolution image of ${subject} ${action}, ${style}, vibrant colors, detailed, 4k resolution`;
|
|
12
|
+
}
|
|
13
|
+
/**
|
|
14
|
+
* Generate a random prompt specifically for the Anime XL model. Using more anime-focused elements and danbooru-style tags.
|
|
15
|
+
*/
|
|
16
|
+
export function animeXLPromptGenerator() {
|
|
17
|
+
const subjects = ["1girl", "2boys", "catgirl", "mecha", "futuristic city", "magical girl", "samurai", "ninja"];
|
|
18
|
+
const styles = ["anime style", "manga style", "chibi style", "cyberpunk anime", "fantasy anime", "mecha anime"];
|
|
19
|
+
const actions = ["holding a katana", "sitting under cherry blossoms", "flying through the sky", "exploring a futuristic city", "casting a spell", "standing on a rooftop"];
|
|
20
|
+
const subject = subjects[Math.floor(Math.random() * subjects.length)];
|
|
21
|
+
const style = styles[Math.floor(Math.random() * styles.length)];
|
|
22
|
+
const action = actions[Math.floor(Math.random() * actions.length)];
|
|
23
|
+
return `score_9, score_8_up, score_7_up, score_6_up, score_5_up, score_4_up, source_anime, BREAK, ${subject}, ${action}, ${style}`;
|
|
24
|
+
}
|
|
25
|
+
export const NEGATIVE_PROMPT = `modern, recent, old, oldest, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured, long body, lowres, bad anatomy, bad hands, missing fingers, extra digits, fewer digits, cropped, very displeasing, (worst quality, bad quality:1.2), bad anatomy, sketch, jpeg artifacts, signature, watermark, username, signature, simple background, conjoined,bad ai-generated`;
|
|
26
26
|
//# sourceMappingURL=prompt-generator.js.map
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
export declare function pickRandom<T>(arr: T[]): T;
|
|
2
|
-
export declare function randomInt(min: number, max: number): number;
|
|
3
|
-
export declare function randomSeed(): number;
|
|
1
|
+
export declare function pickRandom<T>(arr: T[]): T;
|
|
2
|
+
export declare function randomInt(min: number, max: number): number;
|
|
3
|
+
export declare function randomSeed(): number;
|
|
4
4
|
//# sourceMappingURL=test-helpers.d.ts.map
|
|
@@ -1,10 +1,10 @@
|
|
|
1
|
-
export function pickRandom(arr) {
|
|
2
|
-
return arr[Math.floor(Math.random() * arr.length)];
|
|
3
|
-
}
|
|
4
|
-
export function randomInt(min, max) {
|
|
5
|
-
return Math.floor(Math.random() * (max - min + 1)) + min;
|
|
6
|
-
}
|
|
7
|
-
export function randomSeed() {
|
|
8
|
-
return Math.floor(Math.random() * 2147483647);
|
|
9
|
-
}
|
|
1
|
+
export function pickRandom(arr) {
|
|
2
|
+
return arr[Math.floor(Math.random() * arr.length)];
|
|
3
|
+
}
|
|
4
|
+
export function randomInt(min, max) {
|
|
5
|
+
return Math.floor(Math.random() * (max - min + 1)) + min;
|
|
6
|
+
}
|
|
7
|
+
export function randomSeed() {
|
|
8
|
+
return Math.floor(Math.random() * 2147483647);
|
|
9
|
+
}
|
|
10
10
|
//# sourceMappingURL=test-helpers.js.map
|
|
@@ -1,32 +1,32 @@
|
|
|
1
|
-
export declare class TwoStageUser {
|
|
2
|
-
private userId;
|
|
3
|
-
private shouldRun;
|
|
4
|
-
private totalGenerations;
|
|
5
|
-
private generatedImages;
|
|
6
|
-
private editsPerImage;
|
|
7
|
-
private minDelayMs;
|
|
8
|
-
private maxDelayMs;
|
|
9
|
-
stats: {
|
|
10
|
-
generationsStarted: number;
|
|
11
|
-
generationsCompleted: number;
|
|
12
|
-
generationsFailed: number;
|
|
13
|
-
editsStarted: number;
|
|
14
|
-
editsCompleted: number;
|
|
15
|
-
editsFailed: number;
|
|
16
|
-
};
|
|
17
|
-
constructor(userId: string, options?: {
|
|
18
|
-
totalGenerations?: number;
|
|
19
|
-
editsPerImage?: number;
|
|
20
|
-
minDelayMs?: number;
|
|
21
|
-
maxDelayMs?: number;
|
|
22
|
-
});
|
|
23
|
-
stop(): void;
|
|
24
|
-
start(): Promise<void>;
|
|
25
|
-
private generationLoop;
|
|
26
|
-
private editLoop;
|
|
27
|
-
private generateImage;
|
|
28
|
-
private editImage;
|
|
29
|
-
private delay;
|
|
30
|
-
private printStats;
|
|
31
|
-
}
|
|
1
|
+
export declare class TwoStageUser {
|
|
2
|
+
private userId;
|
|
3
|
+
private shouldRun;
|
|
4
|
+
private totalGenerations;
|
|
5
|
+
private generatedImages;
|
|
6
|
+
private editsPerImage;
|
|
7
|
+
private minDelayMs;
|
|
8
|
+
private maxDelayMs;
|
|
9
|
+
stats: {
|
|
10
|
+
generationsStarted: number;
|
|
11
|
+
generationsCompleted: number;
|
|
12
|
+
generationsFailed: number;
|
|
13
|
+
editsStarted: number;
|
|
14
|
+
editsCompleted: number;
|
|
15
|
+
editsFailed: number;
|
|
16
|
+
};
|
|
17
|
+
constructor(userId: string, options?: {
|
|
18
|
+
totalGenerations?: number;
|
|
19
|
+
editsPerImage?: number;
|
|
20
|
+
minDelayMs?: number;
|
|
21
|
+
maxDelayMs?: number;
|
|
22
|
+
});
|
|
23
|
+
stop(): void;
|
|
24
|
+
start(): Promise<void>;
|
|
25
|
+
private generationLoop;
|
|
26
|
+
private editLoop;
|
|
27
|
+
private generateImage;
|
|
28
|
+
private editImage;
|
|
29
|
+
private delay;
|
|
30
|
+
private printStats;
|
|
31
|
+
}
|
|
32
32
|
//# sourceMappingURL=two-stage-edit-simulation.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"two-stage-edit-simulation.d.ts","sourceRoot":"","sources":["../../../src/multipool/tests/two-stage-edit-simulation.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"two-stage-edit-simulation.d.ts","sourceRoot":"","sources":["../../../src/multipool/tests/two-stage-edit-simulation.ts"],"names":[],"mappings":"AAqFA,qBAAa,YAAY;IACvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAiB;IAClC,OAAO,CAAC,gBAAgB,CAAa;IACrC,OAAO,CAAC,eAAe,CAAwB;IAC/C,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,UAAU,CAAS;IAGpB,KAAK;;;;;;;MAOV;gBAGA,MAAM,EAAE,MAAM,EACd,OAAO,GAAE;QACP,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,UAAU,CAAC,EAAE,MAAM,CAAC;KAChB;IASR,IAAI;IAIE,KAAK;YAcG,cAAc;YAWd,QAAQ;YAiBR,aAAa;YAqDb,SAAS;IAyCvB,OAAO,CAAC,KAAK;IAIb,OAAO,CAAC,UAAU;CASnB"}
|
|
@@ -40,8 +40,7 @@ const editPrompts = [
|
|
|
40
40
|
// POOL SETUP
|
|
41
41
|
// ============================================================================
|
|
42
42
|
const pool = new MultiWorkflowPool({
|
|
43
|
-
enableMonitoring: true
|
|
44
|
-
logLevel: "info"
|
|
43
|
+
enableMonitoring: true
|
|
45
44
|
});
|
|
46
45
|
const genWorkflow = Workflow.fromAugmented(GenerationGraph);
|
|
47
46
|
const editWorkflow = Workflow.fromAugmented(EditGraph);
|