Update multi_agent.py

multi_agent.py

@@ -1,14 +1,7 @@
 import base64, datetime, json, os
 
-from autogen import ConversableAgent, AssistantAgent, register_function
+from autogen import ConversableAgent, AssistantAgent
 from autogen.coding import LocalCommandLineCodeExecutor
-from datetime import date
-from typing_extensions import Annotated
-
-###
-def get_today() -> Annotated[str, "Get today's date"]:
-    return str(date.today())
-###
 
 def read_file(file_path: str) -> str:
     with open(file_path, "r", encoding="utf-8") as file:
@@ -56,7 +49,7 @@ def run_multi_agent(llm, task):
 
     code_executor_agent = ConversableAgent(
         name="code_executor_agent",
-        llm_config=llm_config,
+        llm_config=False,
         code_execution_config={"executor": executor},
         human_input_mode="NEVER",
         default_auto_reply="TERMINATE",
@@ -65,35 +58,24 @@ def run_multi_agent(llm, task):
     code_writer_agent = AssistantAgent(
         name="code_writer_agent",
         llm_config=llm_config,
-        system_message="You are a Sr. Python Programmer, an expert in writing Pylint-compliant code. "
-            "If needed, call get_today() to get today's date.",
+        system_message="You are a Sr. Python Programmer, an expert in writing Pylint-compliant code.",
         code_execution_config=False,
         human_input_mode="NEVER",
     )
-
-    ###
-    register_function(
-        get_today,
-        caller=code_executor_agent,
-        executor=code_writer_agent,
-        name="get_today",
-        description="Call this tool to get today's date.",
-    )    
-    ###
     
     chat_result = code_executor_agent.initiate_chat(
         code_writer_agent,
         message=task,
-        max_turns=15
+        max_turns=10
     )
 
     chat = ""
-    #first_message = True
+    first_message = True
     
     for message in chat_result.chat_history:
-        #if not first_message:
-        chat += f"**{message['role'].replace('assistant', 'Code Executor').replace('user', 'Code Writer')}**\n{message['content']}\n\n"
-        #first_message = False
+        if not first_message:
+            chat += f"**{message['role'].replace('assistant', 'Code Executor').replace('user', 'Code Writer')}**\n{message['content']}\n\n"
+        first_message = False
 
     file_name_png = get_latest_file("coding", ".png")