Skip to content

Commit

Permalink
fix: mock OpenAI API calls at a lower level
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 7972291 commit 9249441
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion src/tools/__tests__/evaluator.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,19 @@ jest.mock('@ai-sdk/google', () => ({
}));

jest.mock('@ai-sdk/openai', () => ({
OpenAIChatLanguageModel: jest.fn()
OpenAIChatLanguageModel: jest.fn().mockImplementation(() => ({
doGenerate: jest.fn().mockResolvedValue({
object: mockEvalResponse,
usage: { totalTokens: 0, promptTokens: 0, completionTokens: 0 },
finishReason: 'stop',
warnings: [],
request: {},
response: {},
cached: false,
created: Date.now(),
model: 'test-model'
})
}))
}));

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

0 comments on commit 9249441

Please sign in to comment.