models.py

from __future__ import annotations
from dataclasses import dataclass
from typing import Dict, Any

@dataclass
class Vitals:
    sbp: int
    dbp: int
    hr: int
    rr: int
    temp_c: float
    spo2: int

@dataclass
class PatientState:
    scenario: str
    patient_type: str
    notes: str
    labs: Dict[str, Any]
    vitals: Vitals