diff --git a/packages/ui/src/components/tool-call.tsx b/packages/ui/src/components/tool-call.tsx index c6f51592..458de294 100644 --- a/packages/ui/src/components/tool-call.tsx +++ b/packages/ui/src/components/tool-call.tsx @@ -601,6 +601,9 @@ export default function ToolCall(props: ToolCallProps) { const next = !prev if (!next) { setInputSectionExpanded(false) + } else { + setInputSectionExpanded(true) + setOutputSectionExpanded(true) } return next }) @@ -881,44 +884,69 @@ export default function ToolCall(props: ToolCallProps) { {expanded() && (