Arnaud
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>
44 lines
1.7 KiB
TypeScript
44 lines
1.7 KiB
TypeScript
import { NextRequest, NextResponse } from "next/server";
|
|
import { z } from "zod";
|
|
import { requireSession } from "@/lib/api-auth";
|
|
import { withAiQuota } from "@/lib/ai/ai-error";
|
|
import { applyRateLimit } from "@/lib/rate-limit";
|
|
import { getDefaultProviderWithKey } from "@/lib/ai/resolve-user-key";
|
|
import { substituteIngredient } from "@/lib/ai/features/substitute-ingredient";
|
|
|
|
const Schema = z.object({
|
|
ingredient: z.string().min(1).max(200),
|
|
recipeTitle: z.string().max(200).optional(),
|
|
provider: z.enum(["openai", "anthropic", "openrouter", "ollama"]).optional(),
|
|
model: z.string().optional(),
|
|
});
|
|
|
|
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" }, { status: 400 });
|
|
|
|
const rateLimitRes = await applyRateLimit(`ai:substitute:${session!.user.id}`, 10, 60);
|
|
if (rateLimitRes) return rateLimitRes;
|
|
|
|
const context = parsed.data.recipeTitle
|
|
? `recipe "${parsed.data.recipeTitle}"`
|
|
: "a general recipe";
|
|
|
|
const aiConfig = parsed.data.provider
|
|
? { provider: parsed.data.provider, model: parsed.data.model }
|
|
: await getDefaultProviderWithKey(session!.user.id);
|
|
|
|
const locale = (session!.user as { locale?: string }).locale ?? "en";
|
|
|
|
const result = await withAiQuota(session!.user.id, session!.user.tier as "free" | "pro", () =>
|
|
substituteIngredient(parsed.data.ingredient, context, aiConfig, locale)
|
|
);
|
|
if (!result.ok) return result.response;
|
|
|
|
return NextResponse.json({ substitutions: result.data });
|
|
}
|