core/models.py

from dataclasses import dataclass
from typing import Dict, List, Optional


@dataclass
class Section:
    section_id: str
    title: str
    parent_id: Optional[str]


@dataclass
class PageMeta:
    page_id: str
    guideline_id: str
    section_id: str
    title: str
    summary: str
    intent_ids: List[str]


@dataclass
class Passage:
    passage_id: str
    guideline_id: str
    page_id: str
    section_id: str
    order_in_section: int
    text: str
    source_page: int
    source_lines: List[int]
    tags: List[str]


@dataclass
class IntentMeta:
    intent_id: str
    name: str
    description: str
    topic_group: str
    guideline_ids: List[str]
    primary_section_ids: List[str]
    example_questions: List[str]


@dataclass
class ClarifyingChoice:
    id: str
    text: str
    update_state: Dict[str, object]


@dataclass
class ClarifyingQuestion:
    question_id: str
    applies_to_topic_groups: List[str]
    text: str
    choices: List[ClarifyingChoice]