Update src/agentics/agents.py

src/agentics/agents.py

@@ -22,126 +22,121 @@ llm_config = {
          "temperature": 0,
      }
 
-def termination_msg(self, x):
+def termination_msg(x):
         return isinstance(x, dict) and "TERMINATE" == str(x.get("content", ""))[-9:].upper()
 
-class AgentsFactory:
-    def __init__(self, llm_config, db_path):
-        self.llm_config = llm_config
-        self.db_path = db_path
-
-    
-
-    def tonic(self) :
-        return autogen.UserProxyAgent(
-            name="Boss",
-            is_termination_msg=termination_msg,
-            human_input_mode="NEVER",
-            system_message="The boss who asks questions and gives tasks.",
-            code_execution_config=False,
-            default_auto_reply="Reply `TERMINATE` if the task is done.",
-        )
-
-    # Create the RetrieveUserProxyAgent (Boss Assistant)
-    def scitonic(self) :
-        return RetrieveUserProxyAgent(
-            name="Boss_Assistant",
-            is_termination_msg=termination_msg,
-            system_message="Assistant who has extra content retrieval power for solving difficult problems.",
-            human_input_mode="NEVER",
-            max_consecutive_auto_reply=3,
-            retrieve_config={
-                "task": "QuoraRetrieval",
-                "docs_path": self.db_path,
-                "chunk_token_size": 1000,
-                "model": llm_config["config_list"][0]["model"],
-                "client": chromadb.PersistentClient(path="/tmp/chromadb"),
-                "collection_name": "groupchat",
-                "get_or_create": True,
-            },
-            code_execution_config=False,
-        )
-    # Placeholder definitions for agents used in team functions
-    def coder(self) : 
-        return AssistantAgent(
-            name="Coder",
-            system_message="You are a coder. Help in writing and reviewing code.",
-            llm_config=llm_config
-        )
-
-    def pm(self) :
-        return AssistantAgent(
-            name="Project_Manager",
-            system_message="You are a project manager. Coordinate tasks and ensure project success.",
-            llm_config=llm_config
-        )
-
-    def reviewer(self) :
-        return AssistantAgent(
-            name="Reviewer",
-            system_message="You are a code reviewer. Provide feedback on code quality.",
-            llm_config=llm_config
-        )
-
-    # Define more agents for each team
-    def finance_expert(self) :
-        return AssistantAgent(
-            name="Finance_Expert",
-            system_message="You are a finance expert. Provide insights on financial matters.",
-            llm_config=llm_config
-        )
-
-    def debate_champion(self) :
-        return AssistantAgent(
-            name="Debate_Champion",
-            system_message="You are a debate champion. Contribute to meaningful debates.",
-            llm_config=llm_config
-        )
-
-    def academic_whiz(self) :
-        return AssistantAgent(
+def tonic():
+    return autogen.UserProxyAgent(
+        name="Boss",
+        is_termination_msg=termination_msg,
+        human_input_mode="NEVER",
+        system_message="The boss who asks questions and gives tasks.",
+        code_execution_config=False,
+        default_auto_reply="Reply `TERMINATE` if the task is done.",
+    )
+
+# Create the RetrieveUserProxyAgent (Boss Assistant)
+def scitonic():
+    return RetrieveUserProxyAgent(
+        name="Boss_Assistant",
+        is_termination_msg=termination_msg,
+        system_message="Assistant who has extra content retrieval power for solving difficult problems.",
+        human_input_mode="NEVER",
+        max_consecutive_auto_reply=3,
+        retrieve_config={
+            "task": "QuoraRetrieval",
+            "docs_path": self.db_path,
+            "chunk_token_size": 1000,
+            "model": llm_config["config_list"][0]["model"],
+            "client": chromadb.PersistentClient(path="/tmp/chromadb"),
+            "collection_name": "groupchat",
+            "get_or_create": True,
+        },
+        code_execution_config=False,
+    )
+
+# Placeholder definitions for agents used in team functions
+def coder():
+    return AssistantAgent(
+        name="Coder",
+        system_message="You are a coder. Help in writing and reviewing code.",
+        llm_config=llm_config
+    )
+
+def pm():
+    return AssistantAgent(
+        name="Project_Manager",
+        system_message="You are a project manager. Coordinate tasks and ensure project success.",
+        llm_config=llm_config
+    )
+
+def reviewer():
+    return AssistantAgent(
+        name="Reviewer",
+        system_message="You are a code reviewer. Provide feedback on code quality.",
+        llm_config=llm_config
+    )
+
+# Define more agents for each team
+def finance_expert():
+    return AssistantAgent(
+        name="Finance_Expert",
+        system_message="You are a finance expert. Provide insights on financial matters.",
+        llm_config=llm_config
+    )
+
+def debate_champion():
+    return AssistantAgent(
+        name="Debate_Champion",
+        system_message="You are a debate champion. Contribute to meaningful debates.",
+        llm_config=llm_config
+    )
+
+def academic_whiz():
+    return AssistantAgent(
         name="Academic_Whiz",
         system_message="You are an academic whiz. Offer solutions to academic challenges.",
         llm_config=llm_config
     )
 
-    def consulting_pro(self) :
-            return AssistantAgent(
-            name="Consulting_Pro",
-            system_message="You are a consulting professional. Offer professional advice and solutions.",
-            llm_config=llm_config
-        )
-    def covid19_scientist(self) : 
-        return AssistantAgent(
-            name="Covid19_Scientist",
-            system_message="You are a scientist studying Covid-19 trends. Provide analysis and insights.",
-            llm_config=llm_config
-        )
-
-    def healthcare_expert(self) : 
-        return AssistantAgent(
-            name="Healthcare_Expert",
-            system_message="You are a healthcare expert focused on managing and mitigating the impact of Covid-19.",
-            llm_config=llm_config
-        )
-
-    def finance_analyst(self) :
-        return AssistantAgent(
-            name="Finance_Analyst",
-            system_message="You are a finance analyst. Provide insights on the economic impact of Covid-19.",
-            llm_config=llm_config
-        )
-
-    def debate_expert(self) :
-        return AssistantAgent(
-            name="Debate_Expert",
-            system_message="You are an expert in debate strategies and communication. Participate in meaningful debates.",
-            llm_config=llm_config
-        )
-
-    def academic_expert(self) :
-        return AssistantAgent(
-            name="Academic_Expert",
-            system_message="You are an academic expert. Provide assistance and insights for academic challenges.",
-            llm_config=llm_config
-        )
+def consulting_pro():
+    return AssistantAgent(
+        name="Consulting_Pro",
+        system_message="You are a consulting professional. Offer professional advice and solutions.",
+        llm_config=llm_config
+    )
+
+def covid19_scientist():
+    return AssistantAgent(
+        name="Covid19_Scientist",
+        system_message="You are a scientist studying Covid-19 trends. Provide analysis and insights.",
+        llm_config=llm_config
+    )
+
+def healthcare_expert():
+    return AssistantAgent(
+        name="Healthcare_Expert",
+        system_message="You are a healthcare expert focused on managing and mitigating the impact of Covid-19.",
+        llm_config=llm_config
+    )
+
+def finance_analyst():
+    return AssistantAgent(
+        name="Finance_Analyst",
+        system_message="You are a finance analyst. Provide insights on the economic impact of Covid-19.",
+        llm_config=llm_config
+    )
+
+def debate_expert():
+    return AssistantAgent(
+        name="Debate_Expert",
+        system_message="You are an expert in debate strategies and communication. Participate in meaningful debates.",
+        llm_config=llm_config
+    )
+
+def academic_expert():
+    return AssistantAgent(
+        name="Academic_Expert",
+        system_message="You are an academic expert. Provide assistance and insights for academic challenges.",
+        llm_config=llm_config
+    )