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


class Intention(str, Enum):
    LEARN = "learn"
    REVIEW = "review"
    DRAFT = "draft"


class Session(BaseModel):
    project_id: str
    topic: str
    intention: Intention
    active_node_id: Optional[str] = None