Spaces:
Sleeping
Sleeping
Vaishak G Kumar
commited on
Update src/agentics/agents.py
Browse files- src/agentics/agents.py +14 -14
src/agentics/agents.py
CHANGED
@@ -32,7 +32,7 @@ class AgentsFactory:
|
|
32 |
self.json_file_path = json_file_path
|
33 |
|
34 |
|
35 |
-
def tonic() :
|
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() :
|
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() :
|
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,62 +84,62 @@ class AgentsFactory:
|
|
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.",
|
|
|
32 |
self.json_file_path = json_file_path
|
33 |
|
34 |
|
35 |
+
def tonic(self) :
|
36 |
return autogen.UserProxyAgent(
|
37 |
name="Boss",
|
38 |
is_termination_msg=termination_msg,
|
|
|
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 |
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 |
)
|
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.",
|