feat(ai): Vercel AI SDK provider factory with BYOK and per-use-case model prefs
Provider factory (OpenAI/Anthropic/OpenRouter/Ollama). generateObject for all outputs. Features: recipe generation, photo-to-recipe vision, variations, drink/meal pairings, ingredient substitution, recipe adaptation, nutrition analysis, URL import, meal plan gen. User BYOK keys (AES-256-GCM). Per-use-case model preferences (text/vision/mealPlan). Site-level key override via admin settings.
This commit is contained in:
@@ -0,0 +1,142 @@
|
||||
import { NextRequest, NextResponse } from "next/server";
|
||||
import { z } from "zod";
|
||||
import { db, recipes, recipeIngredients, recipeSteps, mealPlans, mealPlanEntries, pantryItems, eq, and } from "@epicure/db";
|
||||
import { requireSession } from "@/lib/api-auth";
|
||||
import { applyRateLimit } from "@/lib/rate-limit";
|
||||
import { getDefaultProviderWithKey } from "@/lib/ai/resolve-user-key";
|
||||
import { generateMealPlan } from "@/lib/ai/features/generate-meal-plan";
|
||||
|
||||
const DAYS = ["mon", "tue", "wed", "thu", "fri", "sat", "sun"] as const;
|
||||
|
||||
const Schema = z.object({
|
||||
weekStart: z.string().regex(/^\d{4}-\d{2}-\d{2}$/),
|
||||
dietaryPrefs: z.string().max(200).optional(),
|
||||
servings: z.number().int().min(1).max(20).default(2),
|
||||
days: z.array(z.enum(DAYS)).min(1).max(7).default([...DAYS]),
|
||||
usePantry: z.boolean().default(false),
|
||||
pantryMode: z.boolean().default(false),
|
||||
});
|
||||
|
||||
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}`, 3, 60);
|
||||
if (limited) return limited;
|
||||
|
||||
const userId = session!.user.id;
|
||||
const locale = (session!.user as { locale?: string }).locale ?? "en";
|
||||
const config = await getDefaultProviderWithKey(userId);
|
||||
|
||||
// pantryMode forces usePantry on so pantry items are always fetched when maximizing pantry use
|
||||
const effectiveUsePantry = parsed.data.usePantry || parsed.data.pantryMode;
|
||||
|
||||
// Optionally fetch pantry items
|
||||
let pantryItemNames: string[] = [];
|
||||
if (effectiveUsePantry) {
|
||||
const pantry = await db
|
||||
.select({ rawName: pantryItems.rawName })
|
||||
.from(pantryItems)
|
||||
.where(eq(pantryItems.userId, userId));
|
||||
pantryItemNames = pantry.map((p) => p.rawName);
|
||||
}
|
||||
|
||||
const plan = await generateMealPlan(
|
||||
{
|
||||
dietaryPrefs: parsed.data.dietaryPrefs,
|
||||
servings: parsed.data.servings,
|
||||
pantryItems: pantryItemNames,
|
||||
days: parsed.data.days,
|
||||
pantryMode: parsed.data.pantryMode,
|
||||
},
|
||||
config,
|
||||
locale
|
||||
);
|
||||
|
||||
// Ensure meal plan row exists for the week
|
||||
let mealPlan = await db.query.mealPlans.findFirst({
|
||||
where: and(eq(mealPlans.userId, userId), eq(mealPlans.weekStart, parsed.data.weekStart)),
|
||||
});
|
||||
|
||||
if (!mealPlan) {
|
||||
const planId = crypto.randomUUID();
|
||||
await db.insert(mealPlans).values({ id: planId, userId, weekStart: parsed.data.weekStart });
|
||||
mealPlan = { id: planId, userId, weekStart: parsed.data.weekStart, createdAt: new Date() };
|
||||
}
|
||||
|
||||
const createdEntries: Array<{ id: string; day: string; mealType: string; recipeId: string; recipeTitle: string }> = [];
|
||||
|
||||
for (const entry of plan.entries) {
|
||||
// Create draft recipe
|
||||
const recipeId = crypto.randomUUID();
|
||||
await db.insert(recipes).values({
|
||||
id: recipeId,
|
||||
authorId: userId,
|
||||
title: entry.recipe.title,
|
||||
description: entry.recipe.description,
|
||||
baseServings: entry.servings,
|
||||
visibility: "private",
|
||||
aiGenerated: true,
|
||||
difficulty: entry.recipe.difficulty ?? null,
|
||||
prepMins: entry.recipe.prepMins ?? null,
|
||||
cookMins: entry.recipe.cookMins ?? null,
|
||||
});
|
||||
|
||||
if (entry.recipe.ingredients.length > 0) {
|
||||
await db.insert(recipeIngredients).values(
|
||||
entry.recipe.ingredients.map((ing, i) => ({
|
||||
id: crypto.randomUUID(),
|
||||
recipeId,
|
||||
rawName: ing.rawName,
|
||||
quantity: ing.quantity ?? null,
|
||||
unit: ing.unit ?? null,
|
||||
order: i,
|
||||
}))
|
||||
);
|
||||
}
|
||||
|
||||
if (entry.recipe.steps.length > 0) {
|
||||
await db.insert(recipeSteps).values(
|
||||
entry.recipe.steps.map((step, i) => ({
|
||||
id: crypto.randomUUID(),
|
||||
recipeId,
|
||||
instruction: step.instruction,
|
||||
order: i,
|
||||
}))
|
||||
);
|
||||
}
|
||||
|
||||
// Remove any existing entry for this day+mealType, then insert new
|
||||
const existingEntry = await db.query.mealPlanEntries.findFirst({
|
||||
where: and(
|
||||
eq(mealPlanEntries.mealPlanId, mealPlan!.id),
|
||||
eq(mealPlanEntries.day, entry.day),
|
||||
eq(mealPlanEntries.mealType, entry.mealType)
|
||||
),
|
||||
});
|
||||
|
||||
if (existingEntry) {
|
||||
await db.delete(mealPlanEntries).where(eq(mealPlanEntries.id, existingEntry.id));
|
||||
}
|
||||
|
||||
const entryId = crypto.randomUUID();
|
||||
await db.insert(mealPlanEntries).values({
|
||||
id: entryId,
|
||||
mealPlanId: mealPlan!.id,
|
||||
day: entry.day,
|
||||
mealType: entry.mealType,
|
||||
recipeId,
|
||||
servings: entry.servings,
|
||||
});
|
||||
|
||||
createdEntries.push({ id: entryId, day: entry.day, mealType: entry.mealType, recipeId, recipeTitle: entry.recipe.title });
|
||||
}
|
||||
|
||||
return NextResponse.json({ weekStart: parsed.data.weekStart, entries: createdEntries });
|
||||
}
|
||||
Reference in New Issue
Block a user