fix: mock Google language model for tests

Co-Authored-By: Han Xiao <[email protected]>
jina-ai · Feb 13, 2025 · e6821cd · e6821cd
1 parent 17c33e0
commit e6821cd
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/tools/__tests__/evaluator.test.ts b/src/tools/__tests__/evaluator.test.ts
@@ -66,6 +66,16 @@ jest.mock('@ai-sdk/google', () => ({
   GoogleChatLanguageModel: jest.fn()
 }));
 
+// Mock the language models
+jest.mock('@ai-sdk/google', () => ({
+  GoogleChatLanguageModel: jest.fn().mockImplementation(() => ({
+    generateObject: jest.fn().mockResolvedValue({
+      object: mockEvalResponse,
+      usage: { totalTokens: 0, promptTokens: 0, completionTokens: 0 }
+    })
+  }))
+}));
+
 // Mock safe generator to prevent actual API calls
 jest.mock('../../utils/safe-generator', () => ({
   ObjectGeneratorSafe: jest.fn().mockImplementation(() => ({