vaishakgkumar commited on
Commit
2bf02f0
1 Parent(s): d94e2c3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -8
main.py CHANGED
@@ -164,16 +164,16 @@ def process_query(oai_key, query, max_auto_reply):
164
  boss_aid.initiate_chat(manager, homework_problem=HOMEWORK_PROBLEM, n_results=3)
165
 
166
  def consultingteam():
167
- _reset_agents()
168
  team = autogen.GroupChat(
169
- agents=[boss_aid, consultant, pm, reviewer, consulting_pro],
170
- messages=[],
171
- max_round=12,
172
- speaker_selection_method="round_robin"
173
- )
174
 
175
- manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
176
- boss_aid.initiate_chat(manager, consulting_problem=CONSULTING_PROBLEM, n_results=3)
177
 
178
  # Select and initiate team based on team mapping
179
  team_function = {
 
164
  boss_aid.initiate_chat(manager, homework_problem=HOMEWORK_PROBLEM, n_results=3)
165
 
166
  def consultingteam():
167
+ reset_agents()
168
  team = autogen.GroupChat(
169
+ agents=[boss_aid, consultant(), pm(), reviewer(), consulting_pro()],
170
+ messages=[],
171
+ max_round=12,
172
+ speaker_selection_method="round_robin"
173
+ )
174
 
175
+ manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
176
+ boss_aid.initiate_chat(manager, consulting_problem=CONSULTING_PROBLEM, n_results=3)
177
 
178
  # Select and initiate team based on team mapping
179
  team_function = {