From 6135548534fcfacc0ffa7bd1d1f7d32ce05f137d Mon Sep 17 00:00:00 2001
From: David Luzar <5153846+dwelle@users.noreply.github.com>
Date: Wed, 21 Jan 2026 14:07:48 +0100
Subject: [PATCH] fix: rerender TTD preview on message remove (#10681)

---
 .../TTDDialog/hooks/useMermaidRenderer.ts         | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/packages/excalidraw/components/TTDDialog/hooks/useMermaidRenderer.ts b/packages/excalidraw/components/TTDDialog/hooks/useMermaidRenderer.ts
index a554a5b257..9f169c06f1 100644
--- a/packages/excalidraw/components/TTDDialog/hooks/useMermaidRenderer.ts
+++ b/packages/excalidraw/components/TTDDialog/hooks/useMermaidRenderer.ts
@@ -161,23 +161,18 @@ export const useMermaidRenderer = ({
   useEffect(() => {
     if (lastAssistantMessage?.content && lastAssistantMessage?.isGenerating) {
       throttledRenderMermaid(lastAssistantMessage.content);
-    }
-  }, [
-    throttledRenderMermaid,
-    lastAssistantMessage?.isGenerating,
-    lastAssistantMessage?.content,
-  ]);
-
-  // make sure the last bits are rendered once the streaming is completed
-  useEffect(() => {
-    if (!lastAssistantMessage?.isGenerating) {
+    } else if (!lastAssistantMessage?.isGenerating) {
       throttledRenderMermaid.flush();
       resetThrottleState();
+      if (lastAssistantMessage?.content) {
+        throttledRenderMermaid(lastAssistantMessage.content);
+      }
     }
   }, [
     resetThrottleState,
     throttledRenderMermaid,
     lastAssistantMessage?.isGenerating,
+    lastAssistantMessage?.content,
   ]);
 
   // render the last message if the user navigates between the existing chats