File size: 672 Bytes
5a2d62e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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