scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

address greptile review feedback (greploop iteration 1)

Browse source 
Add cleanup helper to delete models created during tests, preventing
stale data accumulation across repeated test runs.
This commit is contained in:
Yuneng Jiang 2026-04-11 20:21:37 -07:00
parent 4f364a8138
commit 5e07c1cbc9
No known key found for this signature in database
1 changed files with 34 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
ui/litellm-dashboard/e2e_tests/tests/modelsPage/addModel.spec.ts
View file

@ -4,6 +4,34 @@ import { Role, users } from "../../fixtures/users";
import { navigateToPage } from "../../helpers/navigation";
import { Page } from "../../fixtures/pages";
/**
* Helper to delete a model by searching for it via the API and deleting matching entries.
* Accepts a partial model name to match against.
*/
async function cleanupModels(request: any, searchTerm: string) {
try {
const response = await request.get("/v2/model/info?include_team_models=true&page=1&size=100", {
headers: { Authorization: "Bearer sk-1234" },
});
const data = await response.json();
const models = data?.data || [];
for (const model of models) {
const name = model.model_name || "";
if (name.includes(searchTerm)) {
await request.post("/model/delete", {
headers: {
Authorization: "Bearer sk-1234",
"Content-Type": "application/json",
},
data: { id: model.model_info?.id },
});
}
}
} catch {
// Best-effort cleanup; don't fail the test
}
}
test.describe("Add Model", () => {
test.use({ storageState: ADMIN_STORAGE_PATH });
@ -110,7 +138,9 @@ test.describe("Add Model", () => {
await expect(page.getByTestId("connection-failure-msg")).toContainText("failed");
});
test("Add specific model and verify it appears in All Models", async ({ page }) => {
test("Add specific model and verify it appears in All Models", async ({ page, request }) => {
// Clean up any leftover models from previous runs
await cleanupModels(request, "claude-haiku-4-5");
await navigateToPage(page, Page.Models);
await page.getByRole("tab", { name: "Add Model" }).click();
@ -155,7 +185,9 @@ test.describe("Add Model", () => {
await expect(tableBody.getByText("claude-haiku-4-5").first()).toBeVisible({ timeout: 15_000 });
});
test("Add wildcard route and verify it appears in All Models", async ({ page }) => {
test("Add wildcard route and verify it appears in All Models", async ({ page, request }) => {
// Clean up any leftover models from previous runs
await cleanupModels(request, "cohere/");
await navigateToPage(page, Page.Models);
await page.getByRole("tab", { name: "Add Model" }).click();