Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): handle enriched events

Browse Source
This commit is contained in:
psychedelicious
2024-05-23 15:21:01 +10:00
parent 1e622a5706
commit aa329ea811
4 changed files with 23 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketInvocationError.ts
View File

@ -39,13 +39,21 @@ export const addInvocationErrorEventListener = (startAppListening: AppStartListe
actionCreator: socketInvocationError,
effect: (action, { getState }) => {
log.error(action.payload, `Invocation error (${action.payload.data.node.type})`);
const { source_node_id, error_type, graph_execution_state_id } = action.payload.data;
const { source_node_id, error_type, error_message, error_traceback, graph_execution_state_id } =
action.payload.data;
const nes = deepClone($nodeExecutionStates.get()[source_node_id]);
if (nes) {
nes.status = zNodeStatus.enum.FAILED;
nes.error = action.payload.data.error;
nes.progress = null;
nes.progressImage = null;
if (error_type && error_message && error_traceback) {
nes.error = {
error_type,
error_message,
error_traceback,
};
}
upsertExecutionState(nes.nodeId, nes);
}