Vaishak G Kumar commited on
Commit
c5cb0f3
·
verified ·
1 Parent(s): 7eb7ed0

Update src/agentics/agents.py

Browse files
Files changed (1) hide show
  1. src/agentics/agents.py +18 -18
src/agentics/agents.py CHANGED
@@ -22,17 +22,17 @@ llm_config = {
22
  "temperature": 0,
23
  }
24
 
25
- def termination_msg(self, x):
26
  return isinstance(x, dict) and "TERMINATE" == str(x.get("content", ""))[-9:].upper()
27
 
28
  class AgentsFactory:
29
- def __init__(self, llm_config, db_path):
30
- self.llm_config = llm_config
31
- self.db_path = db_path
32
 
33
 
34
 
35
- def tonic(self) :
36
  return autogen.UserProxyAgent(
37
  name="Boss",
38
  is_termination_msg=termination_msg,
@@ -43,7 +43,7 @@ class AgentsFactory:
43
  )
44
 
45
  # Create the RetrieveUserProxyAgent (Boss Assistant)
46
- def scitonic(self) :
47
  return RetrieveUserProxyAgent(
48
  name="Boss_Assistant",
49
  is_termination_msg=termination_msg,
@@ -62,21 +62,21 @@ class AgentsFactory:
62
  code_execution_config=False,
63
  )
64
  # Placeholder definitions for agents used in team functions
65
- def coder(self) :
66
  return AssistantAgent(
67
  name="Coder",
68
  system_message="You are a coder. Help in writing and reviewing code.",
69
  llm_config=llm_config
70
  )
71
 
72
- def pm(self) :
73
  return AssistantAgent(
74
  name="Project_Manager",
75
  system_message="You are a project manager. Coordinate tasks and ensure project success.",
76
  llm_config=llm_config
77
  )
78
 
79
- def reviewer(self) :
80
  return AssistantAgent(
81
  name="Reviewer",
82
  system_message="You are a code reviewer. Provide feedback on code quality.",
@@ -84,62 +84,62 @@ class AgentsFactory:
84
  )
85
 
86
  # Define more agents for each team
87
- def finance_expert(self) :
88
  return AssistantAgent(
89
  name="Finance_Expert",
90
  system_message="You are a finance expert. Provide insights on financial matters.",
91
  llm_config=llm_config
92
  )
93
 
94
- def debate_champion(self) :
95
  return AssistantAgent(
96
  name="Debate_Champion",
97
  system_message="You are a debate champion. Contribute to meaningful debates.",
98
  llm_config=llm_config
99
  )
100
 
101
- def academic_whiz(self) :
102
  return AssistantAgent(
103
  name="Academic_Whiz",
104
  system_message="You are an academic whiz. Offer solutions to academic challenges.",
105
  llm_config=llm_config
106
  )
107
 
108
- def consulting_pro(self) :
109
  return AssistantAgent(
110
  name="Consulting_Pro",
111
  system_message="You are a consulting professional. Offer professional advice and solutions.",
112
  llm_config=llm_config
113
  )
114
- def covid19_scientist(self) :
115
  return AssistantAgent(
116
  name="Covid19_Scientist",
117
  system_message="You are a scientist studying Covid-19 trends. Provide analysis and insights.",
118
  llm_config=llm_config
119
  )
120
 
121
- def healthcare_expert(self) :
122
  return AssistantAgent(
123
  name="Healthcare_Expert",
124
  system_message="You are a healthcare expert focused on managing and mitigating the impact of Covid-19.",
125
  llm_config=llm_config
126
  )
127
 
128
- def finance_analyst(self) :
129
  return AssistantAgent(
130
  name="Finance_Analyst",
131
  system_message="You are a finance analyst. Provide insights on the economic impact of Covid-19.",
132
  llm_config=llm_config
133
  )
134
 
135
- def debate_expert(self) :
136
  return AssistantAgent(
137
  name="Debate_Expert",
138
  system_message="You are an expert in debate strategies and communication. Participate in meaningful debates.",
139
  llm_config=llm_config
140
  )
141
 
142
- def academic_expert(self) :
143
  return AssistantAgent(
144
  name="Academic_Expert",
145
  system_message="You are an academic expert. Provide assistance and insights for academic challenges.",
 
22
  "temperature": 0,
23
  }
24
 
25
+ def termination_msg(x):
26
  return isinstance(x, dict) and "TERMINATE" == str(x.get("content", ""))[-9:].upper()
27
 
28
  class AgentsFactory:
29
+ def __init__(llm_config, db_path):
30
+ llm_config = llm_config
31
+ db_path = db_path
32
 
33
 
34
 
35
+ def tonic() :
36
  return autogen.UserProxyAgent(
37
  name="Boss",
38
  is_termination_msg=termination_msg,
 
43
  )
44
 
45
  # Create the RetrieveUserProxyAgent (Boss Assistant)
46
+ def scitonic() :
47
  return RetrieveUserProxyAgent(
48
  name="Boss_Assistant",
49
  is_termination_msg=termination_msg,
 
62
  code_execution_config=False,
63
  )
64
  # Placeholder definitions for agents used in team functions
65
+ def coder() :
66
  return AssistantAgent(
67
  name="Coder",
68
  system_message="You are a coder. Help in writing and reviewing code.",
69
  llm_config=llm_config
70
  )
71
 
72
+ def pm() :
73
  return AssistantAgent(
74
  name="Project_Manager",
75
  system_message="You are a project manager. Coordinate tasks and ensure project success.",
76
  llm_config=llm_config
77
  )
78
 
79
+ def reviewer() :
80
  return AssistantAgent(
81
  name="Reviewer",
82
  system_message="You are a code reviewer. Provide feedback on code quality.",
 
84
  )
85
 
86
  # Define more agents for each team
87
+ def finance_expert() :
88
  return AssistantAgent(
89
  name="Finance_Expert",
90
  system_message="You are a finance expert. Provide insights on financial matters.",
91
  llm_config=llm_config
92
  )
93
 
94
+ def debate_champion() :
95
  return AssistantAgent(
96
  name="Debate_Champion",
97
  system_message="You are a debate champion. Contribute to meaningful debates.",
98
  llm_config=llm_config
99
  )
100
 
101
+ def academic_whiz() :
102
  return AssistantAgent(
103
  name="Academic_Whiz",
104
  system_message="You are an academic whiz. Offer solutions to academic challenges.",
105
  llm_config=llm_config
106
  )
107
 
108
+ def consulting_pro() :
109
  return AssistantAgent(
110
  name="Consulting_Pro",
111
  system_message="You are a consulting professional. Offer professional advice and solutions.",
112
  llm_config=llm_config
113
  )
114
+ def covid19_scientist() :
115
  return AssistantAgent(
116
  name="Covid19_Scientist",
117
  system_message="You are a scientist studying Covid-19 trends. Provide analysis and insights.",
118
  llm_config=llm_config
119
  )
120
 
121
+ def healthcare_expert() :
122
  return AssistantAgent(
123
  name="Healthcare_Expert",
124
  system_message="You are a healthcare expert focused on managing and mitigating the impact of Covid-19.",
125
  llm_config=llm_config
126
  )
127
 
128
+ def finance_analyst() :
129
  return AssistantAgent(
130
  name="Finance_Analyst",
131
  system_message="You are a finance analyst. Provide insights on the economic impact of Covid-19.",
132
  llm_config=llm_config
133
  )
134
 
135
+ def debate_expert() :
136
  return AssistantAgent(
137
  name="Debate_Expert",
138
  system_message="You are an expert in debate strategies and communication. Participate in meaningful debates.",
139
  llm_config=llm_config
140
  )
141
 
142
+ def academic_expert() :
143
  return AssistantAgent(
144
  name="Academic_Expert",
145
  system_message="You are an academic expert. Provide assistance and insights for academic challenges.",