mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-06 07:20:43 +00:00
test: reuse model override normalizer
This commit is contained in:
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user