| 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 |