inline agent inherits onReponseChunk

Author: jahooma

backend/src/run-programmatic-step.ts

@@ -200,7 +200,6 @@ export async function runProgrammaticStep(
           role: 'assistant' as const,
           content: toolCallString,
         })
-        onResponseChunk(toolCallString)
         state.sendSubagentChunk({
           userInputId,
           agentId: agentState.agentId,

backend/src/tools/handlers/handler-function-type.ts

@@ -4,6 +4,7 @@ import type {
   ClientToolName,
   CodebuffToolCall,
 } from '@codebuff/common/tools/list'
+import { PrintModeEvent } from '@codebuff/common/types/print-mode'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
 
 type PresentOrAbsent<K extends PropertyKey, V> =
@@ -22,7 +23,7 @@ export type CodebuffToolHandlerFunction<T extends ToolName = ToolName> = (
 
     fullResponse: string
 
-    writeToClient: (chunk: string) => void
+    writeToClient: (chunk: string | PrintModeEvent) => void
 
     getLatestState: () => any
     state: { [K in string]?: any }

backend/src/tools/handlers/tool/spawn-agent-inline.ts

@@ -6,7 +6,6 @@ import {
   executeAgent,
 } from './spawn-agent-utils'
 import { MAX_AGENT_STEPS_DEFAULT } from '@codebuff/common/constants/agents'
-import { expireMessages } from '../../../util/messages'
 
 import type { CodebuffToolHandlerFunction } from '../handler-function-type'
 import type { CodebuffToolCall } from '@codebuff/common/tools/list'
@@ -24,6 +23,7 @@ export const handleSpawnAgentInline = ((params: {
   fileContext: ProjectFileContext
   clientSessionId: string
   userInputId: string
+  writeToClient: (chunk: string | PrintModeEvent) => void
 
   getLatestState: () => { messages: CodebuffMessage[] }
   state: {
@@ -44,6 +44,7 @@ export const handleSpawnAgentInline = ((params: {
     userInputId,
     getLatestState,
     state,
+    writeToClient,
   } = params
   const {
     agent_type: agentTypeStr,
@@ -102,10 +103,8 @@ export const handleSpawnAgentInline = ((params: {
       localAgentTemplates,
       userId,
       clientSessionId,
-      onResponseChunk: (chunk: string | PrintModeEvent) => {
-        // Child agent output is streamed directly to parent's output
-        // No need for special handling since we share message history
-      },
+      // Inherits parent's onResponseChunk
+      onResponseChunk: writeToClient,
     })
 
     // Update parent's message history with child's final state