import { z } from 'zod';
import { AI_MODEL_STATUS } from '../../constants';
import { IconVariantsSchema } from '../icon-variants.schema';

export const SolvingEduTaskModelParamsSchema = z.object({
    imageAttachment: z.object({
        supported: z.boolean(),
        maxImages: z.number(),
        acceptedTypes: z.array(z.string()),
    }),
});

export type SolvingEduTaskModelParams = z.infer<typeof SolvingEduTaskModelParamsSchema>;

export const SolvingEduTaskModelSchema = z.object({
    uuid: z.string().uuid(),
    title: z.string(),
    description: z.string(),
    price: z.number(),
    order: z.number(),
    status: z.nativeEnum(AI_MODEL_STATUS),
    maxPromptLength: z.number(),
    params: SolvingEduTaskModelParamsSchema,
    icons: IconVariantsSchema,
});

export type SolvingEduTaskModel = z.infer<typeof SolvingEduTaskModelSchema>;