from typing import List
from pydantic import BaseModel, Field
from .learning_objectives import LearningObjective
from .questions import RankedMultipleChoiceQuestion


class Assessment(BaseModel):
    """Model for an assessment."""
    learning_objectives: List[LearningObjective] = Field(description="List of learning objectives")
    questions: List[RankedMultipleChoiceQuestion] = Field(description="List of ranked questions")