524310433c
AI routes had no consistent error handling — a provider failure (e.g. OpenRouter returning a degraded-model error) crashed the route as an uncaught 500 and, worse, still charged the user's monthly aiCall quota for a request that never succeeded. Adds withAiQuota()/aiErrorResponse() (lib/ai/ai-error.ts) and applies them across all 12 AI generation routes: charges the quota, runs the AI call, refunds the credit and returns a clean user-facing message if it throws. Frontend needs no changes — existing dialogs already toast whatever `error` string comes back. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
94 lines
3.0 KiB
TypeScript
94 lines
3.0 KiB
TypeScript
import { NextRequest, NextResponse } from "next/server";
|
|
import { z } from "zod";
|
|
import { db, recipes, recipeIngredients, recipeSteps } from "@epicure/db";
|
|
import { requireSession } from "@/lib/api-auth";
|
|
import { applyRateLimit } from "@/lib/rate-limit";
|
|
import { withAiQuota } from "@/lib/ai/ai-error";
|
|
import { importFromPhoto } from "@/lib/ai/features/import-photo";
|
|
import { getModelConfigForUseCase } from "@/lib/ai/resolve-user-key";
|
|
|
|
const Schema = z.object({
|
|
imageBase64: z.string().max(14_000_000),
|
|
mimeType: z.enum(["image/jpeg", "image/png", "image/webp"]),
|
|
});
|
|
|
|
export async function POST(req: NextRequest) {
|
|
const { session, response } = await requireSession();
|
|
if (response) return response;
|
|
|
|
const body = await req.json() as unknown;
|
|
const parsed = Schema.safeParse(body);
|
|
if (!parsed.success) {
|
|
return NextResponse.json({ error: "Validation error", issues: parsed.error.issues }, { status: 400 });
|
|
}
|
|
|
|
const limited = await applyRateLimit(`rl:ai:${session!.user.id}`, 5, 60);
|
|
if (limited) return limited;
|
|
|
|
const userId = session!.user.id;
|
|
const locale = (session!.user as { locale?: string }).locale ?? "en";
|
|
|
|
const aiConfig = await getModelConfigForUseCase(userId, "vision");
|
|
|
|
// Fall back to vision-capable defaults if no explicit model configured
|
|
if (!aiConfig.model) {
|
|
if (aiConfig.provider === "openai") aiConfig.model = "gpt-4o";
|
|
else if (aiConfig.provider === "anthropic") aiConfig.model = "claude-sonnet-4-6";
|
|
}
|
|
|
|
const result = await withAiQuota(userId, session!.user.tier as "free" | "pro", () =>
|
|
importFromPhoto(parsed.data.imageBase64, parsed.data.mimeType, aiConfig, locale)
|
|
);
|
|
if (!result.ok) return result.response;
|
|
const recipe = result.data;
|
|
|
|
const newRecipeId = crypto.randomUUID();
|
|
const now = new Date();
|
|
|
|
await db.transaction(async (tx) => {
|
|
await tx.insert(recipes).values({
|
|
id: newRecipeId,
|
|
authorId: userId,
|
|
title: recipe.title,
|
|
description: recipe.description,
|
|
baseServings: recipe.baseServings ?? 4,
|
|
visibility: "private",
|
|
difficulty: recipe.difficulty,
|
|
prepMins: recipe.prepMins,
|
|
cookMins: recipe.cookMins,
|
|
dietaryTags: recipe.dietaryTags ?? {},
|
|
aiGenerated: true,
|
|
createdAt: now,
|
|
updatedAt: now,
|
|
});
|
|
|
|
if (recipe.ingredients.length > 0) {
|
|
await tx.insert(recipeIngredients).values(
|
|
recipe.ingredients.map((ing, i) => ({
|
|
id: crypto.randomUUID(),
|
|
recipeId: newRecipeId,
|
|
rawName: ing.rawName,
|
|
quantity: ing.quantity !== undefined ? String(ing.quantity) : undefined,
|
|
unit: ing.unit,
|
|
note: ing.note,
|
|
order: i,
|
|
}))
|
|
);
|
|
}
|
|
|
|
if (recipe.steps.length > 0) {
|
|
await tx.insert(recipeSteps).values(
|
|
recipe.steps.map((step, i) => ({
|
|
id: crypto.randomUUID(),
|
|
recipeId: newRecipeId,
|
|
instruction: step.instruction,
|
|
timerSeconds: step.timerSeconds,
|
|
order: i,
|
|
}))
|
|
);
|
|
}
|
|
});
|
|
|
|
return NextResponse.json({ id: newRecipeId });
|
|
}
|