From f00630ad20f46785e8c7223ded441cb711e2a1d7 Mon Sep 17 00:00:00 2001
From: Angie Jones <jones.angie@gmail.com>
Date: Fri, 13 Feb 2026 18:21:55 -0600
Subject: [PATCH] fix: fork session (#7219)

---
 ui/desktop/src/components/ProgressiveMessageList.tsx | 2 +-
 ui/desktop/src/components/UserMessage.tsx            | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ui/desktop/src/components/ProgressiveMessageList.tsx b/ui/desktop/src/components/ProgressiveMessageList.tsx
index 05b41dbc3d..b1fa8ae029 100644
--- a/ui/desktop/src/components/ProgressiveMessageList.tsx
+++ b/ui/desktop/src/components/ProgressiveMessageList.tsx
@@ -36,7 +36,7 @@ interface ProgressiveMessageListProps {
   // Custom render function for messages
   renderMessage?: (message: Message, index: number) => React.ReactNode | null;
   isStreamingMessage?: boolean; // Whether messages are currently being streamed
-  onMessageUpdate?: (messageId: string, newContent: string) => void;
+  onMessageUpdate?: (messageId: string, newContent: string, editType?: 'fork' | 'edit') => void;
   onRenderingComplete?: () => void; // Callback when all messages are rendered
   submitElicitationResponse?: (
     elicitationId: string,
diff --git a/ui/desktop/src/components/UserMessage.tsx b/ui/desktop/src/components/UserMessage.tsx
index 2b9bda8bc1..a46d400c53 100644
--- a/ui/desktop/src/components/UserMessage.tsx
+++ b/ui/desktop/src/components/UserMessage.tsx
@@ -80,7 +80,7 @@ export default function UserMessage({ message, onMessageUpdate }: UserMessagePro
 
       setIsEditing(false);
 
-      if (editContent.trim() === textContent.trim()) {
+      if (editType === 'edit' && editContent.trim() === textContent.trim()) {
         return;
       }