src/state.py

from typing import TypedDict, List, Dict, Any, Annotated
import operator

class AgentState(TypedDict):
    disruption_event: str
    impact_analysis: Dict[str, Any]
    alternative_routes: List[Dict[str, Any]]
    audit_passed: bool
    critique_feedback: str
    final_mitigation_plan: str
    messages: Annotated[List[str], operator.add]