Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-04-30 19:21:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ws message schema on reconnecting (#390)

Browse Source 
* Update ws message schema on reconnecting

* nit
This commit is contained in:
Chenlei Hu
2024-08-12 13:21:48 -04:00
committed by GitHub
parent a5be1f6072
commit 91e21b1387
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/scripts/ui/menu/interruptButton.ts
View File

@@ -16,7 +16,7 @@ export function getInteruptButton(visibility: string) {
api.addEventListener( api.addEventListener(
'status', 'status',
({ detail }: CustomEvent<StatusWsMessageStatus>) => { ({ detail }: CustomEvent<StatusWsMessageStatus>) => {
const sz = detail.exec_info.queue_remaining const sz = detail?.exec_info?.queue_remaining
btn.enabled = sz > 0 btn.enabled = sz > 0
} }
) )

2
src/scripts/ui/menu/queueButton.ts
View File

@@ -85,7 +85,7 @@ export class ComfyQueueButton {
api.addEventListener( api.addEventListener(
'status', 'status',
({ detail }: CustomEvent<StatusWsMessageStatus>) => { ({ detail }: CustomEvent<StatusWsMessageStatus>) => {
this.#internalQueueSize = detail.exec_info.queue_remaining this.#internalQueueSize = detail?.exec_info?.queue_remaining
if (this.#internalQueueSize != null) { if (this.#internalQueueSize != null) {
if (!this.#internalQueueSize && !app.lastExecutionError) { if (!this.#internalQueueSize && !app.lastExecutionError) {
if ( if (

2
src/stores/queueStore.ts
View File

@@ -214,7 +214,7 @@ export const useQueuePendingTaskCountStore = defineStore(
}), }),
actions: { actions: {
update(e: CustomEvent<StatusWsMessageStatus>) { update(e: CustomEvent<StatusWsMessageStatus>) {
this.count = e.detail.exec_info.queue_remaining this.count = e.detail?.exec_info?.queue_remaining || 0
} }
} }
} }

2
src/types/apiTypes.ts
View File

@@ -19,7 +19,7 @@ const zStatusWsMessageStatus = z.object({
}) })
const zStatusWsMessage = z.object({ const zStatusWsMessage = z.object({
status: zStatusWsMessageStatus status: zStatusWsMessageStatus.nullable().optional()
}) })
const zProgressWsMessage = z.object({ const zProgressWsMessage = z.object({