@@ -2,7 +2,11 @@ import { has, isEqual } from 'lodash'
22import { useCallback , useEffect , useRef } from 'react'
33
44import { getCodebuffClient , formatToolOutput } from '../utils/codebuff-client'
5- import { MAIN_AGENT_ID , shouldHideAgent } from '../utils/constants'
5+ import {
6+ MAIN_AGENT_ID ,
7+ shouldHideAgent ,
8+ shouldCollapseByDefault ,
9+ } from '../utils/constants'
610import { createValidationErrorBlocks } from '../utils/create-validation-error-blocks'
711import { getErrorObject } from '../utils/error'
812import { formatTimestamp } from '../utils/helpers'
@@ -1130,10 +1134,13 @@ export const useSendMessage = ({
11301134 setCollapsedAgents ( ( prev ) => {
11311135 const next = new Set ( prev )
11321136 next . delete ( tempId )
1133- // Only collapse if parent is NOT main agent (i.e., it's a nested agent)
1137+ // Collapse if:
1138+ // 1. Parent is NOT main agent (nested agent), OR
1139+ // 2. Agent type is in the collapsed-by-default list
11341140 if (
1135- event . parentAgentId &&
1136- event . parentAgentId !== MAIN_AGENT_ID
1141+ ( event . parentAgentId &&
1142+ event . parentAgentId !== MAIN_AGENT_ID ) ||
1143+ shouldCollapseByDefault ( event . agentType )
11371144 ) {
11381145 next . add ( event . agentId )
11391146 }
@@ -1233,10 +1240,13 @@ export const useSendMessage = ({
12331240 )
12341241
12351242 setStreamingAgents ( ( prev ) => new Set ( prev ) . add ( event . agentId ) )
1236- // Only collapse if parent is NOT main agent (i.e., it's a nested agent)
1243+ // Collapse if:
1244+ // 1. Parent is NOT main agent (nested agent), OR
1245+ // 2. Agent type is in the collapsed-by-default list
12371246 if (
1238- event . parentAgentId &&
1239- event . parentAgentId !== MAIN_AGENT_ID
1247+ ( event . parentAgentId &&
1248+ event . parentAgentId !== MAIN_AGENT_ID ) ||
1249+ shouldCollapseByDefault ( event . agentType )
12401250 ) {
12411251 setCollapsedAgents ( ( prev ) =>
12421252 new Set ( prev ) . add ( event . agentId ) ,
0 commit comments