fix: sdk max agent steps bug

Author: charleslien

common/src/json-config/constants.ts

@@ -64,6 +64,8 @@ export const FileChangeHook = z
   })
   .describe('Defines a single file change hook.')
 
+export const DEFAULT_MAX_AGENT_STEPS = 12
+
 export const CodebuffConfigSchema = z
   .object({
     description: z
@@ -81,7 +83,7 @@ export const CodebuffConfigSchema = z
     maxAgentSteps: z
       .number()
       .optional()
-      .default(12)
+      .default(DEFAULT_MAX_AGENT_STEPS)
       .describe(
         'Maximum number of turns agent will take before being forced to end',
       ),

sdk/CHANGELOG.md

@@ -12,6 +12,10 @@ All notable changes to the @codebuff/sdk package will be documented in this file
 
 - Automatic parsing of `knowledgeFiles` if not provided
 
+### Fixed
+
+- `maxAgentSteps` resets every run
+
 ## [0.1.8] - 2025-08-13
 
 ### Added

sdk/src/client.ts

@@ -11,6 +11,7 @@ import {
   type ServerAction,
 } from '../../common/src/actions'
 import { API_KEY_ENV_VAR } from '../../common/src/constants'
+import { DEFAULT_MAX_AGENT_STEPS } from '../../common/src/json-config/constants'
 
 import type { AgentDefinition } from '../../common/src/templates/initial-agents-dir/types/agent-definition'
 import type { PrintModeEvent } from '../../common/src/types/print-mode'
@@ -131,7 +132,7 @@ export class CodebuffClient {
     projectFiles,
     knowledgeFiles,
     agentDefinitions,
-    maxAgentSteps,
+    maxAgentSteps = DEFAULT_MAX_AGENT_STEPS,
   }: {
     agent: string
     prompt: string
@@ -154,6 +155,7 @@ export class CodebuffClient {
         projectFiles,
         maxAgentSteps,
       })
+    sessionState.mainAgentState.stepsRemaining = maxAgentSteps
     const toolResults = previousRun?.toolResults ?? []
     if (handleEvent) {
       this.promptIdToHandleEvent[promptId] = handleEvent