diff --git a/src/auto-reply/reply/agent-runner-execution.test.ts b/src/auto-reply/reply/agent-runner-execution.test.ts index 56ab48719ac..7c988bcbbc2 100644 --- a/src/auto-reply/reply/agent-runner-execution.test.ts +++ b/src/auto-reply/reply/agent-runner-execution.test.ts @@ -27,9 +27,15 @@ vi.mock("../../agents/model-fallback.js", () => ({ Array.isArray((err as { attempts?: unknown[] }).attempts), })); -vi.mock("../../agents/model-selection.js", () => ({ - isCliProvider: () => false, -})); +vi.mock("../../agents/model-selection.js", async () => { + const actual = await vi.importActual( + "../../agents/model-selection.js", + ); + return { + ...actual, + isCliProvider: () => false, + }; +}); vi.mock("../../agents/bootstrap-budget.js", () => ({ resolveBootstrapWarningSignaturesSeen: () => [],