from typing import List, Optional
from pydantic import BaseModel


class PartModel(BaseModel):
    page_url: Optional[str] = None
    title: Optional[str] = None
    content: Optional[str] = None

    img_url: Optional[List[str]] = None
    video_url: Optional[List[str]] = None
    audio_url: Optional[List[str]] = None
    pdf_url: Optional[List[str]] = None

    agent_recommendation_rank: Optional[float] = None
    agent_recommendation_notes: Optional[str] = None


class UnitSubtopicOutputModel(BaseModel):
    unit_title: Optional[str] = None
    subtopic_title: Optional[str] = None
    query: Optional[str] = None

    parts: Optional[List[PartModel]] = None