fix: mock safe generator at a higher level

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