131 lines
4.1 KiB
TypeScript
131 lines
4.1 KiB
TypeScript
import { and, eq } from "drizzle-orm";
|
|
import { NextResponse } from "next/server";
|
|
import { BANANAIMAGE_MODEL_PRESETS } from "@/lib/ai/bananaimage-models";
|
|
import { db } from "@/lib/db";
|
|
import { getAiProviderById } from "@/lib/db/queries";
|
|
import { type AiModel, aiModel } from "@/lib/db/schema";
|
|
import { requireAdmin } from "@/lib/middleware/admin";
|
|
|
|
export async function GET() {
|
|
try {
|
|
await requireAdmin();
|
|
return NextResponse.json({
|
|
presets: BANANAIMAGE_MODEL_PRESETS,
|
|
counts: { total: BANANAIMAGE_MODEL_PRESETS.length },
|
|
});
|
|
} catch (error: any) {
|
|
console.error("[BananaImage Models API] GET error:", error);
|
|
return NextResponse.json(
|
|
{ error: error.message || "Failed to fetch BananaImage model presets" },
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|
|
|
|
export async function POST(request: Request) {
|
|
try {
|
|
await requireAdmin();
|
|
const body = await request.json();
|
|
const providerId = String(body?.providerId || "").trim();
|
|
const modelIds = Array.isArray(body?.modelIds)
|
|
? body.modelIds
|
|
.map((item: unknown) => String(item || "").trim())
|
|
.filter(Boolean)
|
|
: null;
|
|
|
|
if (!providerId) {
|
|
return NextResponse.json({ error: "Provider ID is required" }, { status: 400 });
|
|
}
|
|
|
|
const provider = await getAiProviderById({ id: providerId });
|
|
if (!provider || provider.type !== "bananaimage") {
|
|
return NextResponse.json(
|
|
{ error: "Provider must be a BananaImage provider" },
|
|
{ status: 400 }
|
|
);
|
|
}
|
|
|
|
const presets = modelIds
|
|
? BANANAIMAGE_MODEL_PRESETS.filter((preset) => modelIds.includes(preset.modelId))
|
|
: BANANAIMAGE_MODEL_PRESETS;
|
|
const addedModels: AiModel[] = [];
|
|
const updatedModels: AiModel[] = [];
|
|
const errors: { modelId: string; error: string }[] = [];
|
|
|
|
for (const preset of presets) {
|
|
try {
|
|
const existing = await db.query.aiModel.findFirst({
|
|
where: and(
|
|
eq(aiModel.providerId, providerId),
|
|
eq(aiModel.modelId, preset.modelId)
|
|
),
|
|
});
|
|
const values = {
|
|
displayName: preset.displayName,
|
|
description: preset.description,
|
|
supportsVision: preset.supportsVision,
|
|
supportsImageGen: preset.supportsImageGen,
|
|
isEnabled: true,
|
|
isHidden: false,
|
|
sortOrder: preset.sortOrder,
|
|
capabilities: preset.capabilities || null,
|
|
defaultParams: preset.defaultParams || null,
|
|
imageGenConfig: preset.imageGenConfig || null,
|
|
};
|
|
|
|
if (existing) {
|
|
const [updated] = await db
|
|
.update(aiModel)
|
|
.set(values)
|
|
.where(
|
|
and(
|
|
eq(aiModel.providerId, providerId),
|
|
eq(aiModel.modelId, preset.modelId)
|
|
)
|
|
)
|
|
.returning();
|
|
if (updated) updatedModels.push(updated);
|
|
continue;
|
|
}
|
|
|
|
const [created] = await db
|
|
.insert(aiModel)
|
|
.values({
|
|
providerId,
|
|
modelId: preset.modelId,
|
|
...values,
|
|
isEnabled: true,
|
|
isHidden: false,
|
|
isDefault: false,
|
|
isSubModel: false,
|
|
parentModelId: null,
|
|
})
|
|
.returning();
|
|
if (created) addedModels.push(created);
|
|
} catch (error: any) {
|
|
console.error(`[BananaImage Models API] Failed to sync ${preset.modelId}:`, error);
|
|
errors.push({
|
|
modelId: preset.modelId,
|
|
error: error.message || "Sync failed",
|
|
});
|
|
}
|
|
}
|
|
|
|
return NextResponse.json({
|
|
success: true,
|
|
addedCount: addedModels.length,
|
|
updatedCount: updatedModels.length,
|
|
addedModels,
|
|
updatedModels,
|
|
errors: errors.length > 0 ? errors : undefined,
|
|
counts: { requested: presets.length, total: BANANAIMAGE_MODEL_PRESETS.length },
|
|
});
|
|
} catch (error: any) {
|
|
console.error("[BananaImage Models API] POST error:", error);
|
|
return NextResponse.json(
|
|
{ error: error.message || "Failed to sync BananaImage models" },
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|