refactor(logger): move logger into AgentStepContext\n\nCentralizes logging in AgentStepContext to simplify step signatures and improve context-aware tracing. This reduces parameter surface area and ensures logging is consistently tied to agent context.\n\n🤖 Generated with Codebuff\nCo-Authored-By: Codebuff <noreply@codebuff.com>

Author: brandonkachen

.agents/__tests__/context-pruner.test.ts

@@ -72,7 +72,10 @@ describe('context-pruner handleSteps', () => {
       warn: () => {},
       error: () => {},
     }
-    const generator = contextPruner.handleSteps!({ agentState: mockAgentState }, mockLogger)
+    const generator = contextPruner.handleSteps!({
+      agentState: mockAgentState,
+      logger: mockLogger,
+    })
     const results: any[] = []
     let result = generator.next()
     while (!result.done) {
@@ -336,7 +339,10 @@ describe('context-pruner edge cases', () => {
       warn: () => {},
       error: () => {},
     }
-    const generator = contextPruner.handleSteps!({ agentState: mockAgentState }, mockLogger)
+    const generator = contextPruner.handleSteps!({
+      agentState: mockAgentState,
+      logger: mockLogger,
+    })
     const results: ReturnType<typeof generator.next>['value'][] = []
     let result = generator.next()
     while (!result.done) {

.agents/types/agent-definition.ts

@@ -155,7 +155,7 @@ export interface AgentDefinition {
    * Or use 'return' to end the turn.
    *
    * Example 1:
-   * function* handleSteps({ agentStep, prompt, params}, logger) {
+   * function* handleSteps({ agentState, prompt, params, logger }) {
    *   logger.info('Starting file read process')
    *   const { toolResult } = yield {
    *     toolName: 'read_files',
@@ -174,7 +174,7 @@ export interface AgentDefinition {
    * }
    *
    * Example 2:
-   * handleSteps: function* ({ agentState, prompt, params }, logger) {
+   * handleSteps: function* ({ agentState, prompt, params, logger }) {
    *   while (true) {
    *     logger.debug('Spawning thinker agent')
    *     yield {
@@ -193,10 +193,7 @@ export interface AgentDefinition {
    * }
    * }
    */
-  handleSteps?: (
-    context: AgentStepContext,
-    logger: Logger,
-  ) => Generator<
+  handleSteps?: (context: AgentStepContext) => Generator<
     ToolCall | 'STEP' | 'STEP_ALL',
     void,
     {
@@ -230,6 +227,7 @@ export interface AgentStepContext {
   agentState: AgentState
   prompt?: string
   params?: Record<string, any>
+  logger: Logger
 }
 
 /**

backend/src/run-programmatic-step.ts

@@ -118,20 +118,19 @@ export async function runProgrammaticStep(
           agentState,
           prompt,
           params,
+          logger: streamingLogger,
         },
         undefined, // config
-        streamingLogger, // pass the streaming logger instance
+        streamingLogger, // pass the streaming logger instance for internal use
       )
     } else {
       // Initialize native generator
-      generator = (template.handleSteps as any)(
-        {
-          agentState,
-          prompt,
-          params,
-        },
-        streamingLogger,
-      )
+      generator = (template.handleSteps as any)({
+        agentState,
+        prompt,
+        params,
+        logger: streamingLogger,
+      })
       runIdToGenerator[agentState.runId] = generator
     }
   }

backend/src/util/quickjs-sandbox.ts

@@ -143,8 +143,10 @@ export class QuickJSSandbox {
         // Agent function
         const handleSteps = ${generatorCode};
         
-        // Create generator instance
-        let generator = handleSteps(${JSON.stringify(initialInput)}, logger);
+        // Create generator instance with logger injected into context
+        const context = ${JSON.stringify(initialInput)};
+        context.logger = logger;
+        let generator = handleSteps(context);
         
         // Generator management
         globalThis._generator = generator;

common/src/templates/initial-agents-dir/types/agent-definition.ts

@@ -155,7 +155,7 @@ export interface AgentDefinition {
    * Or use 'return' to end the turn.
    *
    * Example 1:
-   * function* handleSteps({ agentStep, prompt, params}, logger) {
+   * function* handleSteps({ agentState, prompt, params, logger }) {
    *   logger.info('Starting file read process')
    *   const { toolResult } = yield {
    *     toolName: 'read_files',
@@ -174,7 +174,7 @@ export interface AgentDefinition {
    * }
    *
    * Example 2:
-   * handleSteps: function* ({ agentState, prompt, params }, logger) {
+   * handleSteps: function* ({ agentState, prompt, params, logger }) {
    *   while (true) {
    *     logger.debug('Spawning thinker agent')
    *     yield {
@@ -193,10 +193,7 @@ export interface AgentDefinition {
    * }
    * }
    */
-  handleSteps?: (
-    context: AgentStepContext,
-    logger: Logger,
-  ) => Generator<
+  handleSteps?: (context: AgentStepContext) => Generator<
     ToolCall | 'STEP' | 'STEP_ALL',
     void,
     {
@@ -230,6 +227,7 @@ export interface AgentStepContext {
   agentState: AgentState
   prompt?: string
   params?: Record<string, any>
+  logger: Logger
 }
 
 /**