add finishAgentRun to AgentRuntimeDeps

Author: charleslien

backend/src/impl/agent-runtime.ts

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

backend/src/run-agent-step.ts

@@ -45,7 +45,10 @@ import type {
 } from '@codebuff/common/types/session-state'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
 import type { ParamsExcluding } from '@codebuff/types/common'
-import type { StartAgentRunFn } from '@codebuff/types/database'
+import type {
+  FinishAgentRunFn,
+  StartAgentRunFn,
+} from '@codebuff/types/database'
 import type { Logger } from '@codebuff/types/logger'
 import type { WebSocket } from 'ws'
 
@@ -424,6 +427,7 @@ export const loopAgentSteps = async (
     onResponseChunk: (chunk: string | PrintModeEvent) => void
 
     startAgentRun: StartAgentRunFn
+    finishAgentRun: FinishAgentRunFn
     logger: Logger
   } & ParamsExcluding<
     typeof runProgrammaticStep,

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

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

evals/impl/agent-runtime.ts

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

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

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