comfyui-node 1.6.4 → 1.6.5

package/dist/types/event.d.ts

@@ -122,7 +122,7 @@ export type TExecutionInterrupted = TExecution & {
 /**
  * Union type of all ComfyUI API event keys
  */
-export type ComfyApiEventKey = "all" | "auth_error" | "connection_error" | "auth_success" | "status" | "progress" | "executing" | "executed" | "disconnected" | "execution_success" | "execution_start" | "execution_error" | "execution_cached" | "queue_error" | "reconnected" | "connected" | "log" | "terminal" | "reconnecting" | "b_preview" | "b_preview_meta" | "b_text" | "b_text_meta" | "b_preview_raw" | "node_text_update";
+export type ComfyApiEventKey = "all" | "auth_error" | "connection_error" | "auth_success" | "status" | "progress" | "executing" | "executed" | "disconnected" | "execution_success" | "execution_start" | "execution_error" | "execution_cached" | "queue_error" | "reconnected" | "reconnection_failed" | "connected" | "log" | "terminal" | "reconnecting" | "b_preview" | "b_preview_meta" | "b_text" | "b_text_meta" | "b_preview_raw" | "node_text_update" | "websocket_unavailable" | "shutdown_started" | "shutdown_complete";
 /**
  * Type mapping ComfyUI API event keys to their respective CustomEvent types
  */
@@ -170,6 +170,22 @@ export type TComfyAPIEventMap = {
      * Reconnected event
      */
     reconnected: CustomEvent<null>;
+    /**
+     * Reconnection failed event (after all retry attempts exhausted)
+     */
+    reconnection_failed: CustomEvent<null>;
+    /**
+     * WebSocket unavailable event (when WebSocket creation fails and polling fallback is used)
+     */
+    websocket_unavailable: CustomEvent<Error>;
+    /**
+     * Shutdown started event
+     */
+    shutdown_started: CustomEvent<null>;
+    /**
+     * Shutdown complete event
+     */
+    shutdown_complete: CustomEvent<null>;
     /**
      * Binary preview image event
      */

package/dist/types/event.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"event.d.ts","sourceRoot":"","sources":["../../src/types/event.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D;;GAEG;AACH,MAAM,MAAM,YAAY,GAAG;IACzB;;OAEG;IACH,MAAM,EAAE;QACN;;WAEG;QACH,SAAS,EAAE;YACT;;eAEG;YACH,eAAe,EAAE,MAAM,CAAC;SACzB,CAAC;KACH,CAAC;IACF;;OAEG;IACH,GAAG,EAAE,MAAM,CAAC;CACb,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG;IACvB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG,UAAU,GAAG;IACpC;;OAEG;IACH,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;CACrB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,SAAS,GAAG,UAAU,GAAG;IACnC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,GAAG,EAAE,MAAM,CAAC;CACb,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,SAAS,CAAC,CAAC,GAAG,OAAO,IAAI,UAAU,GAAG;IAChD;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb;;OAEG;IACH,MAAM,EAAE,CAAC,CAAC;CACX,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,UAAU,GAAG;IAC1C;;OAEG;IACH,KAAK,EAAE,MAAM,EAAE,CAAC;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,UAAU,GAAG;IACzC;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,iBAAiB,EAAE,MAAM,CAAC;IAC1B;;OAEG;IACH,cAAc,EAAE,MAAM,CAAC;IACvB;;OAEG;IACH,SAAS,EAAE,MAAM,EAAE,CAAC;CACrB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,UAAU,GAAG;IAC/C;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gBAAgB,GACxB,KAAK,GACL,YAAY,GACZ,kBAAkB,GAClB,cAAc,GACd,QAAQ,GACR,UAAU,GACV,WAAW,GACX,UAAU,GACV,cAAc,GACd,mBAAmB,GACnB,iBAAiB,GACjB,iBAAiB,GACjB,kBAAkB,GAClB,aAAa,GACb,aAAa,GACb,WAAW,GACX,KAAK,GACL,UAAU,GACV,cAAc,GACd,WAAW,GACX,gBAAgB,GAChB,QAAQ,GACR,aAAa,GACb,eAAe,GACf,kBAAkB,CAAC;AAEvB;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG;IAC9B;;OAEG;IACH,GAAG,EAAE,WAAW,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,GAAG,CAAA;KAAE,CAAC,CAAC;IAC9C;;OAEG;IACH,UAAU,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC;IAClC;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAChC;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;IACrC;;OAEG;IACH,iBAAiB,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IAC3C;;OAEG;IACH,MAAM,EAAE,WAAW,CAAC,YAAY,CAAC,CAAC;IAClC;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAChC;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAChC;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAC7B;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAC/B;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAC7B;;OAEG;IACH,cAAc,EAAE,WAAW,CAAC;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,QAAQ,EAAE,GAAG,CAAA;KAAE,CAAC,CAAC;IAC3D;;OAEG;IACH,MAAM,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAC5B;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5D;;OAEG;IACH,aAAa,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IACvC;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC;QAC5B,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,UAAU,GAAG,QAAQ,GAAG,SAAS,CAAC;QACxC,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC9B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;IACH;;OAEG;IACH,GAAG,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IACtE;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC;QAAE,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAChD;;OAEG;IACH,eAAe,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IACzC;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IACnC;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC;IACjC;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC;IACjC;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC;;OAEG;IACH,eAAe,EAAE,WAAW,CAAC,eAAe,CAAC,CAAC;IAC9C;;OAEG;IACH,qBAAqB,EAAE,WAAW,CAAC,qBAAqB,CAAC,CAAC;IAC1D;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC,gBAAgB,CAAC,CAAC;CACjD,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,iBAAiB,GACzB,MAAM,GACN,aAAa,GACb,YAAY,GACZ,kBAAkB,GAClB,cAAc,GACd,OAAO,GACP,SAAS,GACT,SAAS,GACT,UAAU,GACV,MAAM,GACN,UAAU,GACV,OAAO,GACP,aAAa,GACb,WAAW,GACX,cAAc,GACd,aAAa,GACb,WAAW,GACX,UAAU,GACV,uBAAuB,GACvB,iBAAiB,GACjB,gBAAgB,CAAC;AAErB;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B;;OAEG;IACH,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IACxB;;OAEG;IACH,UAAU,EAAE,WAAW,CAAC;QACtB,MAAM,EAAE,QAAQ,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,GAAG,EAAE,QAAQ,CAAC;KACf,CAAC,CAAC;IACH;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC;QAC5B,MAAM,EAAE,QAAQ,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE,KAAK,CAAC;KACd,CAAC,CAAC;IACH;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnE;;OAEG;IACH,KAAK,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5D;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnE;;OAEG;IACH,cAAc,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACrE;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC;QAAE,IAAI,EAAE,UAAU,CAAA;KAAE,CAAC,CAAC;IAC/C;;OAEG;IACH,KAAK,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5D;;OAEG;IACH,OAAO,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9D;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAChE;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnE;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAClE;;OAEG;IACH,OAAO,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACzD;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5D;;OAEG;IACH,IAAI,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAA;KAAE,CAAC,CAAC;IACxC;;OAEG;IACH,qBAAqB,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5E;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAChE;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/D;;OAEG;IACH,eAAe,EAAE,WAAW,CAAC;QAC3B,MAAM,EAAE,QAAQ,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE,KAAK,CAAC;QACb,SAAS,CAAC,EAAE,OAAO,CAAC;QACpB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,CAAC,CAAC;IACH;;OAEG;IACH,cAAc,EAAE,WAAW,CAAC;QAC1B,MAAM,EAAE,QAAQ,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,aAAa,CAAC;KACrB,CAAC,CAAC;CACJ,CAAC"}
+{"version":3,"file":"event.d.ts","sourceRoot":"","sources":["../../src/types/event.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D;;GAEG;AACH,MAAM,MAAM,YAAY,GAAG;IACzB;;OAEG;IACH,MAAM,EAAE;QACN;;WAEG;QACH,SAAS,EAAE;YACT;;eAEG;YACH,eAAe,EAAE,MAAM,CAAC;SACzB,CAAC;KACH,CAAC;IACF;;OAEG;IACH,GAAG,EAAE,MAAM,CAAC;CACb,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG;IACvB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG,UAAU,GAAG;IACpC;;OAEG;IACH,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;CACrB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,SAAS,GAAG,UAAU,GAAG;IACnC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,GAAG,EAAE,MAAM,CAAC;CACb,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,SAAS,CAAC,CAAC,GAAG,OAAO,IAAI,UAAU,GAAG;IAChD;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb;;OAEG;IACH,MAAM,EAAE,CAAC,CAAC;CACX,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,UAAU,GAAG;IAC1C;;OAEG;IACH,KAAK,EAAE,MAAM,EAAE,CAAC;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,UAAU,GAAG;IACzC;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,iBAAiB,EAAE,MAAM,CAAC;IAC1B;;OAEG;IACH,cAAc,EAAE,MAAM,CAAC;IACvB;;OAEG;IACH,SAAS,EAAE,MAAM,EAAE,CAAC;CACrB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,UAAU,GAAG;IAC/C;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gBAAgB,GACxB,KAAK,GACL,YAAY,GACZ,kBAAkB,GAClB,cAAc,GACd,QAAQ,GACR,UAAU,GACV,WAAW,GACX,UAAU,GACV,cAAc,GACd,mBAAmB,GACnB,iBAAiB,GACjB,iBAAiB,GACjB,kBAAkB,GAClB,aAAa,GACb,aAAa,GACb,qBAAqB,GACrB,WAAW,GACX,KAAK,GACL,UAAU,GACV,cAAc,GACd,WAAW,GACX,gBAAgB,GAChB,QAAQ,GACR,aAAa,GACb,eAAe,GACf,kBAAkB,GAClB,uBAAuB,GACvB,kBAAkB,GAClB,mBAAmB,CAAC;AAExB;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG;IAC9B;;OAEG;IACH,GAAG,EAAE,WAAW,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,GAAG,CAAA;KAAE,CAAC,CAAC;IAC9C;;OAEG;IACH,UAAU,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC;IAClC;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAChC;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;IACrC;;OAEG;IACH,iBAAiB,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IAC3C;;OAEG;IACH,MAAM,EAAE,WAAW,CAAC,YAAY,CAAC,CAAC;IAClC;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAChC;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAChC;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAC7B;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAC/B;;OAEG;IACH,mBAAmB,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IACvC;;OAEG;IACH,qBAAqB,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;IAC1C;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IACpC;;OAEG;IACH,iBAAiB,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IACrC;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAC7B;;OAEG;IACH,cAAc,EAAE,WAAW,CAAC;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,QAAQ,EAAE,GAAG,CAAA;KAAE,CAAC,CAAC;IAC3D;;OAEG;IACH,MAAM,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAC5B;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5D;;OAEG;IACH,aAAa,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IACvC;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC;QAC5B,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,UAAU,GAAG,QAAQ,GAAG,SAAS,CAAC;QACxC,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC9B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;IACH;;OAEG;IACH,GAAG,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IACtE;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC;QAAE,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAChD;;OAEG;IACH,eAAe,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IACzC;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IACnC;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC;IACjC;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC;IACjC;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC;;OAEG;IACH,eAAe,EAAE,WAAW,CAAC,eAAe,CAAC,CAAC;IAC9C;;OAEG;IACH,qBAAqB,EAAE,WAAW,CAAC,qBAAqB,CAAC,CAAC;IAC1D;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC,gBAAgB,CAAC,CAAC;CACjD,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,iBAAiB,GACzB,MAAM,GACN,aAAa,GACb,YAAY,GACZ,kBAAkB,GAClB,cAAc,GACd,OAAO,GACP,SAAS,GACT,SAAS,GACT,UAAU,GACV,MAAM,GACN,UAAU,GACV,OAAO,GACP,aAAa,GACb,WAAW,GACX,cAAc,GACd,aAAa,GACb,WAAW,GACX,UAAU,GACV,uBAAuB,GACvB,iBAAiB,GACjB,gBAAgB,CAAC;AAErB;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG;IAC/B;;OAEG;IACH,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IACxB;;OAEG;IACH,UAAU,EAAE,WAAW,CAAC;QACtB,MAAM,EAAE,QAAQ,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,GAAG,EAAE,QAAQ,CAAC;KACf,CAAC,CAAC;IACH;;OAEG;IACH,gBAAgB,EAAE,WAAW,CAAC;QAC5B,MAAM,EAAE,QAAQ,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE,KAAK,CAAC;KACd,CAAC,CAAC;IACH;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnE;;OAEG;IACH,KAAK,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5D;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnE;;OAEG;IACH,cAAc,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACrE;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC;QAAE,IAAI,EAAE,UAAU,CAAA;KAAE,CAAC,CAAC;IAC/C;;OAEG;IACH,KAAK,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5D;;OAEG;IACH,OAAO,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9D;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAChE;;OAEG;IACH,YAAY,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnE;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAClE;;OAEG;IACH,OAAO,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACzD;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5D;;OAEG;IACH,IAAI,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAA;KAAE,CAAC,CAAC;IACxC;;OAEG;IACH,qBAAqB,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5E;;OAEG;IACH,SAAS,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAChE;;OAEG;IACH,QAAQ,EAAE,WAAW,CAAC;QAAE,MAAM,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/D;;OAEG;IACH,eAAe,EAAE,WAAW,CAAC;QAC3B,MAAM,EAAE,QAAQ,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE,KAAK,CAAC;QACb,SAAS,CAAC,EAAE,OAAO,CAAC;QACpB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,CAAC,CAAC;IACH;;OAEG;IACH,cAAc,EAAE,WAAW,CAAC;QAC1B,MAAM,EAAE,QAAQ,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,aAAa,CAAC;KACrB,CAAC,CAAC;CACJ,CAAC"}

package/dist/utils/model-loading.d.ts

@@ -0,0 +1,158 @@
+/**
+ * Model loading detection utilities.
+ *
+ * These utilities help identify when ComfyUI is loading models (a typically slow operation)
+ * vs. executing standard workflow nodes. This is useful for applying adaptive timeouts
+ * and providing better progress feedback to users.
+ */
+/**
+ * List of node types that typically load models from disk.
+ * These operations can be very slow on first execution (cold start).
+ */
+export declare const MODEL_LOADING_NODE_TYPES: readonly ["CheckpointLoaderSimple", "CheckpointLoader", "unCLIPCheckpointLoader", "VAELoader", "VAEDecode", "LoraLoader", "LoraLoaderModelOnly", "ControlNetLoader", "ControlNetApply", "ControlNetApplyAdvanced", "IPAdapterModelLoader", "IPAdapterApply", "CLIPLoader", "DualCLIPLoader", "CLIPVisionLoader", "UNETLoader", "DiffusersLoader", "StyleModelLoader", "UpscaleModelLoader", "GLIGENLoader", "PhotoMakerLoader", "InstantIDModelLoader", "AnimateDiffLoader", "ADE_AnimateDiffLoaderGen1", "ADE_AnimateDiffLoaderGen2", "FluxCheckpointLoader", "SD3CheckpointLoader", "comfy.sd1.CheckpointLoader", "comfy.sd2.CheckpointLoader", "comfy.sdxl.CheckpointLoader"];
+/**
+ * Check if a node type is a model loading node.
+ * Model loading nodes typically have longer execution times on first run
+ * due to loading large files from disk into memory.
+ *
+ * @param nodeType - The class_type of the node to check
+ * @returns true if the node is known to load models from disk
+ *
+ * @example
+ * ```ts
+ * if (isModelLoadingNode("CheckpointLoaderSimple")) {
+ *   console.log("This node will load a checkpoint - may take a while!");
+ * }
+ * ```
+ */
+export declare function isModelLoadingNode(nodeType: string): boolean;
+/**
+ * Check if a workflow contains any model loading nodes.
+ * Useful for determining if a workflow might have a slow cold start.
+ *
+ * @param workflow - The workflow JSON object
+ * @returns true if the workflow contains at least one model loading node
+ *
+ * @example
+ * ```ts
+ * const workflow = { ... };
+ * if (workflowContainsModelLoading(workflow)) {
+ *   console.log("First execution may be slow due to model loading");
+ * }
+ * ```
+ */
+export declare function workflowContainsModelLoading(workflow: object): boolean;
+/**
+ * Extract all model loading nodes from a workflow.
+ * Returns an array of node IDs and their types.
+ *
+ * @param workflow - The workflow JSON object
+ * @returns Array of objects containing nodeId and nodeType for each model loading node
+ *
+ * @example
+ * ```ts
+ * const modelNodes = getModelLoadingNodes(workflow);
+ * console.log(`Found ${modelNodes.length} model loading nodes:`);
+ * modelNodes.forEach(({ nodeId, nodeType }) => {
+ *   console.log(`  - ${nodeId}: ${nodeType}`);
+ * });
+ * ```
+ */
+export declare function getModelLoadingNodes(workflow: object): Array<{
+    nodeId: string;
+    nodeType: string;
+}>;
+/**
+ * Estimated cold start time multiplier for model loading nodes.
+ * These are rough estimates based on typical hardware (HDD vs SSD, VRAM, etc.).
+ */
+export declare const MODEL_LOADING_TIME_ESTIMATES: {
+    /** Checkpoint models (SD 1.5: ~2GB, SDXL: ~6GB, Flux: ~12GB+) */
+    readonly checkpoint: {
+        readonly hdd: 30000;
+        readonly ssd: 10000;
+        readonly nvme: 5000;
+    };
+    /** VAE models (~200-800MB) */
+    readonly vae: {
+        readonly hdd: 5000;
+        readonly ssd: 2000;
+        readonly nvme: 1000;
+    };
+    /** LoRA models (~50-200MB) */
+    readonly lora: {
+        readonly hdd: 3000;
+        readonly ssd: 1000;
+        readonly nvme: 500;
+    };
+    /** ControlNet models (~1-3GB) */
+    readonly controlnet: {
+        readonly hdd: 15000;
+        readonly ssd: 5000;
+        readonly nvme: 2500;
+    };
+    /** Other models (CLIP, upscale, etc.) */
+    readonly other: {
+        readonly hdd: 10000;
+        readonly ssd: 3000;
+        readonly nvme: 1500;
+    };
+};
+/**
+ * Storage type for estimating model loading times.
+ */
+export type StorageType = "hdd" | "ssd" | "nvme";
+/**
+ * Estimate the model loading time for a node type.
+ * This is a rough estimate and actual times may vary significantly based on:
+ * - Storage type and speed
+ * - Model size
+ * - Available VRAM
+ * - System memory
+ * - CPU performance
+ *
+ * @param nodeType - The class_type of the node
+ * @param storageType - The type of storage (defaults to "ssd")
+ * @returns Estimated loading time in milliseconds, or 0 if not a model loading node
+ *
+ * @example
+ * ```ts
+ * const estimatedTime = estimateModelLoadingTime("CheckpointLoaderSimple", "nvme");
+ * console.log(`Estimated loading time: ${estimatedTime}ms`);
+ * ```
+ */
+export declare function estimateModelLoadingTime(nodeType: string, storageType?: StorageType): number;
+/**
+ * Calculate the total estimated model loading time for a workflow.
+ * This assumes all models need to be loaded from disk (cold start).
+ * On subsequent runs, models may be cached in VRAM/RAM.
+ *
+ * @param workflow - The workflow JSON object
+ * @param storageType - The type of storage (defaults to "ssd")
+ * @returns Total estimated loading time in milliseconds
+ *
+ * @example
+ * ```ts
+ * const totalTime = estimateWorkflowModelLoadingTime(workflow, "nvme");
+ * console.log(`Estimated cold start time: ${totalTime / 1000}s`);
+ * ```
+ */
+export declare function estimateWorkflowModelLoadingTime(workflow: object, storageType?: StorageType): number;
+/**
+ * Determine if a timeout should be extended for first execution based on model loading.
+ * Returns a recommended timeout multiplier.
+ *
+ * @param workflow - The workflow JSON object
+ * @param hasExecutedBefore - Whether this workflow has been executed before on this client
+ * @returns Recommended timeout multiplier (1.0 = no change, 2.0 = double timeout, etc.)
+ *
+ * @example
+ * ```ts
+ * const baseTimeout = 60000; // 1 minute
+ * const multiplier = getTimeoutMultiplierForModelLoading(workflow, false);
+ * const adjustedTimeout = baseTimeout * multiplier;
+ * console.log(`Using timeout: ${adjustedTimeout}ms (${multiplier}x base)`);
+ * ```
+ */
+export declare function getTimeoutMultiplierForModelLoading(workflow: object, hasExecutedBefore?: boolean): number;
+//# sourceMappingURL=model-loading.d.ts.map

package/dist/utils/model-loading.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-loading.d.ts","sourceRoot":"","sources":["../../src/utils/model-loading.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH;;;GAGG;AACH,eAAO,MAAM,wBAAwB,kpBAgE3B,CAAC;AAEX;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAE5D;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,4BAA4B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAatE;AAED;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG,KAAK,CAAC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC,CAalG;AAED;;;GAGG;AACH,eAAO,MAAM,4BAA4B;IACvC,iEAAiE;;;;;;IAMjE,8BAA8B;;;;;;IAM9B,8BAA8B;;;;;;IAM9B,iCAAiC;;;;;;IAMjC,yCAAyC;;;;;;CAMjC,CAAC;AAEX;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,CAAC;AAEjD;;;;;;;;;;;;;;;;;;GAkBG;AACH,wBAAgB,wBAAwB,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,GAAE,WAAmB,GAAG,MAAM,CA6BnG;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,gCAAgC,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,GAAE,WAAmB,GAAG,MAAM,CAS3G;AAED;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,mCAAmC,CACjD,QAAQ,EAAE,MAAM,EAChB,iBAAiB,GAAE,OAAe,GACjC,MAAM,CAyBR"}

package/dist/utils/model-loading.js

@@ -0,0 +1,273 @@
+/**
+ * Model loading detection utilities.
+ *
+ * These utilities help identify when ComfyUI is loading models (a typically slow operation)
+ * vs. executing standard workflow nodes. This is useful for applying adaptive timeouts
+ * and providing better progress feedback to users.
+ */
+/**
+ * List of node types that typically load models from disk.
+ * These operations can be very slow on first execution (cold start).
+ */
+export const MODEL_LOADING_NODE_TYPES = [
+    // Checkpoint loaders
+    "CheckpointLoaderSimple",
+    "CheckpointLoader",
+    "unCLIPCheckpointLoader",
+    // VAE loaders
+    "VAELoader",
+    "VAEDecode",
+    // LoRA loaders
+    "LoraLoader",
+    "LoraLoaderModelOnly",
+    // ControlNet loaders
+    "ControlNetLoader",
+    "ControlNetApply",
+    "ControlNetApplyAdvanced",
+    // IP-Adapter loaders
+    "IPAdapterModelLoader",
+    "IPAdapterApply",
+    // CLIP loaders
+    "CLIPLoader",
+    "DualCLIPLoader",
+    "CLIPVisionLoader",
+    // UNET loaders
+    "UNETLoader",
+    // Diffusion model loaders
+    "DiffusersLoader",
+    // Style model loaders
+    "StyleModelLoader",
+    // Upscale model loaders
+    "UpscaleModelLoader",
+    // GLIGEN loaders
+    "GLIGENLoader",
+    // PhotoMaker loaders
+    "PhotoMakerLoader",
+    // InstantID loaders
+    "InstantIDModelLoader",
+    // AnimateDiff loaders
+    "AnimateDiffLoader",
+    "ADE_AnimateDiffLoaderGen1",
+    "ADE_AnimateDiffLoaderGen2",
+    // Flux loaders
+    "FluxCheckpointLoader",
+    // SD3 loaders
+    "SD3CheckpointLoader",
+    // Common custom node loaders
+    "comfy.sd1.CheckpointLoader",
+    "comfy.sd2.CheckpointLoader",
+    "comfy.sdxl.CheckpointLoader",
+];
+/**
+ * Check if a node type is a model loading node.
+ * Model loading nodes typically have longer execution times on first run
+ * due to loading large files from disk into memory.
+ *
+ * @param nodeType - The class_type of the node to check
+ * @returns true if the node is known to load models from disk
+ *
+ * @example
+ * ```ts
+ * if (isModelLoadingNode("CheckpointLoaderSimple")) {
+ *   console.log("This node will load a checkpoint - may take a while!");
+ * }
+ * ```
+ */
+export function isModelLoadingNode(nodeType) {
+    return MODEL_LOADING_NODE_TYPES.includes(nodeType);
+}
+/**
+ * Check if a workflow contains any model loading nodes.
+ * Useful for determining if a workflow might have a slow cold start.
+ *
+ * @param workflow - The workflow JSON object
+ * @returns true if the workflow contains at least one model loading node
+ *
+ * @example
+ * ```ts
+ * const workflow = { ... };
+ * if (workflowContainsModelLoading(workflow)) {
+ *   console.log("First execution may be slow due to model loading");
+ * }
+ * ```
+ */
+export function workflowContainsModelLoading(workflow) {
+    const nodes = Object.values(workflow);
+    for (const node of nodes) {
+        if (typeof node === "object" && node !== null) {
+            const classType = node.class_type;
+            if (classType && isModelLoadingNode(classType)) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+/**
+ * Extract all model loading nodes from a workflow.
+ * Returns an array of node IDs and their types.
+ *
+ * @param workflow - The workflow JSON object
+ * @returns Array of objects containing nodeId and nodeType for each model loading node
+ *
+ * @example
+ * ```ts
+ * const modelNodes = getModelLoadingNodes(workflow);
+ * console.log(`Found ${modelNodes.length} model loading nodes:`);
+ * modelNodes.forEach(({ nodeId, nodeType }) => {
+ *   console.log(`  - ${nodeId}: ${nodeType}`);
+ * });
+ * ```
+ */
+export function getModelLoadingNodes(workflow) {
+    const modelNodes = [];
+    for (const [nodeId, node] of Object.entries(workflow)) {
+        if (typeof node === "object" && node !== null) {
+            const classType = node.class_type;
+            if (classType && isModelLoadingNode(classType)) {
+                modelNodes.push({ nodeId, nodeType: classType });
+            }
+        }
+    }
+    return modelNodes;
+}
+/**
+ * Estimated cold start time multiplier for model loading nodes.
+ * These are rough estimates based on typical hardware (HDD vs SSD, VRAM, etc.).
+ */
+export const MODEL_LOADING_TIME_ESTIMATES = {
+    /** Checkpoint models (SD 1.5: ~2GB, SDXL: ~6GB, Flux: ~12GB+) */
+    checkpoint: {
+        hdd: 30000, // 30 seconds on HDD
+        ssd: 10000, // 10 seconds on SSD
+        nvme: 5000, // 5 seconds on NVMe
+    },
+    /** VAE models (~200-800MB) */
+    vae: {
+        hdd: 5000, // 5 seconds on HDD
+        ssd: 2000, // 2 seconds on SSD
+        nvme: 1000, // 1 second on NVMe
+    },
+    /** LoRA models (~50-200MB) */
+    lora: {
+        hdd: 3000, // 3 seconds on HDD
+        ssd: 1000, // 1 second on SSD
+        nvme: 500, // 0.5 seconds on NVMe
+    },
+    /** ControlNet models (~1-3GB) */
+    controlnet: {
+        hdd: 15000, // 15 seconds on HDD
+        ssd: 5000, // 5 seconds on SSD
+        nvme: 2500, // 2.5 seconds on NVMe
+    },
+    /** Other models (CLIP, upscale, etc.) */
+    other: {
+        hdd: 10000, // 10 seconds on HDD
+        ssd: 3000, // 3 seconds on SSD
+        nvme: 1500, // 1.5 seconds on NVMe
+    },
+};
+/**
+ * Estimate the model loading time for a node type.
+ * This is a rough estimate and actual times may vary significantly based on:
+ * - Storage type and speed
+ * - Model size
+ * - Available VRAM
+ * - System memory
+ * - CPU performance
+ *
+ * @param nodeType - The class_type of the node
+ * @param storageType - The type of storage (defaults to "ssd")
+ * @returns Estimated loading time in milliseconds, or 0 if not a model loading node
+ *
+ * @example
+ * ```ts
+ * const estimatedTime = estimateModelLoadingTime("CheckpointLoaderSimple", "nvme");
+ * console.log(`Estimated loading time: ${estimatedTime}ms`);
+ * ```
+ */
+export function estimateModelLoadingTime(nodeType, storageType = "ssd") {
+    if (!isModelLoadingNode(nodeType)) {
+        return 0;
+    }
+    const lower = nodeType.toLowerCase();
+    // Checkpoint loaders
+    if (lower.includes("checkpoint")) {
+        return MODEL_LOADING_TIME_ESTIMATES.checkpoint[storageType];
+    }
+    // VAE loaders
+    if (lower.includes("vae")) {
+        return MODEL_LOADING_TIME_ESTIMATES.vae[storageType];
+    }
+    // LoRA loaders
+    if (lower.includes("lora")) {
+        return MODEL_LOADING_TIME_ESTIMATES.lora[storageType];
+    }
+    // ControlNet loaders
+    if (lower.includes("controlnet")) {
+        return MODEL_LOADING_TIME_ESTIMATES.controlnet[storageType];
+    }
+    // Default to "other" category
+    return MODEL_LOADING_TIME_ESTIMATES.other[storageType];
+}
+/**
+ * Calculate the total estimated model loading time for a workflow.
+ * This assumes all models need to be loaded from disk (cold start).
+ * On subsequent runs, models may be cached in VRAM/RAM.
+ *
+ * @param workflow - The workflow JSON object
+ * @param storageType - The type of storage (defaults to "ssd")
+ * @returns Total estimated loading time in milliseconds
+ *
+ * @example
+ * ```ts
+ * const totalTime = estimateWorkflowModelLoadingTime(workflow, "nvme");
+ * console.log(`Estimated cold start time: ${totalTime / 1000}s`);
+ * ```
+ */
+export function estimateWorkflowModelLoadingTime(workflow, storageType = "ssd") {
+    const modelNodes = getModelLoadingNodes(workflow);
+    let totalTime = 0;
+    for (const { nodeType } of modelNodes) {
+        totalTime += estimateModelLoadingTime(nodeType, storageType);
+    }
+    return totalTime;
+}
+/**
+ * Determine if a timeout should be extended for first execution based on model loading.
+ * Returns a recommended timeout multiplier.
+ *
+ * @param workflow - The workflow JSON object
+ * @param hasExecutedBefore - Whether this workflow has been executed before on this client
+ * @returns Recommended timeout multiplier (1.0 = no change, 2.0 = double timeout, etc.)
+ *
+ * @example
+ * ```ts
+ * const baseTimeout = 60000; // 1 minute
+ * const multiplier = getTimeoutMultiplierForModelLoading(workflow, false);
+ * const adjustedTimeout = baseTimeout * multiplier;
+ * console.log(`Using timeout: ${adjustedTimeout}ms (${multiplier}x base)`);
+ * ```
+ */
+export function getTimeoutMultiplierForModelLoading(workflow, hasExecutedBefore = false) {
+    // If already executed, models are likely cached
+    if (hasExecutedBefore) {
+        return 1.0;
+    }
+    const modelNodes = getModelLoadingNodes(workflow);
+    // No model loading nodes
+    if (modelNodes.length === 0) {
+        return 1.0;
+    }
+    // 1-2 model nodes: 2x timeout
+    if (modelNodes.length <= 2) {
+        return 2.0;
+    }
+    // 3-4 model nodes: 3x timeout
+    if (modelNodes.length <= 4) {
+        return 3.0;
+    }
+    // 5+ model nodes: 4x timeout (very heavy workflow)
+    return 4.0;
+}
+//# sourceMappingURL=model-loading.js.map

package/dist/utils/model-loading.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-loading.js","sourceRoot":"","sources":["../../src/utils/model-loading.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH;;;GAGG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG;IACtC,qBAAqB;IACrB,wBAAwB;IACxB,kBAAkB;IAClB,wBAAwB;IAExB,cAAc;IACd,WAAW;IACX,WAAW;IAEX,eAAe;IACf,YAAY;IACZ,qBAAqB;IAErB,qBAAqB;IACrB,kBAAkB;IAClB,iBAAiB;IACjB,yBAAyB;IAEzB,qBAAqB;IACrB,sBAAsB;IACtB,gBAAgB;IAEhB,eAAe;IACf,YAAY;IACZ,gBAAgB;IAChB,kBAAkB;IAElB,eAAe;IACf,YAAY;IAEZ,0BAA0B;IAC1B,iBAAiB;IAEjB,sBAAsB;IACtB,kBAAkB;IAElB,wBAAwB;IACxB,oBAAoB;IAEpB,iBAAiB;IACjB,cAAc;IAEd,qBAAqB;IACrB,kBAAkB;IAElB,oBAAoB;IACpB,sBAAsB;IAEtB,sBAAsB;IACtB,mBAAmB;IACnB,2BAA2B;IAC3B,2BAA2B;IAE3B,eAAe;IACf,sBAAsB;IAEtB,cAAc;IACd,qBAAqB;IAErB,6BAA6B;IAC7B,4BAA4B;IAC5B,4BAA4B;IAC5B,6BAA6B;CACrB,CAAC;AAEX;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAgB;IACjD,OAAO,wBAAwB,CAAC,QAAQ,CAAC,QAAe,CAAC,CAAC;AAC5D,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,4BAA4B,CAAC,QAAgB;IAC3D,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAEtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAC9C,MAAM,SAAS,GAAI,IAAY,CAAC,UAAU,CAAC;YAC3C,IAAI,SAAS,IAAI,kBAAkB,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC/C,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,oBAAoB,CAAC,QAAgB;IACnD,MAAM,UAAU,GAAgD,EAAE,CAAC;IAEnE,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAC9C,MAAM,SAAS,GAAI,IAAY,CAAC,UAAU,CAAC;YAC3C,IAAI,SAAS,IAAI,kBAAkB,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC/C,UAAU,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG;IAC1C,iEAAiE;IACjE,UAAU,EAAE;QACV,GAAG,EAAE,KAAK,EAAG,oBAAoB;QACjC,GAAG,EAAE,KAAK,EAAG,oBAAoB;QACjC,IAAI,EAAE,IAAI,EAAG,oBAAoB;KAClC;IACD,8BAA8B;IAC9B,GAAG,EAAE;QACH,GAAG,EAAE,IAAI,EAAI,mBAAmB;QAChC,GAAG,EAAE,IAAI,EAAI,mBAAmB;QAChC,IAAI,EAAE,IAAI,EAAG,mBAAmB;KACjC;IACD,8BAA8B;IAC9B,IAAI,EAAE;QACJ,GAAG,EAAE,IAAI,EAAI,mBAAmB;QAChC,GAAG,EAAE,IAAI,EAAI,kBAAkB;QAC/B,IAAI,EAAE,GAAG,EAAI,sBAAsB;KACpC;IACD,iCAAiC;IACjC,UAAU,EAAE;QACV,GAAG,EAAE,KAAK,EAAG,oBAAoB;QACjC,GAAG,EAAE,IAAI,EAAI,mBAAmB;QAChC,IAAI,EAAE,IAAI,EAAG,sBAAsB;KACpC;IACD,yCAAyC;IACzC,KAAK,EAAE;QACL,GAAG,EAAE,KAAK,EAAG,oBAAoB;QACjC,GAAG,EAAE,IAAI,EAAI,mBAAmB;QAChC,IAAI,EAAE,IAAI,EAAG,sBAAsB;KACpC;CACO,CAAC;AAOX;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,UAAU,wBAAwB,CAAC,QAAgB,EAAE,cAA2B,KAAK;IACzF,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClC,OAAO,CAAC,CAAC;IACX,CAAC;IAED,MAAM,KAAK,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;IAErC,qBAAqB;IACrB,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACjC,OAAO,4BAA4B,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAC9D,CAAC;IAED,cAAc;IACd,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,4BAA4B,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACvD,CAAC;IAED,eAAe;IACf,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,OAAO,4BAA4B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACxD,CAAC;IAED,qBAAqB;IACrB,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACjC,OAAO,4BAA4B,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAC9D,CAAC;IAED,8BAA8B;IAC9B,OAAO,4BAA4B,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;AACzD,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,gCAAgC,CAAC,QAAgB,EAAE,cAA2B,KAAK;IACjG,MAAM,UAAU,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IAClD,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,KAAK,MAAM,EAAE,QAAQ,EAAE,IAAI,UAAU,EAAE,CAAC;QACtC,SAAS,IAAI,wBAAwB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,mCAAmC,CACjD,QAAgB,EAChB,oBAA6B,KAAK;IAElC,gDAAgD;IAChD,IAAI,iBAAiB,EAAE,CAAC;QACtB,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,UAAU,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IAElD,yBAAyB;IACzB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,GAAG,CAAC;IACb,CAAC;IAED,8BAA8B;IAC9B,IAAI,UAAU,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QAC3B,OAAO,GAAG,CAAC;IACb,CAAC;IAED,8BAA8B;IAC9B,IAAI,UAAU,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QAC3B,OAAO,GAAG,CAAC;IACb,CAAC;IAED,mDAAmD;IACnD,OAAO,GAAG,CAAC;AACb,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "comfyui-node",
-  "version": "1.6.4",
+  "version": "1.6.5",
   "description": "ComfyUI Node.js Client",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -39,12 +39,16 @@
   "scripts": {
     "build": "tsc --project tsconfig.json",
     "build:watch": "tsc --project tsconfig.json --watch",
-    "test": "bun test",
-    "test:local": "bun test ./test",
+    "test": "bun test --bail 10 ./test/*.spec.ts",
+    "test:unit": "bun test ./test/*.spec.ts",
+    "test:all": "bun test",
+    "test:integration": "bun test ./test/integration/",
+    "test:integration:simple": "bun test ./test/integration/reconnection-simple.example.spec.ts",
+    "test:integration:reconnection": "bun test ./test/integration/reconnection.integration.spec.ts",
     "test:real": "cross-env COMFY_REAL=1 bun test ./test/real.integration.spec.ts",
     "test:full": "cross-env COMFY_REAL=1 COMFY_FULL=1 bun test ./test/real.full.integration.spec.ts",
-    "coverage": "bun test --coverage",
-    "coverage:lcov": "bun test --coverage --coverage-reporter=lcov",
+    "coverage": "bun test --coverage ./test/*.spec.ts",
+    "coverage:lcov": "bun test --coverage --coverage-reporter=lcov ./test/*.spec.ts",
     "coverage:enforce": "bun run coverage:lcov && bun run coverage:check",
     "coverage:check": "bun scripts/coverage-check.ts",
     "typecov": "type-coverage --detail --ignore-catch --at-least 0",