tweak best-of-n-editor to only reproduce the changes from selected agent, not commentary

Author: jahooma

.agents/base2/best-of-n/best-of-n-editor.ts

@@ -6,7 +6,6 @@ import type {
   StepText,
   ToolCall,
 } from '../../types/agent-definition'
-
 export function createBestOfNEditor(
   model: 'sonnet' | 'gpt-5',
 ): Omit<SecretAgentDefinition, 'id'> {
@@ -132,10 +131,15 @@ function* handleStepsSonnet({
     return
   }
 
-  // Apply the chosen implementation using STEP_TEXT
+  // Apply the chosen implementation using STEP_TEXT (only tool calls, no commentary)
+  const toolCallsOnly = extractToolCallsOnly(
+    typeof chosenImplementation.content === 'string'
+      ? chosenImplementation.content
+      : '',
+  )
   const { agentState: postEditsAgentState } = yield {
     type: 'STEP_TEXT',
-    text: chosenImplementation.content,
+    text: toolCallsOnly,
   } as StepText
   const { messageHistory } = postEditsAgentState
   const lastAssistantMessageIndex = messageHistory.findLastIndex(
@@ -177,6 +181,19 @@ function* handleStepsSonnet({
         },
     )
   }
+
+  // Extract only tool calls from text, removing any commentary
+  function extractToolCallsOnly(text: string): string {
+    const toolExtractionPattern =
+      /<codebuff_tool_call>\n(.*?)\n<\/codebuff_tool_call>/gs
+    const matches: string[] = []
+
+    for (const match of text.matchAll(toolExtractionPattern)) {
+      matches.push(match[0]) // Include the full tool call with tags
+    }
+
+    return matches.join('\n')
+  }
 }
 
 function* handleStepsGpt5({
@@ -258,10 +275,15 @@ function* handleStepsGpt5({
     return
   }
 
-  // Apply the chosen implementation using STEP_TEXT
+  // Apply the chosen implementation using STEP_TEXT (only tool calls, no commentary)
+  const toolCallsOnly = extractToolCallsOnly(
+    typeof chosenImplementation.content === 'string'
+      ? chosenImplementation.content
+      : '',
+  )
   const { agentState: postEditsAgentState } = yield {
     type: 'STEP_TEXT',
-    text: chosenImplementation.content,
+    text: toolCallsOnly,
   } as StepText
   const { messageHistory } = postEditsAgentState
   const lastAssistantMessageIndex = messageHistory.findLastIndex(
@@ -302,6 +324,19 @@ function* handleStepsGpt5({
         },
     )
   }
+
+  // Extract only tool calls from text, removing any commentary
+  function extractToolCallsOnly(text: string): string {
+    const toolExtractionPattern =
+      /<codebuff_tool_call>\n(.*?)\n<\/codebuff_tool_call>/gs
+    const matches: string[] = []
+
+    for (const match of text.matchAll(toolExtractionPattern)) {
+      matches.push(match[0]) // Include the full tool call with tags
+    }
+
+    return matches.join('\n')
+  }
 }
 
 const definition = {