diff --git a/src/tools/__tests__/evaluator.test.ts b/src/tools/__tests__/evaluator.test.ts
index 38797808..03da503d 100644
--- a/src/tools/__tests__/evaluator.test.ts
+++ b/src/tools/__tests__/evaluator.test.ts
@@ -3,10 +3,14 @@ import { TokenTracker } from '../../utils/token-tracker';
 import { LLMProvider } from '../../config';
 import { jest } from '@jest/globals';
 import { EvaluationResponse } from '../../types';
-import { GoogleChatLanguageModel } from '@ai-sdk/google';
-import { OpenAIChatLanguageModel } from '@ai-sdk/openai';
+import { LanguageModelUsage } from 'ai';
+import { GenerateObjectResult, GenerateOptions } from 'ai';
 import { ObjectGeneratorSafe } from '../../utils/safe-generator';
 
+// Mock these as classes since we don't need their actual implementations
+class GoogleChatLanguageModel {}
+class OpenAIChatLanguageModel {}
+
 jest.mock('@ai-sdk/google');
 jest.mock('@ai-sdk/openai');
 jest.mock('../../utils/safe-generator');
@@ -37,10 +41,12 @@ describe('evaluateAnswer', () => {
     jest.resetModules();
     process.env = { ...originalEnv };
 
-    const mockGenerateObject = jest.fn().mockResolvedValue({ object: mockEvalResponse });
+    const mockGenerateObject = jest.fn().mockResolvedValue({ 
+      object: mockEvalResponse 
+    } as GenerateObjectResult<EvaluationResponse>);
 
     jest.mocked(TokenTracker).mockImplementation(() => ({
-      getTotalUsage: jest.fn().mockReturnValue({ totalTokens: 0 })
+      getTotalUsage: jest.fn().mockReturnValue({ totalTokens: 0 } as LanguageModelUsage)
     }));
 
     jest.mocked(ObjectGeneratorSafe).mockImplementation(() => ({