All files / api/src/connectors/evaluations/__mocks__ openai.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import { vi } from 'vitest';
 
// Mock OpenAI client for evaluation tests
export const mockParse = vi.fn();
export const mockWithOptions = vi.fn().mockReturnValue({
  chat: {
    completions: {
      parse: mockParse,
    },
  },
});
 
// Setup the OpenAI mock
export function setupOpenAIMock() {
  vi.mock('openai', () => {
    return {
      default: vi.fn().mockImplementation(() => ({
        chat: {
          completions: {
            parse: mockParse,
          },
        },
        withOptions: mockWithOptions,
      })),
    };
  });
}
 
// Helper to mock a successful evaluation response
export function mockSuccessfulEvaluation(score: number, reasoning: string) {
  mockParse.mockResolvedValueOnce({
    choices: [
      {
        message: {
          parsed: {
            score,
            reasoning,
          },
        },
      },
    ],
  });
}
 
// Helper to reset all mocks
export function resetOpenAIMocks() {
  mockParse.mockReset();
  mockWithOptions.mockClear();
}