@@ -3,7 +3,6 @@ import { computed, ref } from 'vue'
3
3
4
4
import type {
5
5
ExecutedWsMessage ,
6
- ExecutingWsMessage ,
7
6
ExecutionCachedWsMessage ,
8
7
ExecutionErrorWsMessage ,
9
8
ExecutionStartWsMessage ,
@@ -37,7 +36,7 @@ export const useExecutionStore = defineStore('execution', () => {
37
36
const queuedPrompts = ref < Record < NodeId , QueuedPrompt > > ( { } )
38
37
const lastNodeErrors = ref < Record < NodeId , NodeError > | null > ( null )
39
38
const lastExecutionError = ref < ExecutionErrorWsMessage | null > ( null )
40
- const executingNodeId = ref < string | null > ( null )
39
+ const executingNodeId = ref < NodeId | null > ( null )
41
40
const executingNode = computed < ComfyNode | null > ( ( ) => {
42
41
if ( ! executingNodeId . value ) return null
43
42
@@ -142,7 +141,7 @@ export const useExecutionStore = defineStore('execution', () => {
142
141
activePrompt . value . nodes [ e . detail . node ] = true
143
142
}
144
143
145
- function handleExecuting ( e : CustomEvent < ExecutingWsMessage > ) {
144
+ function handleExecuting ( e : CustomEvent < NodeId | null > ) {
146
145
// Clear the current node progress when a new node starts executing
147
146
_executingNodeProgress . value = null
148
147
@@ -152,8 +151,8 @@ export const useExecutionStore = defineStore('execution', () => {
152
151
// Seems sometimes nodes that are cached fire executing but not executed
153
152
activePrompt . value . nodes [ executingNodeId . value ] = true
154
153
}
155
- executingNodeId . value = e . detail ? String ( e . detail ) : null
156
- if ( ! executingNodeId . value ) {
154
+ executingNodeId . value = e . detail
155
+ if ( executingNodeId . value === null ) {
157
156
if ( activePromptId . value ) {
158
157
delete queuedPrompts . value [ activePromptId . value ]
159
158
}
0 commit comments