fix types

Author: jahooma

packages/agent-runtime/src/__tests__/malformed-tool-call.test.ts

@@ -78,6 +78,7 @@ describe('malformed tool call error handling', () => {
       system: 'Test system prompt',
       agentContext: {},
       onResponseChunk: mock(() => {}),
+      onCostCalculated: mock(async () => {}),
       fullResponse: '',
     }
 

packages/agent-runtime/src/__tests__/run-programmatic-step.test.ts

@@ -127,6 +127,7 @@ describe('runProgrammaticStep', () => {
       clientSessionId: 'test-session',
       fingerprintId: 'test-fingerprint',
       onResponseChunk: () => {},
+      onCostCalculated: async () => {},
       fileContext: mockFileContext,
       localAgentTemplates: {},
       system: undefined,

packages/agent-runtime/src/__tests__/sandbox-generator.test.ts

@@ -82,6 +82,7 @@ describe('QuickJS Sandbox Generator', () => {
       clientSessionId: 'test-session',
       fingerprintId: 'test-fingerprint',
       onResponseChunk: () => {},
+      onCostCalculated: async () => {},
       fileContext: mockFileContext,
       localAgentTemplates: {},
       stepsComplete: false,