Spaces:
Sleeping
Sleeping
Vaishak G Kumar
commited on
Update src/agentics/agents.py
Browse files- 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(
|
26 |
return isinstance(x, dict) and "TERMINATE" == str(x.get("content", ""))[-9:].upper()
|
27 |
|
28 |
class AgentsFactory:
|
29 |
-
def __init__(
|
30 |
-
|
31 |
-
|
32 |
|
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.",
|
|
|
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.",
|