Skip to content

Commit e6821cd

Browse files
fix: mock Google language model for tests
Co-Authored-By: Han Xiao <[email protected]>
1 parent 17c33e0 commit e6821cd

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/tools/__tests__/evaluator.test.ts

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,16 @@ jest.mock('@ai-sdk/google', () => ({
6666
GoogleChatLanguageModel: jest.fn()
6767
}));
6868

69+
// Mock the language models
70+
jest.mock('@ai-sdk/google', () => ({
71+
GoogleChatLanguageModel: jest.fn().mockImplementation(() => ({
72+
generateObject: jest.fn().mockResolvedValue({
73+
object: mockEvalResponse,
74+
usage: { totalTokens: 0, promptTokens: 0, completionTokens: 0 }
75+
})
76+
}))
77+
}));
78+
6979
// Mock safe generator to prevent actual API calls
7080
jest.mock('../../utils/safe-generator', () => ({
7181
ObjectGeneratorSafe: jest.fn().mockImplementation(() => ({

0 commit comments

Comments
 (0)