Fix web search: pass api key through

Author: jahooma

packages/agent-runtime/src/llm-api/codebuff-web-api.ts

@@ -1,6 +1,5 @@
 import { withTimeout } from '@codebuff/common/util/promise'
 import { env } from '@codebuff/common/env'
-
 import type { Logger } from '@codebuff/common/types/contracts/logger'
 
 const FETCH_TIMEOUT_MS = 30_000
@@ -68,7 +67,8 @@ export async function callWebSearchAPI(params: {
     if (data && typeof data.result === 'string') {
       return {
         result: data.result,
-        creditsUsed: typeof data.creditsUsed === 'number' ? data.creditsUsed : undefined,
+        creditsUsed:
+          typeof data.creditsUsed === 'number' ? data.creditsUsed : undefined,
       }
     }
     if (data && typeof data.error === 'string') return { error: data.error }
@@ -154,7 +154,8 @@ export async function callDocsSearchAPI(params: {
     if (data && typeof data.documentation === 'string') {
       return {
         documentation: data.documentation,
-        creditsUsed: typeof data.creditsUsed === 'number' ? data.creditsUsed : undefined,
+        creditsUsed:
+          typeof data.creditsUsed === 'number' ? data.creditsUsed : undefined,
       }
     }
     if (data && typeof data.error === 'string') return { error: data.error }

packages/agent-runtime/src/tools/handlers/handler-function-type.ts

@@ -30,6 +30,7 @@ export type CodebuffToolHandlerFunction<T extends ToolName = ToolName> = (
     repoUrl: string | undefined
     repoId: string | undefined
     fileContext: ProjectFileContext
+    apiKey: string
 
     signal: AbortSignal
 

packages/agent-runtime/src/tools/handlers/tool/web-search.ts

@@ -12,6 +12,7 @@ export const handleWebSearch = ((params: {
   previousToolCallFinished: Promise<void>
   toolCall: CodebuffToolCall<'web_search'>
   logger: Logger
+  apiKey: string
 
   agentStepId: string
   clientSessionId: string
@@ -36,6 +37,7 @@ export const handleWebSearch = ((params: {
     repoUrl,
     state,
     fetch,
+    apiKey
   } = params
   const { query, depth } = toolCall.input
   const { userId, fingerprintId, repoId } = state
@@ -63,6 +65,7 @@ export const handleWebSearch = ((params: {
           repoUrl: repoUrl ?? null,
           fetch,
           logger,
+          apiKey
         })
 
         if (webApi.error) {