@@ -139,14 +139,13 @@ def convert_hierarchical_team_to_dict(
139
139
return teams
140
140
141
141
142
- def convert_sequential_team_to_dict (team : Team ) -> Mapping [str , GraphMember ]:
142
+ def convert_sequential_team_to_dict (members : list [ Member ] ) -> Mapping [str , GraphMember ]:
143
143
team_dict : dict [str , GraphMember ] = {}
144
144
145
145
in_counts : defaultdict [int , int ] = defaultdict (int )
146
146
out_counts : defaultdict [int , list [int ]] = defaultdict (list [int ])
147
147
members_lookup : dict [int , Member ] = {}
148
-
149
- for member in team .members :
148
+ for member in members :
150
149
assert member .id is not None , "member.id is unexpectedly None"
151
150
if member .source :
152
151
in_counts [member .id ] += 1
@@ -171,7 +170,7 @@ def convert_sequential_team_to_dict(team: Team) -> Mapping[str, GraphMember]:
171
170
managed = skill .managed ,
172
171
definition = skill .tool_definition ,
173
172
)
174
- for skill in member .skills
173
+ for skill in memberModel .skills
175
174
]
176
175
tools += [
177
176
GraphUpload (
@@ -180,7 +179,7 @@ def convert_sequential_team_to_dict(team: Team) -> Mapping[str, GraphMember]:
180
179
owner_id = upload .owner_id ,
181
180
upload_id = cast (int , upload .id ),
182
181
)
183
- for upload in member .uploads
182
+ for upload in memberModel .uploads
184
183
if upload .owner_id is not None
185
184
]
186
185
graph_member = GraphMember (
@@ -453,7 +452,6 @@ async def generator(
453
452
454
453
try :
455
454
memory = await AsyncPostgresSaver .from_conn_string (settings .PG_DATABASE_URI )
456
-
457
455
if team .workflow == "hierarchical" :
458
456
teams = convert_hierarchical_team_to_dict (team , members )
459
457
team_leader = list (teams .keys ())[0 ]
@@ -466,7 +464,7 @@ async def generator(
466
464
"main_task" : formatted_messages ,
467
465
}
468
466
else :
469
- member_dict = convert_sequential_team_to_dict (team )
467
+ member_dict = convert_sequential_team_to_dict (members )
470
468
root = create_sequential_graph (member_dict , memory )
471
469
first_member = list (member_dict .values ())[0 ]
472
470
state = {
0 commit comments