Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-16 16:30:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

fix: Prevent memory leaks in useResumableSSE by clearing handler maps on stream completion and cleanup

Browse source
This commit is contained in:
Danny Avila 2025-12-12 01:15:25 -05:00
parent 1b2d3f30ef
commit 08052844cf
No known key found for this signature in database
GPG key ID: BF31EEB2C5CA0956
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
client/src/hooks/SSE/useResumableSSE.ts
View file

@ -151,6 +151,8 @@ export default function useResumableSSE(
setIsSubmitting(false);
setShowStopButton(false);
}
// Clear handler maps on stream completion to prevent memory leaks
clearStepMaps();
(startupConfig?.balance?.enabled ?? false) && balanceQuery.refetch();
sse.close();
setStreamId(null);
@ -369,6 +371,7 @@ export default function useResumableSSE(
contentHandler,
resetContentHandler,
syncStepMessage,
clearStepMaps,
messageHandler,
errorHandler,
setIsSubmitting,
@ -476,6 +479,8 @@ export default function useResumableSSE(
sseRef.current.close();
sseRef.current = null;
}
// Clear handler maps to prevent memory leaks and stale state
clearStepMaps();
// Reset UI state on cleanup - useResumeOnLoad will restore if needed
setIsSubmitting(false);
setShowStopButton(false);