Refactor think-deeply, create-plan, and find-files to pass logger as parameter

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

Author: charleslien

backend/src/tools/handlers/tool/create-plan.ts

@@ -2,9 +2,9 @@ import { trackEvent } from '@codebuff/common/analytics'
 import { AnalyticsEvent } from '@codebuff/common/constants/analytics-events'
 
 import { getFileProcessingValues, postStreamProcessing } from './write-file'
-import { logger } from '../../../util/logger'
 
 import type { CodebuffToolHandlerFunction } from '../handler-function-type'
+import type { Logger } from '@codebuff/types/logger'
 import type {
   FileProcessingState,
   OptionalFileProcessingState,
@@ -22,6 +22,7 @@ export const handleCreatePlan = ((params: {
     toolCall: ClientToolCall<'create_plan'>,
   ) => Promise<CodebuffToolOutput<'create_plan'>>
   writeToClient: (chunk: string) => void
+  logger: Logger
 
   getLatestState: () => FileProcessingState
   state: {
@@ -41,6 +42,7 @@ export const handleCreatePlan = ((params: {
     toolCall,
     requestClientToolCall,
     writeToClient,
+    logger,
     getLatestState,
     state,
   } = params

backend/src/tools/handlers/tool/find-files.ts

@@ -6,13 +6,13 @@ import {
 } from '../../../find-files/request-files-prompt'
 import { getFileReadingUpdates } from '../../../get-file-reading-updates'
 import { getSearchSystemPrompt } from '../../../system-prompt/search-system-prompt'
-import { logger } from '../../../util/logger'
 import { renderReadFilesResult } from '../../../util/parse-tool-call-xml'
 import { countTokens, countTokensJson } from '../../../util/token-counter'
 import { requestFiles } from '../../../websockets/websocket-action'
 
 import type { TextBlock } from '../../../llm-apis/claude'
 import type { CodebuffToolHandlerFunction } from '../handler-function-type'
+import type { Logger } from '@codebuff/types/logger'
 import type { GetExpandedFileContextForTrainingBlobTrace } from '@codebuff/bigquery'
 import type {
   CodebuffToolCall,
@@ -29,6 +29,7 @@ const COLLECT_FULL_FILE_CONTEXT = false
 export const handleFindFiles = ((params: {
   previousToolCallFinished: Promise<any>
   toolCall: CodebuffToolCall<'find_files'>
+  logger: Logger
 
   fileContext: ProjectFileContext
   agentStepId: string
@@ -46,6 +47,7 @@ export const handleFindFiles = ((params: {
   const {
     previousToolCallFinished,
     toolCall,
+    logger,
     fileContext,
     agentStepId,
     clientSessionId,
@@ -113,6 +115,7 @@ export const handleFindFiles = ((params: {
           userInputId,
           userId,
           repoId,
+          logger,
         ).catch((error) => {
           logger.error(
             { error },
@@ -178,6 +181,7 @@ async function uploadExpandedFileContextForTraining(
   userInputId: string,
   userId: string | undefined,
   repoId: string | undefined,
+  logger: Logger,
 ) {
   const files = await requestRelevantFilesForTraining({
     messages,

backend/src/tools/handlers/tool/think-deeply.ts

@@ -1,6 +1,5 @@
-import { logger } from '../../../util/logger'
-
 import type { CodebuffToolHandlerFunction } from '../handler-function-type'
+import type { Logger } from '@codebuff/types/logger'
 import type {
   CodebuffToolCall,
   CodebuffToolOutput,
@@ -9,8 +8,9 @@ import type {
 export const handleThinkDeeply = ((params: {
   previousToolCallFinished: Promise<any>
   toolCall: CodebuffToolCall<'think_deeply'>
+  logger: Logger
 }): { result: Promise<CodebuffToolOutput<'think_deeply'>>; state: {} } => {
-  const { previousToolCallFinished, toolCall } = params
+  const { previousToolCallFinished, toolCall, logger } = params
   const { thought } = toolCall.input
 
   logger.debug(