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): improve session creation handling

Browse Source
This commit is contained in:
psychedelicious
2023-05-26 15:53:08 +10:00
parent 1d4d705795
commit 8f190169db
6 changed files with 99 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
invokeai/frontend/web/src/features/system/store/systemSlice.ts
View File

@ -16,7 +16,11 @@ import {
import { ProgressImage } from 'services/events/types';
import { makeToast } from '../../../app/components/Toaster';
import { sessionCanceled, sessionInvoked } from 'services/thunks/session';
import {
sessionCanceled,
sessionCreated,
sessionInvoked,
} from 'services/thunks/session';
import { receivedModels } from 'services/thunks/model';
import { parsedOpenAPISchema } from 'features/nodes/store/nodesSlice';
import { LogLevelName } from 'roarr';
@ -353,7 +357,7 @@ export const systemSlice = createSlice({
});
/**
* Session Canceled
* Session Canceled - FULFILLED
*/
builder.addCase(sessionCanceled.fulfilled, (state, action) => {
state.canceledSession = action.meta.arg.sessionId;
@ -370,6 +374,23 @@ export const systemSlice = createSlice({
);
});
/**
* Session Created - REJECTED
*/
builder.addCase(sessionCreated.rejected, (state, action) => {
state.isProcessing = false;
state.isCancelable = false;
state.isCancelScheduled = false;
state.currentStep = 0;
state.totalSteps = 0;
state.statusTranslationKey = 'common.statusConnected';
state.progressImage = null;
state.toastQueue.push(
makeToast({ title: t('toast.problemCreatingSession'), status: 'error' })
);
});
/**
* Session Canceled
*/