fix issues with groupchat and version bump. address #1580

microsoft · Feb 7, 2024 · 95225fb · 95225fb
1 parent 3f4b6e5
commit 95225fb
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 10 deletions.
diff --git a/samples/apps/autogen-studio/autogenstudio/utils/dbdefaults.json b/samples/apps/autogen-studio/autogenstudio/utils/dbdefaults.json
@@ -138,7 +138,7 @@
                 },
                 "human_input_mode": "NEVER",
                 "max_consecutive_auto_reply": 8,
-                "system_message": "You are a helpful assistant that can suggest a travel plan for a user. You are the primary cordinator who will receive suggestions or advice from other agents (local_assistant, language_assistant). You must ensure that the finally plan integrates the suggestions from other agents or team members. YOUR FINAL RESPONSE MUST BE THE COMPLETE PLAN that ends with the word TERMINATE. "
+                "system_message": "You are a helpful assistant that can suggest a travel plan for a user. You are the primary cordinator who will receive suggestions or advice from other agents (local_assistant, language_assistant). You must ensure that the finally plan integrates the suggestions from other agents or team members. YOUR FINAL RESPONSE MUST BE THE COMPLETE PLAN."
               }
             },
             {

diff --git a/samples/apps/autogen-studio/autogenstudio/version.py b/samples/apps/autogen-studio/autogenstudio/version.py
@@ -1,3 +1,3 @@
-VERSION = "0.0.43a"
+VERSION = "0.0.44a"
 __version__ = VERSION
 APP_NAME = "autogenstudio"
diff --git a/samples/apps/autogen-studio/autogenstudio/workflowmanager.py b/samples/apps/autogen-studio/autogenstudio/workflowmanager.py
@@ -38,8 +38,8 @@ def __init__(
 
         if config.receiver.type == "groupchat":
             # append self.sender to the list of agents
-            self.receiver._groupchat.agents.append(self.sender)
-            print(self.receiver)
+            self.receiver._groupchat.agents = self.receiver._groupchat.agents + \
+                [self.sender]
         self.agent_history = []
 
         if history:
@@ -147,11 +147,14 @@ def get_default_system_message(agent_type: str) -> str:
         if agent_spec.skills:
             # get skill prompt, also write skills to a file named skills.py
             skills_prompt = ""
-            skills_prompt = get_skills_from_prompt(agent_spec.skills, self.work_dir)
+            skills_prompt = get_skills_from_prompt(
+                agent_spec.skills, self.work_dir)
             if agent_spec.config.system_message:
-                agent_spec.config.system_message = agent_spec.config.system_message + "\n\n" + skills_prompt
+                agent_spec.config.system_message = agent_spec.config.system_message + \
+                    "\n\n" + skills_prompt
             else:
-                agent_spec.config.system_message = get_default_system_message(agent_spec.type) + "\n\n" + skills_prompt
+                agent_spec.config.system_message = get_default_system_message(
+                    agent_spec.type) + "\n\n" + skills_prompt
 
         return agent_spec
 
@@ -173,8 +176,10 @@ def load(self, agent_spec: AgentFlowSpec) -> autogen.Agent:
             group_chat_config = agent_spec.groupchat_config.dict()
             group_chat_config["agents"] = agents
             groupchat = autogen.GroupChat(**group_chat_config)
-            agent = autogen.GroupChatManager(groupchat=groupchat, **agent_spec.config.dict())
-            agent.register_reply([autogen.Agent, None], reply_func=self.process_reply, config={"callback": None})
+            agent = autogen.GroupChatManager(
+                groupchat=groupchat, **agent_spec.config.dict())
+            agent.register_reply(
+                [autogen.Agent, None], reply_func=self.process_reply, config={"callback": None})
             return agent
 
         else:
@@ -198,7 +203,8 @@ def load_agent_config(self, agent_config: AgentConfig, agent_type: str) -> autog
             agent = autogen.UserProxyAgent(**agent_config.dict())
         else:
             raise ValueError(f"Unknown agent type: {agent_type}")
-        agent.register_reply([autogen.Agent, None], reply_func=self.process_reply, config={"callback": None})
+        agent.register_reply(
+            [autogen.Agent, None], reply_func=self.process_reply, config={"callback": None})
         return agent
 
     def run(self, message: str, clear_history: bool = False) -> None: