github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): ensure we clean up when session id changes

Browse Source
This commit is contained in:
psychedelicious
2025-07-18 18:05:08 +10:00
committed by Kent Keirsey
parent 89fa43a3b6
commit bed01941a5
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
invokeai/frontend/web/src/features/controlLayers/components/StagingArea/context.tsx
View File

@@ -15,7 +15,7 @@ import { selectBboxRect, selectSelectedEntityIdentifier } from 'features/control
import type { CanvasRasterLayerState } from 'features/controlLayers/store/types';
import { imageNameToImageObject } from 'features/controlLayers/store/util';
import type { PropsWithChildren } from 'react';
import { createContext, memo, useContext, useMemo } from 'react';
import { createContext, memo, useContext, useEffect, useMemo } from 'react';
import { getImageDTOSafe } from 'services/api/endpoints/images';
import { queueApi } from 'services/api/endpoints/queue';
import type { S } from 'services/api/types';
@@ -94,7 +94,16 @@ export const StagingAreaContextProvider = memo(({ children, sessionId }: PropsWi
return _stagingAreaAppApi;
}, [sessionId, socket, store]);
const value = useMemo(() => new StagingAreaApi(sessionId, stagingAreaAppApi), [sessionId, stagingAreaAppApi]);
const value = useMemo(() => {
return new StagingAreaApi(sessionId, stagingAreaAppApi);
}, [sessionId, stagingAreaAppApi]);
useEffect(() => {
const api = value;
return () => {
api.cleanup();
};
}, [value]);
return <StagingAreaContext.Provider value={value}>{children}</StagingAreaContext.Provider>;
});