AdityParbat/disaster-response-env

1from typing import List, Optional, Dict, Any2from pydantic import BaseModel, Field3 4class Incident(BaseModel):5    id: str6    type: str7    severity: str8    requires: List[str]9    location: str10    notes: Optional[str] = None11    time_to_resolve: int = 112    turns_worked: int = 013    assigned_units: List[str] = []14 15class Observation(BaseModel):16    step: int17    max_steps: int18    active_incidents: List[Incident]19    available_units: List[str]20    busy_units: Dict[str, int]21    resources_manifest: Dict[str, str]22    constraints: List[str]23    previous_actions: List[Any] = []24 25class Dispatch(BaseModel):26    unit: str27    incident_id: str28 29class Action(BaseModel):30    dispatches: List[Dispatch] = []31    recalls: List[Dispatch] = []32    reasoning: str = ""33    priority_order: Optional[List[str]] = None34    reserved_units: Optional[List[str]] = None35 36class StepResult(BaseModel):37    observation: Observation38    reward: float39    terminated: bool40    truncated: bool41    info: Dict[str, Any] = {}