diff --git a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketConnected.ts b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketConnected.ts index fe4bce682b..f01c3911da 100644 --- a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketConnected.ts +++ b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketConnected.ts @@ -1,6 +1,7 @@ import { log } from 'app/logging/useLogger'; -import { appSocketConnected, socketConnected } from 'services/events/actions'; +import { modelsApi } from 'services/api/endpoints/models'; import { receivedOpenAPISchema } from 'services/api/thunks/schema'; +import { appSocketConnected, socketConnected } from 'services/events/actions'; import { startAppListening } from '../..'; const moduleLog = log.child({ namespace: 'socketio' }); @@ -23,6 +24,13 @@ export const addSocketConnectedEventListener = () => { // pass along the socket event as an application action dispatch(appSocketConnected(action.payload)); + + // update all server state + dispatch(modelsApi.endpoints.getMainModels.initiate()); + dispatch(modelsApi.endpoints.getControlNetModels.initiate()); + dispatch(modelsApi.endpoints.getLoRAModels.initiate()); + dispatch(modelsApi.endpoints.getTextualInversionModels.initiate()); + dispatch(modelsApi.endpoints.getVaeModels.initiate()); }, }); };