Function updateToolBlockWithOutput

cli/src/utils/message-block-helpers.ts:592–622

(
  blocks: ContentBlock[],
  options: UpdateToolBlockOptions,
)

 * Recursively processes nested agent blocks.
 */
export const updateToolBlockWithOutput = (
  blocks: ContentBlock[],
  options: UpdateToolBlockOptions,
): ContentBlock[] => {
  const { toolCallId, toolOutput } = options

  return blocks.map((block) => {
    if (block.type === 'tool' && block.toolCallId === toolCallId) {
      let output: string
      if (block.toolName === 'run_terminal_command') {
        const parsed = (toolOutput?.[0] as any)?.value
        if (parsed?.stdout || parsed?.stderr) {
          output = (parsed.stdout || '') + (parsed.stderr || '')
        } else {
          output = formatToolOutput(toolOutput)
        }
      } else {
        output = formatToolOutput(toolOutput)
      }
      return { ...block, output }
    } else if (block.type === 'agent' && block.blocks) {
      const updatedBlocks = updateToolBlockWithOutput(block.blocks, options)
      // Avoid creating new block if nested blocks didn't change
      if (isEqual(block.blocks, updatedBlocks)) {
        return block
      }
      return { ...block, blocks: updatedBlocks }
    }
    return block
  })
}