fix(cli): collapse set_output tool toggle by default

Author: brandonkachen

cli/src/components/blocks/tool-branch.tsx

@@ -3,6 +3,7 @@ import { memo, useCallback } from 'react'
 import { ContentWithMarkdown } from './content-with-markdown'
 import { useTheme } from '../../hooks/use-theme'
 import { getToolDisplayInfo } from '../../utils/codebuff-client'
+import { shouldCollapseToolByDefault } from '../../utils/constants'
 import { renderToolComponent } from '../tools/registry'
 import { ToolCallItem } from '../tools/tool-call-item'
 
@@ -43,7 +44,7 @@ export const ToolBranch = memo(
     }
 
     const displayInfo = getToolDisplayInfo(toolBlock.toolName)
-    const isCollapsed = toolBlock.isCollapsed ?? false
+    const isCollapsed = toolBlock.isCollapsed ?? shouldCollapseToolByDefault(toolBlock.toolName)
     const isStreaming = streamingAgents.has(toolBlock.toolCallId)
 
     const inputContent = `\`\`\`json\n${JSON.stringify(toolBlock.input, null, 2)}\n\`\`\``

cli/src/utils/constants.ts

@@ -1,6 +1,21 @@
+import type { ToolName } from '@codebuff/sdk'
+
 // Agent IDs that should not be rendered in the CLI UI
 export const HIDDEN_AGENT_IDS = ['codebuff/context-pruner'] as const
 
+// Tool names that should be collapsed by default when rendered
+// Uses ToolName type to ensure only valid tool names are added
+export const COLLAPSED_BY_DEFAULT_TOOL_NAMES: readonly ToolName[] = [
+  'set_output',
+] as const
+
+/**
+ * Check if a tool should be collapsed by default
+ */
+export const shouldCollapseToolByDefault = (toolName: string): boolean => {
+  return COLLAPSED_BY_DEFAULT_TOOL_NAMES.includes(toolName as ToolName)
+}
+
 /**
  * Check if an agent ID should be hidden from rendering
  */