app/schemas/schemas.py

from enum import Enum
from pydantic import BaseModel, Field

class SOLOLevel(str,Enum):
    """
    Enum for SOLO levels.
    """
    UNISTRUCTURAL = "unistructural"
    MULTISTRUCTURAL = "multistructural"


class MCQResponse(BaseModel):
    """
    Response model for MCQ generation.
    """
    question_text: str = Field(..., description="The generated question text.")
    options: list[str] = Field(..., description="List of answer options.")
    correct_answer: str = Field(..., description="The correct answer from the options.")