From 3d21f97422cd51153bd0c073f189ae07aa1830b1 Mon Sep 17 00:00:00 2001
From: Brooklyn Nicholson <brooklyn.bb.nicholson@gmail.com>
Date: Sun, 26 Apr 2026 14:06:42 -0500
Subject: [PATCH] fix(tui): keep live tool state before stream segments

---
 ui-tui/src/components/appLayout.tsx | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/ui-tui/src/components/appLayout.tsx b/ui-tui/src/components/appLayout.tsx
index b302fed66f..c4739c0555 100644
--- a/ui-tui/src/components/appLayout.tsx
+++ b/ui-tui/src/components/appLayout.tsx
@@ -36,21 +36,8 @@ const StreamingAssistant = memo(function StreamingAssistant({
 
   return (
     <>
-      {progress.streamSegments.map((msg, i) => (
-        <MessageLine
-          cols={cols}
-          compact={compact}
-          detailsMode={detailsMode}
-          detailsModeCommandOverride={detailsModeCommandOverride}
-          key={`seg:${i}`}
-          msg={msg}
-          sections={sections}
-          t={t}
-        />
-      ))}
-
       {progress.showProgressArea && (
-        <Box flexDirection="column" marginBottom={progress.showStreamingArea ? 1 : 0}>
+        <Box flexDirection="column" marginBottom={progress.streamSegments.length || progress.showStreamingArea ? 1 : 0}>
           <ToolTrail
             activity={progress.activity}
             busy={busy}
@@ -71,6 +58,19 @@ const StreamingAssistant = memo(function StreamingAssistant({
         </Box>
       )}
 
+      {progress.streamSegments.map((msg, i) => (
+        <MessageLine
+          cols={cols}
+          compact={compact}
+          detailsMode={detailsMode}
+          detailsModeCommandOverride={detailsModeCommandOverride}
+          key={`seg:${i}`}
+          msg={msg}
+          sections={sections}
+          t={t}
+        />
+      ))}
+
       {progress.showStreamingArea && (
         <MessageLine
           cols={cols}