scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(ui): cover auto-router tier model replacement

Browse source
This commit is contained in:
Onat Özmen 2026-08-30 16:40:36 +03:00
parent 5e4b3838aa
commit 40f75b1dd7
1 changed files with 42 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
ui/litellm-dashboard/src/components/edit_auto_router/auto_router_tier_model_replacement.test.ts Normal file
View file

@ -0,0 +1,42 @@
import { describe, expect, it } from "vitest";
import { applyTierSetAction } from "../add_model/tier_set_actions";
import { buildUpdatedComplexityRouterConfig, hydrateComplexityRouterConfig } from "./edit_auto_router_modal";
describe("auto-router tier model replacement", () => {
it("writes the replacement model and drops params tied to the removed model", () => {
const stored = {
tiers: {
SIMPLE: ["gemma4:12b-it-qat"],
MEDIUM: [],
COMPLEX: [],
REASONING: [],
},
tier_model_configs: {
SIMPLE: [
{
model_name: "gemma4:12b-it-qat",
litellm_params: { reasoning_effort: "medium" },
},
],
},
classifier_type: "heuristic" as const,
};
const hydrated = hydrateComplexityRouterConfig(stored, "gemma4:12b-it-qat");
const changed = applyTierSetAction(hydrated, [], {
kind: "models",
id: "SIMPLE",
models: ["gemma4-12b-it-optiq-4bit"],
}).value;
const saved = buildUpdatedComplexityRouterConfig(stored, changed);
expect(saved.tiers).toEqual({
SIMPLE: ["gemma4-12b-it-optiq-4bit"],
MEDIUM: [],
COMPLEX: [],
REASONING: [],
});
expect(saved).not.toHaveProperty("tier_model_configs");
});
});