from enum import Enum
from typing import Dict, Optional
from pydantic import BaseModel


class EvaluationType(str, Enum):
    WRITING = "writing"
    SPEAKING_INTERACTIVE = "speaking_interactive"
    SPEAKING = "speaking"

class EvaluationRecord(BaseModel):
    id: str
    session_id: str
    exercise_id: str
    type: EvaluationType
    task: int
    status: str = "pending"
    result: Optional[Dict] = None