diff --git a/src/tools/__tests__/evaluator.test.ts b/src/tools/__tests__/evaluator.test.ts index 0d681061..03fa1b20 100644 --- a/src/tools/__tests__/evaluator.test.ts +++ b/src/tools/__tests__/evaluator.test.ts @@ -66,21 +66,6 @@ jest.mock('@ai-sdk/google', () => ({ GoogleChatLanguageModel: jest.fn() })); -// Mock provider utils to prevent API calls -jest.mock('@ai-sdk/provider-utils', () => ({ - postToApi: jest.fn().mockResolvedValue({ - data: { - choices: [{ - message: { - function_call: { - arguments: JSON.stringify(mockEvalResponse) - } - } - }] - } - }) -})); - // Mock OpenAI API jest.mock('@ai-sdk/openai', () => ({ OpenAIChatLanguageModel: jest.fn().mockImplementation(() => ({ @@ -91,6 +76,14 @@ jest.mock('@ai-sdk/openai', () => ({ })) })); +// Mock response handler to prevent API calls +jest.mock('@ai-sdk/provider-utils/src/response-handler', () => ({ + handleResponse: jest.fn().mockImplementation(() => ({ + object: mockEvalResponse, + usage: { totalTokens: 0 } + })) +})); + jest.mock('../../utils/token-tracker'); // Mock readUrl to prevent actual API calls