Refactor clearAgentGeneratorCache to pass logger as parameter

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

Author: charleslien

backend/src/__tests__/loop-agent-steps.test.ts

@@ -101,7 +101,7 @@ describe('loopAgentSteps - runAgentStep vs runProgrammaticStep behavior', () =>
   })
 
   beforeEach(() => {
-    clearAgentGeneratorCache()
+    clearAgentGeneratorCache({ logger })
 
     llmCallCount = 0
 
@@ -172,7 +172,7 @@ describe('loopAgentSteps - runAgentStep vs runProgrammaticStep behavior', () =>
   })
 
   afterEach(() => {
-    clearAgentGeneratorCache()
+    clearAgentGeneratorCache({ logger })
 
     mock.restore()
   })

backend/src/__tests__/run-agent-step-tools.test.ts

@@ -113,15 +113,15 @@ describe('runAgentStep - set_output tool', () => {
     spyOn(aisdk, 'promptAiSdk').mockImplementation(() =>
       Promise.resolve('Test response'),
     )
-    clearAgentGeneratorCache()
+    clearAgentGeneratorCache({ logger })
   })
 
   afterEach(() => {
     mock.restore()
   })
 
   afterAll(() => {
-    clearAgentGeneratorCache()
+    clearAgentGeneratorCache({ logger })
   })
 
   class MockWebSocket {

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

@@ -143,7 +143,7 @@ describe('runProgrammaticStep', () => {
   afterEach(() => {
     mock.restore()
     // Clear the generator cache between tests
-    clearAgentGeneratorCache()
+    clearAgentGeneratorCache({ logger })
   })
 
   describe('generator lifecycle', () => {

backend/src/__tests__/sandbox-generator.test.ts

@@ -30,7 +30,7 @@ describe('QuickJS Sandbox Generator', () => {
   let mockTemplate: AgentTemplate
 
   beforeEach(() => {
-    clearAgentGeneratorCache()
+    clearAgentGeneratorCache({ logger })
 
     // Mock dependencies
     spyOn(agentRun, 'addAgentStep').mockImplementation(
@@ -100,7 +100,7 @@ describe('QuickJS Sandbox Generator', () => {
   })
 
   afterEach(() => {
-    clearAgentGeneratorCache()
+    clearAgentGeneratorCache({ logger })
   })
 
   test('should execute string-based generator in QuickJS sandbox', async () => {

backend/src/run-programmatic-step.ts

@@ -4,7 +4,6 @@ import { cloneDeep } from 'lodash'
 
 import { addAgentStep } from './agent-run'
 import { executeToolCall } from './tools/tool-executor'
-import { logger } from './util/logger'
 import { SandboxManager } from './util/quickjs-sandbox'
 import { getRequestContext } from './websockets/request-context'
 import { sendAction } from './websockets/websocket-action'
@@ -22,6 +21,7 @@ import type {
 import type { PrintModeEvent } from '@codebuff/common/types/print-mode'
 import type { AgentState } from '@codebuff/common/types/session-state'
 import type { ProjectFileContext } from '@codebuff/common/util/file'
+import type { ParamsOf } from '@codebuff/types/common'
 import type { Logger } from '@codebuff/types/logger'
 import type { WebSocket } from 'ws'
 
@@ -33,13 +33,15 @@ const runIdToGenerator: Record<string, StepGenerator | undefined> = {}
 export const runIdToStepAll: Set<string> = new Set()
 
 // Function to clear the generator cache for testing purposes
-export function clearAgentGeneratorCache() {
+export function clearAgentGeneratorCache(
+  params: ParamsOf<typeof sandboxManager.dispose>,
+) {
   for (const key in runIdToGenerator) {
     delete runIdToGenerator[key]
   }
   runIdToStepAll.clear()
   // Clean up QuickJS sandboxes
-  sandboxManager.dispose({ logger })
+  sandboxManager.dispose(params)
 }
 
 // Function to handle programmatic agents