diff --git a/src/cron/isolated-agent/run.test-harness.ts b/src/cron/isolated-agent/run.test-harness.ts index 4daa91fee7e..48c2c5fd65f 100644 --- a/src/cron/isolated-agent/run.test-harness.ts +++ b/src/cron/isolated-agent/run.test-harness.ts @@ -87,15 +87,19 @@ vi.mock("../../agents/model-catalog.js", () => ({ loadModelCatalog: vi.fn().mockResolvedValue({ models: [] }), })); -vi.mock("../../agents/model-selection.js", () => ({ - getModelRefStatus: getModelRefStatusMock, - isCliProvider: isCliProviderMock, - normalizeModelSelection: normalizeModelSelectionForTest, - resolveAllowedModelRef: resolveAllowedModelRefMock, - resolveConfiguredModelRef: resolveConfiguredModelRefMock, - resolveHooksGmailModel: resolveHooksGmailModelMock, - resolveThinkingDefault: resolveThinkingDefaultMock, -})); +vi.mock("../../agents/model-selection.js", async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, + getModelRefStatus: getModelRefStatusMock, + isCliProvider: isCliProviderMock, + normalizeModelSelection: normalizeModelSelectionForTest, + resolveAllowedModelRef: resolveAllowedModelRefMock, + resolveConfiguredModelRef: resolveConfiguredModelRefMock, + resolveHooksGmailModel: resolveHooksGmailModelMock, + resolveThinkingDefault: resolveThinkingDefaultMock, + }; +}); vi.mock("../../agents/model-fallback.js", () => ({ runWithModelFallback: runWithModelFallbackMock,