Custom Structured Output Schemas — Type-Safe AI Agent Responses

import { z } from "zod";

export const ReviewExtractionSchema = z.object({
  product_name: z.string().describe("Name of the reviewed product"),
  rating: z.number().min(1).max(5).describe("Star rating (1-5)"),
  sentiment: z.enum(["positive", "negative", "mixed", "neutral"]),
  key_pros: z.array(z.string()).max(5)
    .describe("Top positive points mentioned"),
  key_cons: z.array(z.string()).max(5)
    .describe("Top negative points mentioned"),
  summary: z.string().max(200)
    .describe("One-sentence summary of the review"),
  recommended: z.boolean()
    .describe("Whether the reviewer recommends the product"),
  confidence: z.number().min(0).max(1)
    .describe("Extraction confidence score"),
});

export type ReviewExtraction = z.infer<typeof ReviewExtractionSchema>;