Vaishak G Kumar commited on
Commit
ead5031
·
verified ·
1 Parent(s): 98971c8

Update src/teams/agentteam.py

Browse files

just changes from boos_aid to scitonic

Files changed (1) hide show
  1. src/teams/agentteam.py +8 -8
src/teams/agentteam.py CHANGED
@@ -1,11 +1,11 @@
1
  import autogen
2
- from src.agentics.agents import AgentsFactory
3
 
4
  # agents_factory = AgentsFactory()
5
 
6
  # Function to reset agents
7
  def _reset_agents():
8
- boss_aid.reset()
9
 
10
 
11
  # Define functions for each team
@@ -19,7 +19,7 @@ def codingteam():
19
  )
20
 
21
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
22
- boss_aid.initiate_chat(manager, problem=PROBLEM, n_results=3)
23
 
24
  def covid19team():
25
  _reset_agents()
@@ -30,7 +30,7 @@ def covid19team():
30
  )
31
 
32
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
33
- boss_aid.initiate_chat(manager, covid19_problem=COVID19_PROBLEM, n_results=3)
34
 
35
  def financeteam():
36
  _reset_agents()
@@ -42,7 +42,7 @@ def financeteam():
42
  )
43
 
44
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
45
- boss_aid.initiate_chat(manager, finance_problem=FINANCE_PROBLEM, n_results=3)
46
 
47
  def debateteam():
48
  _reset_agents()
@@ -54,7 +54,7 @@ def debateteam():
54
  )
55
 
56
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
57
- boss_aid.initiate_chat(manager, debate_problem=DEBATE_PROBLEM, n_results=3)
58
 
59
  def homeworkteam():
60
  _reset_agents()
@@ -66,7 +66,7 @@ def homeworkteam():
66
  )
67
 
68
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
69
- boss_aid.initiate_chat(manager, homework_problem=HOMEWORK_PROBLEM, n_results=3)
70
 
71
  def consultingteam():
72
  _reset_agents()
@@ -78,4 +78,4 @@ def consultingteam():
78
  )
79
 
80
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
81
- boss_aid.initiate_chat(manager, consulting_problem=CONSULTING_PROBLEM, n_results=3)
 
1
  import autogen
2
+ from src.agentics.agents import AgentsFactory,scitonic, coder, pm, reviewer, covid19_scientist, healthcare_expert, finance_analyst, finance_expert, debate_expert, debate_champion, academic_expert, academic_whiz, consultant, consulting_pro
3
 
4
  # agents_factory = AgentsFactory()
5
 
6
  # Function to reset agents
7
  def _reset_agents():
8
+ scitonic.reset()
9
 
10
 
11
  # Define functions for each team
 
19
  )
20
 
21
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
22
+ scitonic.initiate_chat(manager, problem=PROBLEM, n_results=3)
23
 
24
  def covid19team():
25
  _reset_agents()
 
30
  )
31
 
32
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
33
+ scitonic.initiate_chat(manager, covid19_problem=COVID19_PROBLEM, n_results=3)
34
 
35
  def financeteam():
36
  _reset_agents()
 
42
  )
43
 
44
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
45
+ scitonic.initiate_chat(manager, finance_problem=FINANCE_PROBLEM, n_results=3)
46
 
47
  def debateteam():
48
  _reset_agents()
 
54
  )
55
 
56
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
57
+ scitonic.initiate_chat(manager, debate_problem=DEBATE_PROBLEM, n_results=3)
58
 
59
  def homeworkteam():
60
  _reset_agents()
 
66
  )
67
 
68
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
69
+ scitonic.initiate_chat(manager, homework_problem=HOMEWORK_PROBLEM, n_results=3)
70
 
71
  def consultingteam():
72
  _reset_agents()
 
78
  )
79
 
80
  manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
81
+ scitonic.initiate_chat(manager, consulting_problem=CONSULTING_PROBLEM, n_results=3)