From 21658764013e07a5e7d192b252a8aa0db00a3212 Mon Sep 17 00:00:00 2001
From: Zane Staggs <zane@squareup.com>
Date: Thu, 9 Oct 2025 15:48:42 -0700
Subject: [PATCH] cleanup and fix messages re-rendering while chat is streaming

---
 ui/desktop/src/components/ProgressiveMessageList.tsx | 2 +-
 ui/desktop/src/hooks/useSessionStream.ts             | 5 -----
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/ui/desktop/src/components/ProgressiveMessageList.tsx b/ui/desktop/src/components/ProgressiveMessageList.tsx
index 9bdd0ab689..24c8c74541 100644
--- a/ui/desktop/src/components/ProgressiveMessageList.tsx
+++ b/ui/desktop/src/components/ProgressiveMessageList.tsx
@@ -189,7 +189,7 @@ export default function ProgressiveMessageList({
 
         return (
           <div
-            key={message.id && `${message.id}-${message.content.length}`}
+            key={message.id || `message-${index}`}
             className={`relative ${index === 0 ? 'mt-0' : 'mt-4'} ${isUser ? 'user' : 'assistant'}`}
             data-testid="message-container"
           >
diff --git a/ui/desktop/src/hooks/useSessionStream.ts b/ui/desktop/src/hooks/useSessionStream.ts
index f76adca3f0..397ae5d260 100644
--- a/ui/desktop/src/hooks/useSessionStream.ts
+++ b/ui/desktop/src/hooks/useSessionStream.ts
@@ -3,10 +3,6 @@ import { Session } from '../api';
 import { useSessionStreamContext } from '../contexts/SessionStreamContext';
 
 interface UseSessionStreamOptions {
-  /**
-   * Whether to enable streaming. If false, will not connect.
-   * @default true
-   */
   enabled?: boolean;
   onUpdate?: (session: Session) => void;
   onError?: (error: string) => void;
@@ -45,7 +41,6 @@ export function useSessionStream(
   const onUpdateRef = useRef(onUpdate);
   const onErrorRef = useRef(onError);
 
-  // Keep refs up to date
   useEffect(() => {
     onUpdateRef.current = onUpdate;
     onErrorRef.current = onError;