Use max agent steps constant in more places

Author: jahooma

backend/src/loop-main-prompt.ts

@@ -1,3 +1,4 @@
+import { MAX_AGENT_STEPS_DEFAULT } from '@codebuff/common/constants/agents'
 import { mainPrompt } from './main-prompt'
 
 import type { MainPromptOptions } from './main-prompt'
@@ -9,8 +10,6 @@ import type {
 } from '@codebuff/common/types/session-state'
 import type { WebSocket } from 'ws'
 
-const DEFAULT_MAX_ITERATIONS = 20
-
 export async function loopMainPrompt(
   ws: WebSocket,
   action: ClientAction<'prompt'>,
@@ -20,7 +19,7 @@ export async function loopMainPrompt(
   toolCalls: Array<ClientToolCall>
   toolResults: Array<ToolResult>
 }> {
-  const maxIterations = options.maxIterations ?? DEFAULT_MAX_ITERATIONS
+  const maxIterations = options.maxIterations ?? MAX_AGENT_STEPS_DEFAULT
   let { sessionState, toolResults, toolCalls } = await mainPrompt(
     ws,
     action,

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

@@ -15,6 +15,7 @@ import type {
 } from '@codebuff/common/types/session-state'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
 import type { WebSocket } from 'ws'
+import { MAX_AGENT_STEPS_DEFAULT } from '@codebuff/common/constants/agents'
 
 export const handleSpawnAgentInline = ((params: {
   previousToolCallFinished: Promise<void>
@@ -135,7 +136,7 @@ export const handleSpawnAgentInline = ((params: {
       agentContext: agentState!.agentContext, // Inherit parent context directly
       subagents: [],
       messageHistory: getLatestState().messages, // Share the same message array
-      stepsRemaining: 20, // MAX_AGENT_STEPS
+      stepsRemaining: MAX_AGENT_STEPS_DEFAULT,
       output: undefined,
       parentId: agentState!.agentId,
     }

backend/src/tools/handlers/tool/spawn-agents-async.ts

@@ -18,6 +18,7 @@ import type {
 } from '@codebuff/common/types/session-state'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
 import type { WebSocket } from 'ws'
+import { MAX_AGENT_STEPS_DEFAULT } from '@codebuff/common/constants/agents'
 
 export const handleSpawnAgentsAsync = ((params: {
   previousToolCallFinished: Promise<void>
@@ -184,7 +185,7 @@ export const handleSpawnAgentsAsync = ((params: {
           agentContext: {},
           subagents: [],
           messageHistory: subAgentMessages,
-          stepsRemaining: 20, // MAX_AGENT_STEPS
+          stepsRemaining: MAX_AGENT_STEPS_DEFAULT,
           output: undefined,
           // Add parent ID to agent state for communication
           parentId: agentState!.agentId,

backend/src/tools/handlers/tool/spawn-agents.ts

@@ -14,6 +14,7 @@ import type {
 } from '@codebuff/common/types/session-state'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
 import type { WebSocket } from 'ws'
+import { MAX_AGENT_STEPS_DEFAULT } from '@codebuff/common/constants/agents'
 
 export type SendSubagentChunk = (data: {
   userInputId: string
@@ -163,7 +164,7 @@ export const handleSpawnAgents = ((params: {
           agentContext: {},
           subagents: [],
           messageHistory: subAgentMessages,
-          stepsRemaining: 20, // MAX_AGENT_STEPS
+          stepsRemaining: MAX_AGENT_STEPS_DEFAULT,
           output: undefined,
           parentId: agentState!.agentId,
         }