Remove print message

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,7 +123,9 @@ 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"
+)
 template.main.append(
     pn.Row(
         pn.pane.Markdown("## Add or Remove Agents: "),
@@ -155,7 +174,7 @@ btn_remove.on_click(remove_agent)
 
 
 async def send_messages(recipient, messages, sender, config):
-    print(f"{sender.name} -> {recipient.name}: {messages[-1]['content']}")
+    # print(f"{sender.name} -> {recipient.name}: {messages[-1]['content']}")
     chatiface.send(messages[-1]["content"], user=sender.name, respond=False)
     return False, None  # required to ensure the agent communication flow continues
 
@@ -173,7 +192,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"])
@@ -220,13 +240,24 @@ 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(
-            agents=agents, messages=[], max_round=12, speaker_selection_method="auto", allow_repeat_speaker=False
+            agents=agents,
+            messages=[],
+            max_round=12,
+            speaker_selection_method="auto",
+            allow_repeat_speaker=False,
         )
         manager = myGroupChatManager(groupchat=groupchat, llm_config=llm_config)
     else:
@@ -235,7 +266,13 @@ def init_groupchat(event, collection_name):
 
 
 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:
@@ -250,7 +287,10 @@ async def reply_chat(contents, user, instance):
         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:
+    if (
+        not hasattr(instance, "agent_list")
+        or instance.agents_list != column_agents_list
+    ):
         agents, manager = init_groupchat(None, collection_name)
         instance.manager = manager
         instance.agents = agents
@@ -321,10 +361,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"),
@@ -492,7 +540,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)