fix: rerender TTD preview on message remove (#10681)

This commit is contained in:
David Luzar
2026-01-21 14:07:48 +01:00
committed by GitHub
parent acf54c6f38
commit 6135548534
@@ -161,23 +161,18 @@ export const useMermaidRenderer = ({
useEffect(() => { useEffect(() => {
if (lastAssistantMessage?.content && lastAssistantMessage?.isGenerating) { if (lastAssistantMessage?.content && lastAssistantMessage?.isGenerating) {
throttledRenderMermaid(lastAssistantMessage.content); throttledRenderMermaid(lastAssistantMessage.content);
} } else if (!lastAssistantMessage?.isGenerating) {
}, [
throttledRenderMermaid,
lastAssistantMessage?.isGenerating,
lastAssistantMessage?.content,
]);
// make sure the last bits are rendered once the streaming is completed
useEffect(() => {
if (!lastAssistantMessage?.isGenerating) {
throttledRenderMermaid.flush(); throttledRenderMermaid.flush();
resetThrottleState(); resetThrottleState();
if (lastAssistantMessage?.content) {
throttledRenderMermaid(lastAssistantMessage.content);
}
} }
}, [ }, [
resetThrottleState, resetThrottleState,
throttledRenderMermaid, throttledRenderMermaid,
lastAssistantMessage?.isGenerating, lastAssistantMessage?.isGenerating,
lastAssistantMessage?.content,
]); ]);
// render the last message if the user navigates between the existing chats // render the last message if the user navigates between the existing chats