import { z } from 'zod';

export namespace WriterGenerateTextCommand {
    export const RequestParamsSchema = z.object({
        uuid: z.string().uuid(),
    });
    export type RequestParams = z.infer<typeof RequestParamsSchema>;

    export const RequestBodySchema = z.object({
        prompt: z.string(),
        selectionText: z.string(),
        contextBefore: z.string().max(2000).optional(),
        contextAfter: z.string().max(2000).optional(),
    });
    export type RequestBody = z.infer<typeof RequestBodySchema>;

    export const ResponseSchema = z.object({
        data: z.object({
            output: z.string(),
        }),
    });
    export type Response = z.infer<typeof ResponseSchema>;
}