fix: update the logic for add model from admin side

This commit is contained in:
2026-07-21 17:17:14 +07:00
parent a1706fb40f
commit 06d6873d52
5 changed files with 61 additions and 14 deletions
+6 -5
View File
@@ -4,7 +4,7 @@ import {
addCustomModel,
deleteCustomModel,
} from "@/models";
import { isDeletedModel } from "@/lib/db";
import { restoreDeletedModel } from "@/lib/db";
import { requireAdminUser } from "@/lib/auth/currentUser";
export const dynamic = "force-dynamic";
@@ -45,11 +45,12 @@ export async function POST(request) {
return NextResponse.json({ error: "providerAlias and id required" }, { status: 400 });
}
await requireCustomModelCatalogAdmin();
if (await isDeletedModel(providerAlias, id)) {
return NextResponse.json({ error: "This model was permanently deleted" }, { status: 409 });
}
// Adding a model is an explicit administrator restore action. Older
// tombstones otherwise reject the request and make the Test action return
// the misleading "deleted by an administrator" response.
const added = await addCustomModel({ providerAlias, id, type: type || "llm", name });
return NextResponse.json({ success: true, added });
const restored = await restoreDeletedModel(providerAlias, id);
return NextResponse.json({ success: true, added, restored });
} catch (error) {
const accessError = getAccessErrorResponse(error);
if (accessError) return accessError;
+1 -1
View File
@@ -81,7 +81,7 @@ export {
// Permanently deleted models
export {
getDeletedModels, isDeletedModel, isDeletedModelReference, deleteModelPermanently,
getDeletedModels, isDeletedModel, isDeletedModelReference, restoreDeletedModel, deleteModelPermanently,
} from "./repos/deletedModelsRepo.js";
// Usage
+33
View File
@@ -302,6 +302,39 @@ export async function isDeletedModelReference(reference) {
});
}
/**
* Remove permanent-deletion tombstones for a model that an administrator adds
* back to the shared catalog. This deliberately does not restore dependencies
* that were removed when the model was deleted (combos, aliases, pricing, and
* request details); it only makes the newly added model routable again.
*/
export async function restoreDeletedModel(providerAlias, modelId) {
if (!providerAlias || !modelId) return false;
const db = await getAdapter();
let restored = false;
db.transaction(() => {
const providerAliases = getProviderAliasesSync(db, providerAlias);
for (const alias of providerAliases) {
const row = db.get(`SELECT value FROM kv WHERE scope = ? AND key = ?`, [SCOPE, alias]);
const current = normalizeIds(row ? parseJson(row.value, []) : []);
const next = current.filter((deletedModelId) => !matchesDeletedModelId(modelId, deletedModelId));
if (next.length === current.length) continue;
restored = true;
if (next.length === 0) {
db.run(`DELETE FROM kv WHERE scope = ? AND key = ?`, [SCOPE, alias]);
} else {
db.run(`UPDATE kv SET value = ? WHERE scope = ? AND key = ?`, [stringifyJson(next), SCOPE, alias]);
}
}
});
return restored;
}
export async function deleteModelPermanently(providerAlias, modelId) {
if (!providerAlias || !modelId) return null;
+16 -8
View File
@@ -3,7 +3,7 @@ import { beforeEach, describe, expect, it, vi } from "vitest";
const getCustomModels = vi.fn();
const addCustomModel = vi.fn();
const deleteCustomModel = vi.fn();
const isDeletedModel = vi.fn();
const restoreDeletedModel = vi.fn();
const requireAdminUser = vi.fn();
vi.mock("@/models", () => ({
@@ -11,7 +11,7 @@ vi.mock("@/models", () => ({
addCustomModel,
deleteCustomModel,
}));
vi.mock("@/lib/db", () => ({ isDeletedModel }));
vi.mock("@/lib/db", () => ({ restoreDeletedModel }));
vi.mock("@/lib/auth/currentUser", () => ({ requireAdminUser }));
const { GET, POST, DELETE } = await import("../../src/app/api/models/custom/route.js");
@@ -21,9 +21,9 @@ describe("/api/models/custom", () => {
getCustomModels.mockReset();
addCustomModel.mockReset();
deleteCustomModel.mockReset();
isDeletedModel.mockReset();
restoreDeletedModel.mockReset();
requireAdminUser.mockReset();
isDeletedModel.mockResolvedValue(false);
restoreDeletedModel.mockResolvedValue(false);
});
it("keeps the shared catalog readable to authenticated model selectors", async () => {
@@ -68,17 +68,25 @@ describe("/api/models/custom", () => {
});
});
it("does not let an administrator re-add a permanently deleted model", async () => {
it("restores a permanently deleted model when an administrator adds it again", async () => {
requireAdminUser.mockResolvedValue({ id: "admin", role: "admin" });
isDeletedModel.mockResolvedValue(true);
restoreDeletedModel.mockResolvedValue(true);
addCustomModel.mockResolvedValue(true);
const response = await POST(new Request("http://localhost/api/models/custom", {
method: "POST",
body: JSON.stringify({ providerAlias: "openai", id: "gpt-deleted", type: "llm" }),
}));
expect(response.status).toBe(409);
expect(addCustomModel).not.toHaveBeenCalled();
expect(response.status).toBe(200);
await expect(response.json()).resolves.toMatchObject({ success: true, added: true, restored: true });
expect(restoreDeletedModel).toHaveBeenCalledWith("openai", "gpt-deleted");
expect(addCustomModel).toHaveBeenCalledWith({
providerAlias: "openai",
id: "gpt-deleted",
type: "llm",
name: undefined,
});
});
it("rejects a non-admin deleting a shared custom model", async () => {
@@ -218,5 +218,10 @@ describe("permanent model deletion", () => {
const backup = await db.exportDb();
expect(backup.deletedModels).toMatchObject({ [providerId]: [modelId] });
expect(await db.restoreDeletedModel(providerPrefix, `${modelId}(high)`)).toBe(true);
expect(await db.isDeletedModel(providerId, modelId)).toBe(false);
expect(await db.isDeletedModel(providerPrefix, `${modelId}(high)`)).toBe(false);
expect((await db.getDeletedModels())[providerId]).toBeUndefined();
});
});