Add speaker selection method, update timeout logic, add agents config changes detection, add naming rule

app.py

@@ -34,42 +34,25 @@ def get_description_text():
     """
 
 
-template.main.append(
-    pn.pane.Markdown(get_description_text(), sizing_mode="stretch_width")
-)
+template.main.append(pn.pane.Markdown(get_description_text(), sizing_mode="stretch_width"))
 
 txt_model = TextInput(
-    name="Model Name",
-    placeholder="Enter your model name here...",
-    value="gpt-35-turbo",
-    sizing_mode="stretch_width",
+    name="Model Name", placeholder="Enter your model name here...", value="gpt-35-turbo", sizing_mode="stretch_width"
 )
 pwd_openai_key = PasswordInput(
-    name="OpenAI API Key",
-    placeholder="Enter your OpenAI API Key here...",
-    sizing_mode="stretch_width",
+    name="OpenAI API Key", placeholder="Enter your OpenAI API Key here...", sizing_mode="stretch_width"
 )
 pwd_openai_url = PasswordInput(
-    name="OpenAI Base Url",
-    placeholder="Enter your OpenAI Base Url here...",
-    sizing_mode="stretch_width",
+    name="OpenAI Base Url", placeholder="Enter your OpenAI Base Url here...", sizing_mode="stretch_width"
 )
 pwd_aoai_key = PasswordInput(
-    name="Azure OpenAI API Key",
-    placeholder="Enter your Azure OpenAI API Key here...",
-    sizing_mode="stretch_width",
+    name="Azure OpenAI API Key", placeholder="Enter your Azure OpenAI API Key here...", sizing_mode="stretch_width"
 )
 pwd_aoai_url = PasswordInput(
-    name="Azure OpenAI Base Url",
-    placeholder="Enter your Azure OpenAI Base Url here...",
-    sizing_mode="stretch_width",
+    name="Azure OpenAI Base Url", placeholder="Enter your Azure OpenAI Base Url here...", sizing_mode="stretch_width"
 )
 file_cfg = pn.widgets.FileInput(filename="OAI_CONFIG_LIST", sizing_mode="stretch_width")
-template.main.append(
-    pn.Row(
-        txt_model, pwd_openai_key, pwd_openai_url, pwd_aoai_key, pwd_aoai_url, file_cfg
-    )
-)
+template.main.append(pn.Row(txt_model, pwd_openai_key, pwd_openai_url, pwd_aoai_key, pwd_aoai_url, file_cfg))
 
 
 def get_config(tmpfilename="OAI_CONFIG_LIST"):
@@ -123,9 +106,8 @@ def get_config(tmpfilename="OAI_CONFIG_LIST"):
 
 btn_add = Button(name="+", button_type="success")
 btn_remove = Button(name="-", button_type="danger")
-switch_code = Switch(
-    name="Run Code", sizing_mode="fixed", width=50, height=30, align="end"
-)
+switch_code = Switch(name="Run Code", sizing_mode="fixed", width=50, height=30, align="end")
+select_speaker_method = pn.widgets.Select(name="", options=["round_robin", "auto", "random"], value="round_robin")
 template.main.append(
     pn.Row(
         pn.pane.Markdown("## Add or Remove Agents: "),
@@ -133,6 +115,8 @@ template.main.append(
         btn_remove,
         pn.pane.Markdown("### Run Code: "),
         switch_code,
+        pn.pane.Markdown("### Speaker Selection Method: "),
+        select_speaker_method,
     )
 )
 
@@ -192,8 +176,7 @@ class myGroupChatManager(autogen.GroupChatManager):
                     content = autogen.OpenAIWrapper.instantiate(
                         content,
                         message["context"],
-                        self.llm_config
-                        and self.llm_config.get("allow_format_str_template", False),
+                        self.llm_config and self.llm_config.get("allow_format_str_template", False),
                     )
             if "function_call" in message:
                 function_call = dict(message["function_call"])
@@ -240,39 +223,26 @@ def init_groupchat(event, collection_name):
             else False
         )
         agent = initialize_agents(
-            llm_config,
-            agent_name,
-            system_msg,
-            agent_type,
-            retrieve_config,
-            code_execution_config,
-        )
-        agent.register_reply(
-            [autogen.Agent, None], reply_func=send_messages, config={"callback": None}
+            llm_config, agent_name, system_msg, agent_type, retrieve_config, code_execution_config
         )
+        agent.register_reply([autogen.Agent, None], reply_func=send_messages, config={"callback": None})
         agents.append(agent)
     if len(agents) >= 3:
         groupchat = autogen.GroupChat(
             agents=agents,
             messages=[],
             max_round=12,
-            speaker_selection_method="auto",
+            speaker_selection_method=select_speaker_method.value,
             allow_repeat_speaker=False,
         )
         manager = myGroupChatManager(groupchat=groupchat, llm_config=llm_config)
     else:
         manager = None
-    return agents, manager
+    return agents, manager, groupchat
 
 
 async def agents_chat(init_sender, manager, contents, agents):
-    recipient = (
-        manager
-        if len(agents) > 2
-        else agents[1]
-        if agents[1] != init_sender
-        else agents[0]
-    )
+    recipient = manager if len(agents) > 2 else agents[1] if agents[1] != init_sender else agents[0]
     if isinstance(init_sender, (RetrieveUserProxyAgent, MathUserProxyAgent)):
         await init_sender.a_initiate_chat(recipient, problem=contents)
     else:
@@ -286,12 +256,9 @@ async def reply_chat(contents, user, instance):
         collection_name = f"{int(time.time())}_{random.randint(0, 100000)}"
         instance.collection_name = collection_name
 
-    column_agents_list = [agent[0][0].value for agent in column_agents]
-    if (
-        not hasattr(instance, "agent_list")
-        or instance.agents_list != column_agents_list
-    ):
-        agents, manager = init_groupchat(None, collection_name)
+    column_agents_list = [[a.value for a in agent[0]] for agent in column_agents]
+    if not hasattr(instance, "agent_list") or instance.agents_list != column_agents_list:
+        agents, manager, groupchat = init_groupchat(None, collection_name)
         instance.manager = manager
         instance.agents = agents
         instance.agents_list = column_agents_list
@@ -321,7 +288,7 @@ async def reply_chat(contents, user, instance):
 
     if not init_sender:
         init_sender = agents[0]
-    await generate_code(agents, manager, contents, code_editor)
+    await generate_code(agents, manager, contents, code_editor, groupchat)
     await agents_chat(init_sender, manager, contents, agents)
     return "The task is done. Please start a new task."
 
@@ -361,18 +328,10 @@ btn_msg2.on_click(load_message)
 btn_msg3.on_click(load_message)
 
 
-btn_example1 = Button(
-    name="General 2 agents", button_type="primary", sizing_mode="stretch_width"
-)
-btn_example2 = Button(
-    name="RAG 2 agents", button_type="primary", sizing_mode="stretch_width"
-)
-btn_example3 = Button(
-    name="Software Dev 3 agents", button_type="primary", sizing_mode="stretch_width"
-)
-btn_example4 = Button(
-    name="Research 6 agents", button_type="primary", sizing_mode="stretch_width"
-)
+btn_example1 = Button(name="General 2 agents", button_type="primary", sizing_mode="stretch_width")
+btn_example2 = Button(name="RAG 2 agents", button_type="primary", sizing_mode="stretch_width")
+btn_example3 = Button(name="Software Dev 3 agents", button_type="primary", sizing_mode="stretch_width")
+btn_example4 = Button(name="Research 6 agents", button_type="primary", sizing_mode="stretch_width")
 template.main.append(
     pn.Row(
         pn.pane.Markdown("## Agent Examples: ", sizing_mode="stretch_width"),
@@ -540,9 +499,7 @@ btn_example2.on_click(load_example)
 btn_example3.on_click(load_example)
 btn_example4.on_click(load_example)
 
-code_editor = CodeEditor(
-    value="", sizing_mode="stretch_width", language="python", height=300
-)
+code_editor = CodeEditor(value="", sizing_mode="stretch_width", language="python", height=300)
 template.main.append(code_editor)
 
 template.servable(title=TITLE)

autogen_utils.py

@@ -182,11 +182,11 @@ async def get_human_input(name, prompt: str, instance=None) -> str:
     while True:
         if time.time() - ts > TIMEOUT:
             instance.send(
-                f"You didn't provide your feedback in {TIMEOUT} seconds, skip and use auto-reply.",
+                f"You didn't provide your feedback in {TIMEOUT} seconds, exit.",
                 user=name,
                 respond=False,
             )
-            reply = ""
+            reply = "exit"
             break
         if get_input_widget.value != "" and get_input_checkbox.value is True:
             get_input_widget.disabled = True
@@ -289,7 +289,7 @@ async def format_code(code_to_format: str) -> str:
     return formatted_code
 
 
-async def generate_code(agents, manager, contents, code_editor):
+async def generate_code(agents, manager, contents, code_editor, groupchat):
     code = """import autogen
 import os
 from autogen.agentchat.contrib.retrieve_user_proxy_agent import RetrieveUserProxyAgent
@@ -397,6 +397,7 @@ agent = UserProxyAgent(
     name="{agent.name}",
     is_termination_msg=termination_msg,
     human_input_mode="TERMINATE",
+    system_message=\"\"\"{agent.system_message}\"\"\",
     default_auto_reply="{DEFAULT_AUTO_REPLY}",
     max_consecutive_auto_reply=5,
     code_execution_config={agent._code_execution_config},
@@ -440,9 +441,9 @@ if not init_sender:
     code += _code
 
     if manager:
-        _code = """
+        _code = f"""
 groupchat = autogen.GroupChat(
-    agents=agents, messages=[], max_round=12, speaker_selection_method="auto", allow_repeat_speaker=False
+    agents=agents, messages=[], max_round=12, speaker_selection_method="{groupchat.speaker_selection_method}", allow_repeat_speaker=False
 )  # todo: auto, sometimes message has no name
 manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=llm_config)
 

custom_widgets.py

@@ -28,14 +28,14 @@ class RowAgentWidget(Viewer):
         self._agent_name = TextInput(
             name="",
             value=params.get("value")[0],
-            placeholder="Agent Name",
+            placeholder="Agent Name (can only contain letters, numbers, and underscores))",
             min_width=100,
             sizing_mode="scale_width",
         )
         self._system_msg = TextInput(
             name="",
             value=params.get("value")[1],
-            placeholder="System Message",
+            placeholder="System Message, leave empty to use default",
             min_width=400,
             sizing_mode="scale_width",
         )