@@ -751,6 +751,10 @@ const AssistantMessage: FC = () => {
751751 const planToolTraceDetailText = useMemo ( ( ) => buildPlanToolTraceDetailText ( planToolTrace ) , [ planToolTrace ] ) ;
752752 const hasPlanToolTrace = phase === "plan" && planToolTrace . length > 0 ;
753753
754+ if ( phase === "progress" ) {
755+ return null ;
756+ }
757+
754758 return (
755759 < MessagePrimitive . Root
756760 className = "aui-assistant-message-root fade-in slide-in-from-bottom-1 relative mx-auto w-full max-w-(--thread-max-width) animate-in py-3 duration-150"
@@ -759,7 +763,7 @@ const AssistantMessage: FC = () => {
759763 data-running-message = { isRunning ? "true" : "false" }
760764 >
761765 < div className = "aui-assistant-message-content wrap-break-word px-2 text-foreground leading-relaxed" >
762- { phase === "progress" && ! hasPlanToolTrace ? < ProgressMessageCard text = { progressText } /> : < MessagePrimitive . Parts components = { TOOL_PART_COMPONENTS } /> }
766+ < MessagePrimitive . Parts components = { TOOL_PART_COMPONENTS } />
763767 { hasPlanToolTrace ? < PlanToolTraceCard trace = { planToolTrace } isRunning = { isRunning } detailText = { planToolTraceDetailText || progressText } /> : null }
764768 { experimentProposal ? < ExperimentApprovalCard proposal = { experimentProposal } disabled = { isRunning } /> : null }
765769 { isRunning && ! hasVisibleParts && ! experimentProposal ? (
@@ -778,25 +782,6 @@ const AssistantMessage: FC = () => {
778782 ) ;
779783} ;
780784
781- const ProgressMessageCard : FC < {
782- text : string ;
783- } > = ( { text } ) => {
784- return (
785- < section className = "aui-agentic-trace" aria-label = "Agentic Search run" >
786- < div className = "aui-agentic-trace-header is-static" >
787- < div className = "aui-agentic-trace-heading" >
788- < div className = "aui-agentic-trace-title-row" >
789- < span className = "aui-agentic-trace-title" > Agentic Search</ span >
790- < span className = "aui-agentic-trace-spinner" aria-hidden = "true" />
791- < span className = "aui-agentic-trace-status" > In Progress</ span >
792- { text . trim ( ) . length > 0 ? < DetailedRunOutputButton text = { text } /> : null }
793- </ div >
794- </ div >
795- </ div >
796- </ section >
797- ) ;
798- } ;
799-
800785const DetailedRunOutputButton : FC < {
801786 text : string ;
802787} > = ( { text } ) => {
0 commit comments