Fix type errors: add required tools property to call sites

Author: brandonkachen

backend/src/__tests__/cost-aggregation.integration.test.ts

@@ -222,6 +222,7 @@ describe('Cost Aggregation Integration Tests', () => {
       onResponseChunk: () => {},
       localAgentTemplates: mockLocalAgentTemplates,
       signal: new AbortController().signal,
+      tools: {},
     }
 
     callMainPromptBaseParams = {
@@ -232,6 +233,7 @@ describe('Cost Aggregation Integration Tests', () => {
       promptId: 'test-prompt',
       clientSessionId: 'test-session',
       signal: new AbortController().signal,
+      tools: {},
     }
 
     // Mock getAgentTemplate to return our mock templates

backend/src/__tests__/main-prompt.integration.test.ts

@@ -374,6 +374,7 @@ export function getMessagesSubset(messages: Message[], otherTokens: number) {
         process.stdout.write(chunk)
       },
       signal: new AbortController().signal,
+      tools: {},
     })
     const requestToolCallSpy = agentRuntimeImpl.requestToolCall as any
 
@@ -462,6 +463,7 @@ export function getMessagesSubset(messages: Message[], otherTokens: number) {
           process.stdout.write(chunk)
         },
         signal: new AbortController().signal,
+        tools: {},
       })
 
       const requestToolCallSpy = agentRuntimeImpl.requestToolCall as any

backend/src/websockets/middleware.ts

@@ -158,7 +158,7 @@ export class WebSocketMiddleware {
         ws: WebSocket
       } & AgentRuntimeDeps &
         AgentRuntimeScopedDeps,
-    ) => void
+    ) => void | Promise<void>
     silent?: boolean
   }) {
     const { baseAction, silent } = params

backend/src/websockets/websocket-action.ts

@@ -102,7 +102,7 @@ const onPrompt = async (
     logger: Logger
   } & ParamsExcluding<
     typeof callMainPrompt,
-    'userId' | 'promptId' | 'repoId' | 'repoUrl' | 'signal'
+    'userId' | 'promptId' | 'repoId' | 'repoUrl' | 'signal' | 'tools'
   >,
 ) => {
   const { action, ws, getUserInfoFromApiKey, logger } = params
@@ -151,6 +151,7 @@ const onPrompt = async (
           repoUrl,
           repoId,
           signal: new AbortController().signal,
+          tools: {},
         })
         if (result.output.type === 'error') {
           throw new Error(result.output.message)

sdk/src/run.ts

@@ -841,6 +841,7 @@ export async function runOnce({
     clientSessionId: promptId,
     userId,
     signal: signal ?? new AbortController().signal,
+    tools: {},
   }).catch((error) => {
     // Let retryable errors and PaymentRequiredError propagate so the retry wrapper can handle them
     const isRetryable = isRetryableError(error)