data/models.py

from dataclasses import dataclass
from typing import List, Optional


@dataclass
class Session:
    """A chat session in MSC"""
    session_id: int
    dialog_text: str              # Formatted dialog text
    persona_summary: str          # Summary from sessionlevel_summaries_subsample5.json
    persona_sentences: List[str]  # Persona sentences extracted from summaries_subsample5.json
    followup: Optional[str] = None  # Follow-up question (from msc_personasummary)
    newfact: Optional[str] = None   # New information (from msc_personasummary)
    init_personas: Optional[List[List[str]]] = None  # Initial personas



@dataclass
class Conversation:
    """A complete conversation consisting of multiple sessions"""
    conv_id: str                  # initial_data_id
    sessions: List[Session]       # List of sessions (usually 5)
    init_personas: List[List[str]]  # Initial personas from init_persona_all.json