from typing import TypedDict, List
from enum import Enum

class DiagramType(Enum):
    CLASS = 1
    USE_CASE = 2
    SEQUENCE = 3
    OBJECT = 4
    ACTIVITY = 5
    COMPONENT = 6
    DEPLOYMENT = 7
    STATE = 8
    TIMING = 9

class AgentState(TypedDict):
    project_name: str
    project_description: str
    output_dir: str
    selected_diagrams: List[int]
    entities_classes: str
    actors_use_cases: str
    sequence_interactions: str
    class_diagram: str
    use_case_diagram: str
    sequence_diagram: str
    activity_diagram: str
    component_diagram: str
    deployment_diagram: str
    state_diagram: str
    timing_diagram: str
    object_diagram: str