Skip to content

Commit

Permalink
fix: improve OpenAI mock implementation
Browse files Browse the repository at this point in the history
Co-Authored-By: Han Xiao <[email protected]>
  • Loading branch information
devin-ai-integration[bot] and hanxiao committed Feb 13, 2025
1 parent 5a275ae commit bd29ec1
Showing 1 changed file with 11 additions and 11 deletions.
22 changes: 11 additions & 11 deletions src/tools/__tests__/evaluator.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -67,18 +67,18 @@ jest.mock('@ai-sdk/google', () => ({
}));

// Mock OpenAI API
jest.mock('@ai-sdk/openai', () => ({
OpenAIChatLanguageModel: jest.fn().mockImplementation(() => ({
generateObject: jest.fn().mockImplementation(async () => ({
object: mockEvalResponse,
usage: { totalTokens: 0 }
})),
doGenerate: jest.fn().mockImplementation(async () => ({
object: mockEvalResponse,
usage: { totalTokens: 0 }
jest.mock('@ai-sdk/openai', () => {
const mockResponse = {
object: mockEvalResponse,
usage: { totalTokens: 0 }
};
return {
OpenAIChatLanguageModel: jest.fn().mockImplementation(() => ({
generateObject: jest.fn().mockResolvedValue(mockResponse),
doGenerate: jest.fn().mockResolvedValue(mockResponse)
}))
}))
}));
};
});

jest.mock('../../utils/token-tracker');

Expand Down

0 comments on commit bd29ec1

Please sign in to comment.