from pydantic import BaseModel
from typing import Optional, List


class ChecklistItem(BaseModel):
    category: str
    item: str
    status: str  # "confirmed" | "needs_clarification" | "not_discussed"
    notes: Optional[str] = None


class ChecklistResponse(BaseModel):
    session_id: str
    checklist: List[ChecklistItem]
    markdown: str