Hide editor-implementors / selectors.

Author: jahooma

cli/src/components/message-block.tsx

@@ -2,6 +2,7 @@ import { TextAttributes } from '@opentui/core'
 import { pluralize } from '@codebuff/common/util/string'
 import React, { memo, useCallback, useMemo, type ReactNode } from 'react'
 
+import { shouldRenderAsSimpleText } from '../utils/constants'
 import { AgentBranchItem } from './agent-branch-item'
 import { ElapsedTimer } from './elapsed-timer'
 import { FeedbackIconButton } from './feedback-icon-button'
@@ -28,6 +29,7 @@ import { ContentWithMarkdown } from './blocks/content-with-markdown'
 import { ToolBranch } from './blocks/tool-branch'
 import { PlanBox } from './renderers/plan-box'
 import { AgentListBranch } from './blocks/agent-list-branch'
+import { BULLET_CHAR } from '../utils/strings'
 
 interface MessageBlockProps {
   messageId: string
@@ -85,7 +87,7 @@ export const MessageBlock = memo((props: MessageBlockProps): ReactNode => {
   })
 
   const theme = useTheme()
-  
+
   // Memoize selectors to prevent new function references on every render
   const selectIsFeedbackOpenMemo = useMemo(
     () => selectIsFeedbackOpenForMessage(messageId),
@@ -99,10 +101,12 @@ export const MessageBlock = memo((props: MessageBlockProps): ReactNode => {
     () => selectMessageFeedbackCategory(messageId),
     [messageId],
   )
-  
+
   const isFeedbackOpen = useFeedbackStore(selectIsFeedbackOpenMemo)
   const hasSubmittedFeedback = useFeedbackStore(selectHasSubmittedFeedbackMemo)
-  const selectedFeedbackCategory = useFeedbackStore(selectMessageFeedbackCategoryMemo)
+  const selectedFeedbackCategory = useFeedbackStore(
+    selectMessageFeedbackCategoryMemo,
+  )
 
   const resolvedTextColor = textColor ?? theme.foreground
   const shouldShowLoadingTimer = isAi && isLoading && !isComplete
@@ -578,6 +582,28 @@ const AgentBranchWrapper = memo(
     onBuildMax,
   }: AgentBranchWrapperProps) => {
     const theme = useTheme()
+
+    if (shouldRenderAsSimpleText(agentBlock.agentType)) {
+      return (
+        <box
+          key={keyPrefix}
+          style={{
+            flexDirection: 'column',
+          }}
+        >
+          <text
+            style={{
+              wrapMode: 'word',
+              fg: theme.muted,
+            }}
+            attributes={TextAttributes.ITALIC}
+          >
+            {BULLET_CHAR} Selecting the best implementation...
+          </text>
+        </box>
+      )
+    }
+
     const isCollapsed = agentBlock.isCollapsed ?? false
     const isStreaming =
       agentBlock.status === 'running' || streamingAgents.has(agentBlock.agentId)
@@ -626,11 +652,11 @@ const AgentBranchWrapper = memo(
     const nParameterMessage =
       agentBlock.params?.n !== undefined &&
       (agentBlock.agentType.includes('editor-best-of-n')
-        ? `Generating ${agentBlock.params.n} implementations...`
+        ? `${BULLET_CHAR} Generating ${agentBlock.params.n} implementations...`
         : agentBlock.agentType.includes('thinker-best-of-n')
-          ? `Generating ${agentBlock.params.n} deep thoughts...`
+          ? `${BULLET_CHAR} Generating ${agentBlock.params.n} deep thoughts...`
           : agentBlock.agentType.includes('code-reviewer-best-of-n')
-            ? `Generating ${agentBlock.params.n} code reviews...`
+            ? `${BULLET_CHAR} Generating ${agentBlock.params.n} code reviews...`
             : undefined)
 
     return (

cli/src/hooks/use-send-message.ts

@@ -1419,8 +1419,12 @@ export const useSendMessage = ({
                         ? [...msg.blocks]
                         : []
 
-                      const newAgentBlocks: ContentBlock[] = agents.map(
-                        (agent: any, index: number) => ({
+                      const newAgentBlocks: ContentBlock[] = agents
+                        .filter(
+                          (agent: any) =>
+                            !shouldHideAgent(agent.agent_type || ''),
+                        )
+                        .map((agent: any, index: number) => ({
                           type: 'agent',
                           agentId: `${toolCallId}-${index}`,
                           agentName: agent.agent_type || 'Agent',
@@ -1434,8 +1438,7 @@ export const useSendMessage = ({
                           ) && {
                             isCollapsed: true,
                           }),
-                        }),
-                      )
+                        }))
 
                       return {
                         ...msg,

cli/src/utils/constants.ts

@@ -1,5 +1,10 @@
 // Agent IDs that should not be rendered in the CLI UI
-export const HIDDEN_AGENT_IDS = ['codebuff/context-pruner'] as const
+export const HIDDEN_AGENT_IDS = [
+  'codebuff/context-pruner',
+  'editor-implementor',
+  'editor-implementor-gemini',
+  'editor-implementor-gpt-5',
+] as const
 
 /**
  * Check if an agent ID should be hidden from rendering
@@ -25,6 +30,21 @@ export const shouldCollapseByDefault = (agentType: string): boolean => {
   )
 }
 
+// Agent IDs that should render as simple text instead of full agent boxes
+export const SIMPLE_TEXT_AGENT_IDS = [
+  'best-of-n-selector',
+  'best-of-n-selector-gemini',
+] as const
+
+/**
+ * Check if an agent should render as simple text instead of a full agent box
+ */
+export const shouldRenderAsSimpleText = (agentType: string): boolean => {
+  return SIMPLE_TEXT_AGENT_IDS.some((simpleTextId) =>
+    agentType.includes(simpleTextId),
+  )
+}
+
 /**
  * The parent agent ID for all root-level agents
  */

cli/src/utils/strings.ts

@@ -21,3 +21,5 @@ export function getSubsequenceIndices(
 
   return null
 }
+
+export const BULLET_CHAR = '• '