diff --git a/src/api/providers/__tests__/xai.spec.ts b/src/api/providers/__tests__/xai.spec.ts index 44c3f26f6c..ab02b3b2f4 100644 --- a/src/api/providers/__tests__/xai.spec.ts +++ b/src/api/providers/__tests__/xai.spec.ts @@ -10,19 +10,11 @@ vitest.mock("@roo-code/telemetry", () => ({ }, })) -const mockResponsesCreate = vitest.fn() - -vitest.mock("openai", () => { - const mockConstructor = vitest.fn() - - return { - __esModule: true, - default: mockConstructor.mockImplementation(function () { - return { - responses: { create: mockResponsesCreate }, - } - }), - } +const mockResponsesCreate = vitest.hoisted(() => vitest.fn()) + +vitest.mock("openai", async () => { + const { mockOpenAiResponsesClient } = await import("../../../test-utils/api") + return mockOpenAiResponsesClient(mockResponsesCreate) }) import OpenAI from "openai"