42 lines
1.1 KiB
TypeScript
42 lines
1.1 KiB
TypeScript
import { z } from "zod";
|
|
import { requireAdmin } from "@/lib/middleware/admin";
|
|
import { ChatSDKError } from "@/lib/errors";
|
|
import { probeRunwareSeedImageSupport } from "@/lib/realtime/runware-capability";
|
|
|
|
const schema = z.object({
|
|
apiKey: z.string().min(1),
|
|
modelId: z.string().min(1),
|
|
});
|
|
|
|
function mapStatus(code: string) {
|
|
if (code === "UPSTREAM_ERROR") return 502;
|
|
return 400;
|
|
}
|
|
|
|
export async function POST(request: Request) {
|
|
try {
|
|
await requireAdmin();
|
|
|
|
const body = await request.json();
|
|
const data = schema.parse(body);
|
|
const result = await probeRunwareSeedImageSupport({
|
|
apiKey: data.apiKey,
|
|
modelId: data.modelId,
|
|
});
|
|
|
|
if (!result.supported) {
|
|
return Response.json(result, { status: mapStatus(result.code) });
|
|
}
|
|
|
|
return Response.json(result);
|
|
} catch (error) {
|
|
if (error instanceof ChatSDKError) {
|
|
return error.toResponse();
|
|
}
|
|
if (error instanceof z.ZodError) {
|
|
return new ChatSDKError("bad_request:api", error.message).toResponse();
|
|
}
|
|
return new ChatSDKError("offline:admin").toResponse();
|
|
}
|
|
}
|