fix: mock both language models and safe generator

Co-Authored-By: Han Xiao <[email protected]>

src/tools/__tests__/evaluator.test.ts

@@ -66,16 +66,42 @@ jest.mock('@ai-sdk/google', () => ({
   GoogleChatLanguageModel: jest.fn()
 }));
 
-// Mock the safe generator to prevent actual API calls
-jest.mock('../../utils/safe-generator', () => ({
-  ObjectGeneratorSafe: jest.fn().mockImplementation(() => ({
+// Mock the language models
+jest.mock('@ai-sdk/google', () => ({
+  GoogleChatLanguageModel: jest.fn().mockImplementation(() => ({
     generateObject: jest.fn().mockImplementation(async () => ({
       object: mockEvalResponse,
       usage: { totalTokens: 0, promptTokens: 0, completionTokens: 0 }
     }))
   }))
 }));
 
+// Mock the safe generator to prevent actual API calls
+jest.mock('../../utils/safe-generator', () => ({
+  ObjectGeneratorSafe: jest.fn().mockImplementation(() => ({
+    generateObject: jest.fn().mockImplementation(async (params: any) => {
+      const { schema } = params;
+      // Return different responses based on the schema
+      if (schema.description?.includes('question requires freshness')) {
+        return {
+          object: {
+            needsFreshness: false,
+            needsPlurality: false,
+            reasoning: 'Simple test question',
+            languageStyle: 'plain English'
+          },
+          usage: { totalTokens: 0, promptTokens: 0, completionTokens: 0 }
+        };
+      }
+      // Default response for other schemas
+      return {
+        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(() => ({