Fix groupchat none

app.py

@@ -34,25 +34,42 @@ 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"):
@@ -106,8 +123,12 @@ 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")
-select_speaker_method = pn.widgets.Select(name="", options=["round_robin", "auto", "random"], value="round_robin")
+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: "),
@@ -176,7 +197,8 @@ 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"])
@@ -223,9 +245,16 @@ def init_groupchat(event, collection_name):
             else False
         )
         agent = initialize_agents(
-            llm_config, agent_name, system_msg, agent_type, retrieve_config, code_execution_config
+            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}
         )
-        agent.register_reply([autogen.Agent, None], reply_func=send_messages, config={"callback": None})
         agents.append(agent)
     if len(agents) >= 3:
         groupchat = autogen.GroupChat(
@@ -238,11 +267,18 @@ def init_groupchat(event, collection_name):
         manager = myGroupChatManager(groupchat=groupchat, llm_config=llm_config)
     else:
         manager = None
+        groupchat = None
     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:
@@ -257,7 +293,10 @@ async def reply_chat(contents, user, instance):
         instance.collection_name = 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:
+    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
@@ -328,10 +367,18 @@ 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"),
@@ -499,7 +546,9 @@ 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)