add addAgentStep to AgentRuntimeDeps

Author: charleslien

backend/src/impl/agent-runtime.ts

@@ -1,4 +1,4 @@
-import { finishAgentRun, startAgentRun } from '../agent-run'
+import { addAgentStep, finishAgentRun, startAgentRun } from '../agent-run'
 import { logger } from '../util/logger'
 
 import type { AgentRuntimeDeps } from '@codebuff/types/deps/agent-runtime'
@@ -8,4 +8,5 @@ export const backendAgentRuntimeImpl: AgentRuntimeDeps = {
 
   startAgentRun,
   finishAgentRun,
+  addAgentStep,
 }

backend/src/run-agent-step.ts

@@ -7,7 +7,6 @@ import { buildArray } from '@codebuff/common/util/array'
 import { getErrorObject } from '@codebuff/common/util/error'
 import { cloneDeep } from 'lodash'
 
-import { addAgentStep, finishAgentRun } from './agent-run'
 import { checkLiveUserInput } from './live-user-inputs'
 import { getMCPToolData } from './mcp/util'
 import { getAgentStreamFromTemplate } from './prompt-agent-stream'
@@ -46,6 +45,7 @@ import type {
 import type { ProjectFileContext } from '@codebuff/common/util/file'
 import type { ParamsExcluding } from '@codebuff/types/common'
 import type {
+  AddAgentStepFn,
   FinishAgentRunFn,
   StartAgentRunFn,
 } from '@codebuff/types/database'
@@ -428,6 +428,7 @@ export const loopAgentSteps = async (
 
     startAgentRun: StartAgentRunFn
     finishAgentRun: FinishAgentRunFn
+    addAgentStep: AddAgentStepFn
     logger: Logger
   } & ParamsExcluding<
     typeof runProgrammaticStep,
@@ -460,6 +461,8 @@ export const loopAgentSteps = async (
     clearUserPromptMessagesAfterResponse = true,
     parentSystemPrompt,
     startAgentRun,
+    finishAgentRun,
+    addAgentStep,
     logger,
   } = params
 

common/src/testing/impl/agent-runtime.ts

@@ -11,8 +11,7 @@ export const testLogger: Logger = {
 export const testAgentRuntimeImpl: AgentRuntimeDeps = {
   logger: testLogger,
 
-  startAgentRun: async () => {
-    return 'test-agent-run-id'
-  },
+  startAgentRun: async () => 'test-agent-run-id',
   finishAgentRun: async () => {},
+  addAgentStep: async () => 'test-agent-step-id',
 }

evals/impl/agent-runtime.ts

@@ -5,4 +5,5 @@ export const evalAgentRuntimeImpl: AgentRuntimeDeps = {
 
   startAgentRun: async () => 'test-agent-run-id',
   finishAgentRun: async () => {},
+  addAgentStep: async () => 'test-agent-step-id',
 }

packages/types/src/deps/agent-runtime.ts

@@ -1,9 +1,14 @@
-import type { FinishAgentRunFn, StartAgentRunFn } from '../database'
+import type {
+  AddAgentStepFn,
+  FinishAgentRunFn,
+  StartAgentRunFn,
+} from '../database'
 import type { Logger } from '../logger'
 
 export type AgentRuntimeDeps = {
   logger: Logger
 
   startAgentRun: StartAgentRunFn
   finishAgentRun: FinishAgentRunFn
+  addAgentStep: AddAgentStepFn
 }