Fix for editor multi prompt to show the selected strategy and reason!

Author: jahooma

cli/src/components/blocks/agent-branch-wrapper.tsx

@@ -353,15 +353,17 @@ export const AgentBranchWrapper = memo(
           (b): b is ToolContentBlock =>
             b.type === 'tool' && b.toolName === 'set_output',
         )
-        const implementationId = setOutputBlock?.input?.implementationId as string | undefined
+        // set_output wraps data in a 'data' property, so we need to access input.data
+        const outputData = (setOutputBlock?.input as { data?: Record<string, unknown> })?.data
+        const implementationId = outputData?.implementationId as string | undefined
         if (implementationId) {
           const letterIndex = implementationId.charCodeAt(0) - 65
           const implementors = siblingBlocks.filter(
             (b): b is AgentContentBlock =>
               b.type === 'agent' && isImplementorAgent(b),
           )
 
-          reason = setOutputBlock?.input?.reason as string | undefined
+          reason = outputData?.reason as string | undefined
 
           const selectedAgent = implementors[letterIndex]
           if (selectedAgent) {

cli/src/utils/constants.ts

@@ -85,6 +85,7 @@ export const shouldCollapseForParent = (
 export const SIMPLE_TEXT_AGENT_IDS = [
   'best-of-n-selector',
   'best-of-n-selector-gemini',
+  'best-of-n-selector2',
 ] as const
 
 /**

cli/src/utils/implementor-helpers.ts

@@ -533,26 +533,34 @@ export function getMultiPromptProgress(
   }
 }
 
-/** Expected shape of the set_output input from editor-multi-prompt */
-interface MultiPromptSetOutputInput {
+/** Expected shape of the set_output data from editor-multi-prompt */
+interface MultiPromptSetOutputData {
+  implementationId?: string
   chosenStrategy?: string
   reason?: string
   suggestedImprovements?: string
   toolResults?: unknown[]
   error?: string
 }
 
-/** Type guard for MultiPromptSetOutputInput */
-function isMultiPromptSetOutput(input: unknown): input is MultiPromptSetOutputInput {
+/** Expected shape of the set_output input (data is wrapped in a 'data' property) */
+interface SetOutputInput {
+  data?: MultiPromptSetOutputData
+}
+
+/** Type guard for set_output input with data property */
+function hasSetOutputData(input: unknown): input is SetOutputInput {
   return (
     typeof input === 'object' &&
     input !== null &&
-    ('reason' in input || 'chosenStrategy' in input || 'error' in input)
+    'data' in input &&
+    typeof (input as SetOutputInput).data === 'object'
   )
 }
 
 /**
  * Extract the selection reason from multi-prompt agent's set_output block.
+ * set_output wraps data in a 'data' property, so we need to access input.data.reason
  */
 function extractSelectionReason(blocks: ContentBlock[] | undefined): string | null {
   if (!blocks || blocks.length === 0) return null
@@ -561,15 +569,15 @@ function extractSelectionReason(blocks: ContentBlock[] | undefined): string | nu
     (block): block is ToolContentBlock =>
       block.type === 'tool' &&
       block.toolName === 'set_output' &&
-      isMultiPromptSetOutput(block.input) &&
-      typeof block.input.reason === 'string',
+      hasSetOutputData(block.input) &&
+      typeof block.input.data?.reason === 'string',
   )
 
-  if (!setOutputBlock || !isMultiPromptSetOutput(setOutputBlock.input)) {
+  if (!setOutputBlock || !hasSetOutputData(setOutputBlock.input)) {
     return null
   }
 
-  return setOutputBlock.input.reason ?? null
+  return setOutputBlock.input.data?.reason ?? null
 }
 
 /**