From d872cf5009a46776e05df187eca0a1847e71e41b Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Sat, 27 Dec 2025 18:36:13 +0530
Subject: [PATCH] feat(chat-hook): prevent duplicate blocks

---
 src/lib/hooks/useChat.tsx | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/lib/hooks/useChat.tsx b/src/lib/hooks/useChat.tsx
index fdb5743..11a94b5 100644
--- a/src/lib/hooks/useChat.tsx
+++ b/src/lib/hooks/useChat.tsx
@@ -570,6 +570,20 @@ export const ChatProvider = ({ children }: { children: React.ReactNode }) => {
         setMessages((prev) =>
           prev.map((msg) => {
             if (msg.messageId === messageId) {
+              const exists = msg.responseBlocks.findIndex(
+                (b) => b.id === data.block.id,
+              );
+
+              if (exists !== -1) {
+                const existingBlocks = [...msg.responseBlocks];
+                existingBlocks[exists] = data.block;
+
+                return {
+                  ...msg,
+                  responseBlocks: existingBlocks,
+                };
+              }
+
               return {
                 ...msg,
                 responseBlocks: [...msg.responseBlocks, data.block],