src/web/schemas.py

from __future__ import annotations

from enum import Enum
from typing import Any

from pydantic import BaseModel, Field


class ModoEvaluacion(str, Enum):
    POR_DEFECTO = "por_defecto"
    PERSONALIZADA = "personalizada"


class TipoEvaluacion(str, Enum):
    PREGUNTAS_AGENTE = "preguntas_agente"
    PREGUNTAS_CERRADAS_ESPERADAS = "preguntas_cerradas_esperadas"


class JobRequest(BaseModel):
    modo_evaluacion: ModoEvaluacion
    tipo_evaluacion: TipoEvaluacion = TipoEvaluacion.PREGUNTAS_CERRADAS_ESPERADAS
    modelo_hf: str = Field(..., min_length=3, max_length=256)
    timeout_segundos: int = Field(default=90, ge=10, le=300)
    numero_reintentos: int = Field(default=1, ge=1, le=5)
    max_prompts_por_job: int | None = Field(default=None, ge=1)
    plantilla_personalizada: dict[str, Any] | None = None