feat(app): hide reasoning once agent is done

1 files changed, 8 insertions, 1 deletions

diff --git a/packages/ui/src/components/session-turn.tsx b/packages/ui/src/components/session-turn.tsx
index e2169102f..751327385 100644
--- a/packages/ui/src/components/session-turn.tsx
+++ b/packages/ui/src/components/session-turn.tsx
@@ -78,6 +78,7 @@ function AssistantMessageItem(props: {
   message: AssistantMessage
   responsePartId: string | undefined
   hideResponsePart: boolean
+  hideReasoning: boolean
 }) {
   const data = useData()
   const emptyParts: PartType[] = []
@@ -92,7 +93,12 @@ function AssistantMessageItem(props: {
   })
 
   const filteredParts = createMemo(() => {
-    const parts = msgParts()
+    let parts = msgParts()
+
+    if (props.hideReasoning) {
+      parts = parts.filter((part) => part?.type !== "reasoning")
+    }
+
     if (!props.hideResponsePart) return parts
 
     const responsePartId = props.responsePartId
@@ -556,6 +562,7 @@ export function SessionTurn(
                               message={assistantMessage}
                               responsePartId={responsePartId()}
                               hideResponsePart={hideResponsePart()}
+                              hideReasoning={!working()}
                             />
                           )}
                         </For>