Refactor read-docs, web-search, and str-replace to pass logger as parameter

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

Author: charleslien

backend/src/tools/batch-str-replace.ts

@@ -358,6 +358,7 @@ async function executeSingleStrReplace(params: {
       toolCall,
       requestClientToolCall,
       writeToClient: onResponseChunk,
+      logger,
       getLatestState: () => getFileProcessingValues(isolatedState),
       state: isolatedState,
     })

backend/src/tools/handlers/tool/read-docs.ts

@@ -1,7 +1,7 @@
 import { fetchContext7LibraryDocumentation } from '../../../llm-apis/context7-api'
-import { logger } from '../../../util/logger'
 
 import type { CodebuffToolHandlerFunction } from '../handler-function-type'
+import type { Logger } from '@codebuff/types/logger'
 import type {
   CodebuffToolCall,
   CodebuffToolOutput,
@@ -10,13 +10,15 @@ import type {
 export const handleReadDocs = (({
   previousToolCallFinished,
   toolCall,
+  logger,
   agentStepId,
   clientSessionId,
   userInputId,
   state,
 }: {
   previousToolCallFinished: Promise<void>
   toolCall: CodebuffToolCall<'read_docs'>
+  logger: Logger
 
   agentStepId: string
   clientSessionId: string

backend/src/tools/handlers/tool/str-replace.ts

@@ -1,9 +1,9 @@
 import { getFileProcessingValues, postStreamProcessing } from './write-file'
 import { processStrReplace } from '../../../process-str-replace'
-import { logger } from '../../../util/logger'
 import { requestOptionalFile } from '../../../websockets/websocket-action'
 
 import type { CodebuffToolHandlerFunction } from '../handler-function-type'
+import type { Logger } from '@codebuff/types/logger'
 import type {
   FileProcessingState,
   OptionalFileProcessingState,
@@ -22,6 +22,7 @@ export const handleStrReplace = ((params: {
     toolCall: ClientToolCall<'str_replace'>,
   ) => Promise<CodebuffToolOutput<'str_replace'>>
   writeToClient: (chunk: string) => void
+  logger: Logger
 
   getLatestState: () => FileProcessingState
   state: {
@@ -36,6 +37,7 @@ export const handleStrReplace = ((params: {
     toolCall,
     requestClientToolCall,
     writeToClient,
+    logger,
     getLatestState,
     state,
   } = params

backend/src/tools/handlers/tool/web-search.ts

@@ -3,9 +3,9 @@ import { consumeCreditsWithFallback } from '@codebuff/billing'
 import { getRequestContext } from '../../../context/app-context'
 import { searchWeb } from '../../../llm-apis/linkup-api'
 import { PROFIT_MARGIN } from '../../../llm-apis/message-cost-tracker'
-import { logger } from '../../../util/logger'
 
 import type { CodebuffToolHandlerFunction } from '../handler-function-type'
+import type { Logger } from '@codebuff/types/logger'
 import type {
   CodebuffToolCall,
   CodebuffToolOutput,
@@ -14,6 +14,7 @@ import type {
 export const handleWebSearch = ((params: {
   previousToolCallFinished: Promise<void>
   toolCall: CodebuffToolCall<'web_search'>
+  logger: Logger
 
   agentStepId: string
   clientSessionId: string
@@ -28,6 +29,7 @@ export const handleWebSearch = ((params: {
   const {
     previousToolCallFinished,
     toolCall,
+    logger,
     agentStepId,
     clientSessionId,
     userInputId,