77
88
99class ProjectMetadata :
10- def __init__ (self ,
11- project_name : str = None ,
12- project_slug : str = None ,
13- description : str = "" ,
14- author_name : str = "" ,
15- version : str = "" ,
16- license : str = "" ,
17- year : int = datetime .now ().year ,
18- template : str = "none" ,
19- template_version : str = "0" ,
20- ):
10+ def __init__ (
11+ self ,
12+ project_name : str = None ,
13+ project_slug : str = None ,
14+ description : str = "" ,
15+ author_name : str = "" ,
16+ version : str = "" ,
17+ license : str = "" ,
18+ year : int = datetime .now ().year ,
19+ template : str = "none" ,
20+ template_version : str = "0" ,
21+ ):
2122 self .project_name = clean_input (project_name ) if project_name else "myagent"
22- self .project_slug = clean_input (project_slug ) if project_slug else self .project_name
23+ self .project_slug = (
24+ clean_input (project_slug ) if project_slug else self .project_name
25+ )
2326 self .description = description
2427 self .author_name = author_name
2528 self .version = version
@@ -33,16 +36,16 @@ def __init__(self,
3336
3437 def to_dict (self ):
3538 return {
36- ' project_name' : self .project_name ,
37- ' project_slug' : self .project_slug ,
38- ' description' : self .description ,
39- ' author_name' : self .author_name ,
40- ' version' : self .version ,
41- ' license' : self .license ,
42- ' year' : self .year ,
43- ' agentstack_version' : self .agentstack_version ,
44- ' template' : self .template ,
45- ' template_version' : self .template_version ,
39+ " project_name" : self .project_name ,
40+ " project_slug" : self .project_slug ,
41+ " description" : self .description ,
42+ " author_name" : self .author_name ,
43+ " version" : self .version ,
44+ " license" : self .license ,
45+ " year" : self .year ,
46+ " agentstack_version" : self .agentstack_version ,
47+ " template" : self .template ,
48+ " template_version" : self .template_version ,
4649 }
4750
4851 def to_json (self ):
@@ -62,46 +65,48 @@ def add_task(self, task):
6265
6366 def to_dict (self ):
6467 return {
65- ' agents' : self .agents ,
66- ' tasks' : self .tasks ,
68+ " agents" : self .agents ,
69+ " tasks" : self .tasks ,
6770 }
6871
6972 def to_json (self ):
7073 return json .dumps (self .to_dict (), default = str )
7174
7275
7376class FrameworkData :
74- def __init__ (self ,
75- # name: Optional[Literal["crewai"]] = None
76- name : str = None # TODO: better framework handling, Literal or Enum
77- ):
77+ def __init__ (
78+ self ,
79+ # name: Optional[Literal["crewai"]] = None
80+ name : str = None , # TODO: better framework handling, Literal or Enum
81+ ):
7882 self .name = name
7983
8084 def to_dict (self ):
8185 return {
82- ' name' : self .name ,
86+ " name" : self .name ,
8387 }
8488
8589 def to_json (self ):
8690 return json .dumps (self .to_dict (), default = str )
8791
8892
8993class CookiecutterData :
90- def __init__ (self ,
91- project_metadata : ProjectMetadata ,
92- structure : ProjectStructure ,
93- # framework: Literal["crewai"],
94- framework : str ,
95- ):
94+ def __init__ (
95+ self ,
96+ project_metadata : ProjectMetadata ,
97+ structure : ProjectStructure ,
98+ # framework: Literal["crewai"],
99+ framework : str ,
100+ ):
96101 self .project_metadata = project_metadata
97102 self .framework = framework
98103 self .structure = structure
99104
100105 def to_dict (self ):
101106 return {
102- ' project_metadata' : self .project_metadata .to_dict (),
103- ' framework' : self .framework ,
104- ' structure' : self .structure .to_dict (),
107+ " project_metadata" : self .project_metadata .to_dict (),
108+ " framework" : self .framework ,
109+ " structure" : self .structure .to_dict (),
105110 }
106111
107112 def to_json (self ):
0 commit comments