Arnaud
55 lines
2.4 KiB
TypeScript
55 lines
2.4 KiB
TypeScript
import { generateObject } from "ai";
|
|
import { z } from "zod";
|
|
import { resolveModel, type AiConfig } from "../factory";
|
|
|
|
const RecipeOutputSchema = z.object({
|
|
title: z.string(),
|
|
description: z.string(),
|
|
baseServings: z.number().int().min(1).max(100),
|
|
prepMins: z.number().int().min(0).optional(),
|
|
cookMins: z.number().int().min(0).optional(),
|
|
difficulty: z.enum(["easy", "medium", "hard"]),
|
|
dietaryTags: z.object({
|
|
vegan: z.boolean().optional(),
|
|
vegetarian: z.boolean().optional(),
|
|
glutenFree: z.boolean().optional(),
|
|
dairyFree: z.boolean().optional(),
|
|
nutFree: z.boolean().optional(),
|
|
halal: z.boolean().optional(),
|
|
kosher: z.boolean().optional(),
|
|
}),
|
|
ingredients: z.array(z.object({
|
|
rawName: z.string(),
|
|
quantity: z.number().optional(),
|
|
unit: z.string().optional(),
|
|
note: z.string().optional(),
|
|
})),
|
|
steps: z.array(z.object({
|
|
instruction: z.string(),
|
|
timerSeconds: z.number().int().optional(),
|
|
})),
|
|
});
|
|
|
|
export type GeneratedRecipe = z.infer<typeof RecipeOutputSchema>;
|
|
|
|
export async function generateRecipe(
|
|
prompt: string,
|
|
config?: AiConfig & { language?: string; difficulty?: "easy" | "medium" | "hard"; userContext?: string }
|
|
): Promise<GeneratedRecipe> {
|
|
const model = resolveModel(config);
|
|
const lang = config?.language ?? "en";
|
|
const langInstruction = lang !== "en" ? ` Write the entire recipe (title, description, ingredient names, step instructions) in ${lang}.` : "";
|
|
const difficultyInstruction = config?.difficulty ? ` The recipe must be ${config.difficulty} difficulty: ${{ easy: "simple techniques, minimal steps, common ingredients", medium: "moderate skill required, standard techniques", hard: "advanced techniques, multiple components, skilled cook required" }[config.difficulty]}.` : "";
|
|
const userInstruction = config?.userContext ? `\n\nUser preferences and context:\n${config.userContext}` : "";
|
|
|
|
const { object } = await generateObject({
|
|
model,
|
|
schema: RecipeOutputSchema,
|
|
system:
|
|
`You are a professional chef and culinary writer. Generate detailed, accurate, and delicious recipes. For ingredients: quantity must be a number only (e.g. 0.25, 1.5, 2), unit is a separate string (e.g. 'cup', 'tbsp', 'g', 'ml'). Never combine quantity and unit into one string. Include helpful notes for ingredients when relevant.${langInstruction}${difficultyInstruction}${userInstruction}`,
|
|
prompt: `Create a recipe for: ${prompt}`,
|
|
});
|
|
|
|
return object;
|
|
}
|