import { z } from 'zod';

/**
 * This Zod schema defines the *only* structured output
 * we want the AI to be able to generate. We will pass
 * this to the Vercel AI SDK to guarantee the AI's output
 * conforms to this shape.
 */
export const NodeSuggestionSchema = z.object({
  action: z.literal('suggest_node'),
  title: z
    .string()
    .describe('A concise, descriptive title for the thought node.'),
  body: z
    .string()
    .describe('The full, well-structured content of the thought node.'),
});

export type NodeSuggestion = z.infer<typeof NodeSuggestionSchema>;