test: reuse model override normalizer

This commit is contained in:
Peter Steinberger
2026-04-18 22:26:58 +01:00
parent da22866030
commit f0f4fa6978

View File

@@ -25,29 +25,17 @@ vi.mock("./pi-embedded-runner/runs.js", () => ({
state.consumeEmbeddedRunModelSwitchMock(...args),
}));
vi.mock("./model-selection.js", () => ({
normalizeStoredOverrideModel: (params: { providerOverride?: string; modelOverride?: string }) => {
const providerOverride = params.providerOverride?.trim();
const modelOverride = params.modelOverride?.trim();
if (!providerOverride || !modelOverride) {
return {
providerOverride,
modelOverride,
};
}
const providerPrefix = `${providerOverride.toLowerCase()}/`;
return {
providerOverride,
modelOverride: modelOverride.toLowerCase().startsWith(providerPrefix)
? modelOverride.slice(providerOverride.length + 1).trim() || modelOverride
: modelOverride,
};
},
resolveDefaultModelForAgent: (...args: unknown[]) =>
state.resolveDefaultModelForAgentMock(...args),
resolvePersistedSelectedModelRef: (...args: unknown[]) =>
state.resolvePersistedSelectedModelRefMock(...args),
}));
vi.mock("./model-selection.js", async () => {
const actual =
await vi.importActual<typeof import("./model-selection.js")>("./model-selection.js");
return {
normalizeStoredOverrideModel: actual.normalizeStoredOverrideModel,
resolveDefaultModelForAgent: (...args: unknown[]) =>
state.resolveDefaultModelForAgentMock(...args),
resolvePersistedSelectedModelRef: (...args: unknown[]) =>
state.resolvePersistedSelectedModelRefMock(...args),
};
});
vi.mock("../config/sessions/store.js", () => ({
loadSessionStore: (...args: unknown[]) => state.loadSessionStoreMock(...args),