51 lines
1.9 KiB
JavaScript
51 lines
1.9 KiB
JavaScript
|
|
import { NextResponse } from "next/server";
|
||
|
|
import { getDisabledModels, disableModels, enableModels } from "@/lib/disabledModelsDb";
|
||
|
|
|
||
|
|
export const dynamic = "force-dynamic";
|
||
|
|
|
||
|
|
// GET /api/models/disabled?providerAlias=xxx
|
||
|
|
export async function GET(request) {
|
||
|
|
try {
|
||
|
|
const { searchParams } = new URL(request.url);
|
||
|
|
const providerAlias = searchParams.get("providerAlias");
|
||
|
|
const all = await getDisabledModels();
|
||
|
|
if (providerAlias) return NextResponse.json({ ids: all[providerAlias] || [] });
|
||
|
|
return NextResponse.json({ disabled: all });
|
||
|
|
} catch (error) {
|
||
|
|
console.log("Error fetching disabled models:", error);
|
||
|
|
return NextResponse.json({ error: "Failed to fetch disabled models" }, { status: 500 });
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// POST /api/models/disabled body: { providerAlias, ids: [...] }
|
||
|
|
export async function POST(request) {
|
||
|
|
try {
|
||
|
|
const { providerAlias, ids } = await request.json();
|
||
|
|
if (!providerAlias || !Array.isArray(ids)) {
|
||
|
|
return NextResponse.json({ error: "providerAlias and ids[] required" }, { status: 400 });
|
||
|
|
}
|
||
|
|
await disableModels(providerAlias, ids);
|
||
|
|
return NextResponse.json({ success: true });
|
||
|
|
} catch (error) {
|
||
|
|
console.log("Error disabling models:", error);
|
||
|
|
return NextResponse.json({ error: "Failed to disable models" }, { status: 500 });
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// DELETE /api/models/disabled?providerAlias=xxx[&id=yyy]
|
||
|
|
export async function DELETE(request) {
|
||
|
|
try {
|
||
|
|
const { searchParams } = new URL(request.url);
|
||
|
|
const providerAlias = searchParams.get("providerAlias");
|
||
|
|
const id = searchParams.get("id");
|
||
|
|
if (!providerAlias) {
|
||
|
|
return NextResponse.json({ error: "providerAlias required" }, { status: 400 });
|
||
|
|
}
|
||
|
|
await enableModels(providerAlias, id ? [id] : []);
|
||
|
|
return NextResponse.json({ success: true });
|
||
|
|
} catch (error) {
|
||
|
|
console.log("Error enabling models:", error);
|
||
|
|
return NextResponse.json({ error: "Failed to enable models" }, { status: 500 });
|
||
|
|
}
|
||
|
|
}
|