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): add warning socket event handling

Browse Source
This commit is contained in:
psychedelicious
2023-06-13 20:51:35 +10:00
parent abee37eab3
commit 09f396ce84
8 changed files with 80 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/index.ts
View File

@ -73,6 +73,7 @@ import { addImageCategoriesChangedListener } from './listeners/imageCategoriesCh
import { addControlNetImageProcessedListener } from './listeners/controlNetImageProcessed';
import { addControlNetAutoProcessListener } from './listeners/controlNetAutoProcess';
import { addUpdateImageUrlsOnConnectListener } from './listeners/updateImageUrlsOnConnect';
import { addInvocationWarningEventListener } from './listeners/socketio/socketInvocationWarning';
export const listenerMiddleware = createListenerMiddleware();
@ -149,6 +150,7 @@ addGeneratorProgressListener();
addGraphExecutionStateCompleteListener();
addInvocationCompleteListener();
addInvocationErrorListener();
addInvocationWarningEventListener();
addInvocationStartedListener();
addSocketConnectedListener();
addSocketDisconnectedListener();

21
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketInvocationWarning.ts Normal file
View File

@ -0,0 +1,21 @@
import { startAppListening } from '../..';
import { log } from 'app/logging/useLogger';
import {
appSocketInvocationWarning,
socketInvocationWarning,
} from 'services/events/actions';
const moduleLog = log.child({ namespace: 'socketio' });
export const addInvocationWarningEventListener = () => {
startAppListening({
actionCreator: socketInvocationWarning,
effect: (action, { dispatch, getState }) => {
moduleLog.warn(
action.payload,
`Invocation warning (${action.payload.data.node.type})`
);
dispatch(appSocketInvocationWarning(action.payload));
},
});
};