refactor: deduplicate agent-template.ts by importing base types from the user-facing template; update Logger parameter in handleSteps to ensure correct step logging.

🤖 Generated with Codebuff
Co-Authored-By: Codebuff <noreply@codebuff.com>

Author: brandonkachen

backend/src/__tests__/run-programmatic-step.test.ts

@@ -84,10 +84,9 @@ describe('runProgrammaticStep', () => {
     )
 
     // Mock sendAction
-    sendActionSpy = spyOn(
-      websocketAction,
-      'sendAction',
-    ).mockImplementation(() => {})
+    sendActionSpy = spyOn(websocketAction, 'sendAction').mockImplementation(
+      () => {},
+    )
 
     // Mock crypto.randomUUID
     spyOn(crypto, 'randomUUID').mockImplementation(
@@ -118,7 +117,8 @@ describe('runProgrammaticStep', () => {
     mockAgentState = {
       ...sessionState.mainAgentState,
       agentId: 'test-agent-id',
-      runId: 'test-run-id' as `${string}-${string}-${string}-${string}-${string}`,
+      runId:
+        'test-run-id' as `${string}-${string}-${string}-${string}-${string}`,
       messageHistory: [
         { role: 'user', content: 'Initial message' },
         { role: 'assistant', content: 'Initial response' },

common/src/templates/agent-validation.ts

@@ -284,6 +284,7 @@ export function validateSingleAgent(
       stepPrompt: validatedConfig.stepPrompt ?? '',
       outputSchema,
       inputSchema,
+
     }
 
     return {

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

@@ -196,15 +196,17 @@ export interface AgentDefinition {
   handleSteps?: (
     context: AgentStepContext,
     logger: Logger,
-  ) => Generator<
-    ToolCall | 'STEP' | 'STEP_ALL',
-    void,
-    {
-      agentState: AgentState
-      toolResult: ToolResultOutput[] | undefined
-      stepsComplete: boolean
-    }
-  >
+  ) =>
+    | Generator<
+        ToolCall | 'STEP' | 'STEP_ALL',
+        void,
+        {
+          agentState: AgentState
+          toolResult: ToolResultOutput[] | undefined
+          stepsComplete: boolean
+        }
+      >
+    | string
 }
 
 // ============================================================================

common/src/types/agent-template.ts

@@ -1,16 +1,30 @@
+/**
+ * Backend Agent Template Types
+ * 
+ * This file provides backend-compatible agent template types with strict validation.
+ * It imports base types from the user-facing template to eliminate duplication.
+ */
+
 import type { Model } from '../old-constants'
 import type { ToolResultOutput } from './messages/content-part'
 import type { AgentState, AgentTemplateType } from './session-state'
 import type {
+  AgentDefinition,
   ToolCall,
   AgentState as PublicAgentState,
+  Logger,
+  AgentStepContext,
 } from '../templates/initial-agents-dir/types/agent-definition'
 import type { ToolName } from '../tools/constants'
 import type { OpenRouterProviderOptions } from '@codebuff/internal/openrouter-ai-sdk'
 import type { z } from 'zod/v4'
 
 export type AgentId = `${string}/${string}@${number}.${number}.${number}`
 
+/**
+ * Backend agent template with strict validation and Zod schemas
+ * Extends the user-facing AgentDefinition but with backend-specific requirements
+ */
 export type AgentTemplate<
   P = string | undefined,
   T = Record<string, any> | undefined,
@@ -59,11 +73,4 @@ export type StepHandler<
   logger: Logger,
 ) => StepGenerator
 
-export interface Logger {
-  debug: (data: any, msg?: string) => void
-  info: (data: any, msg?: string) => void
-  warn: (data: any, msg?: string) => void
-  error: (data: any, msg?: string) => void
-}
-
-export { PublicAgentState }
+export { Logger, PublicAgentState }