Update multi_agent.py

multi_agent.py

@@ -63,116 +63,81 @@ def initialize():
     board_svgs = []
     made_move = False
 
-def run_multi_agent(llm_white, llm_black, num_moves):
-    initialize()
+def run_multi_agent(llm, task):
+    #initialize()
     
-    llm_config_white = {"model": llm_white}
-    llm_config_black = {"model": llm_black}
+    llm_config = {"model": llm}
     
-    board_proxy = ConversableAgent(
-        name="Board Proxy",
-        llm_config=False,
-        is_termination_msg=check_made_move,
-        default_auto_reply="Please make a move.",
-        human_input_mode="NEVER",
-    )
-    
-    player_white = ConversableAgent(
-        name="Player White",
-        system_message="You are a chess Grandmaster and you play as white. "
-        "First call get_legal_moves(), to get a list of legal moves. "
-        "Then call make_move(move) to make a legal move. "
-        "Analyze the move in 3 bullet points. Respond in format **Analysis:** move in UCI format, unordered list.",
-        llm_config=llm_config_white,
+    user_proxy = autogen.ConversableAgent(
+        name="Admin",
+        system_message="Give the task, and send "
+        "instructions to writer to refine the blog post.",
+        code_execution_config=False,
+        llm_config=llm_config,
+        human_input_mode="ALWAYS",
     )
-    
-    player_black = ConversableAgent(
-        name="Player Black",
-        system_message="You are a chess Grandmaster and you play as black. "
-        "First call get_legal_moves(), to get a list of legal moves. "
-        "Then call make_move(move) to make a legal move. "
-        "Analyze the move in 3 bullet points. Respond in format **Analysis:** move in UCI format, unordered list.",
-        llm_config=llm_config_black,
-    )
-    
-    for caller in [player_white, player_black]:
-        register_function(
-            get_legal_moves,
-            caller=caller,
-            executor=board_proxy,
-            name="get_legal_moves",
-            description="Call this tool to get legal moves.",
-        )
-    
-        register_function(
-            make_move,
-            caller=caller,
-            executor=board_proxy,
-            name="make_move",
-            description="Call this tool to make a move.",
-        )
-    
-    player_white.register_nested_chats(
-        trigger=player_black,
-        chat_queue=[
-            {
-                "sender": board_proxy,
-                "recipient": player_white,
-                "summary_method": "last_msg",
-                "silent": False,
-            }
-        ],
+
+    planner = autogen.ConversableAgent(
+        name="Planner",
+        system_message="Given a task, please determine "
+        "what information is needed to complete the task. "
+        "Please note that the information will all be retrieved using"
+        " Python code. Please only suggest information that can be "
+        "retrieved using Python code. "
+        "After each step is done by others, check the progress and "
+        "instruct the remaining steps. If a step fails, try to "
+        "workaround",
+        description="Planner. Given a task, determine what "
+        "information is needed to complete the task. "
+        "After each step is done by others, check the progress and "
+        "instruct the remaining steps",
+        llm_config=llm_config,
     )
-    
-    player_black.register_nested_chats(
-        trigger=player_white,
-        chat_queue=[
-            {
-                "sender": board_proxy,
-                "recipient": player_black,
-                "summary_method": "last_msg",
-                "silent": False,
-            }
-        ],
+
+    engineer = autogen.AssistantAgent(
+        name="Engineer",
+        llm_config=llm_config,
+        description="An engineer that writes code based on the plan "
+        "provided by the planner.",
     )
 
-    chat_result = None
-    chat_history = []
-    
-    try:
-        chat_result = player_black.initiate_chat(
-            player_white,
-            message="Let's play chess!",
-            max_turns=get_num_turns(num_moves),
-            verbose=True
-        )
-    except Exception as e:
-        print(f"Error: {e}")
-    finally:
-        if chat_result != None:
-            chat_history = chat_result.chat_history
-    
-    result = ""
-    num_move = 0
+    executor = autogen.ConversableAgent(
+        name="Executor",
+        system_message="Execute the code written by the "
+        "engineer and report the result.",
+        human_input_mode="NEVER",
+        code_execution_config={
+            "last_n_messages": 3,
+            "work_dir": "coding",
+            "use_docker": False,
+        },
+    )
 
-    for chat in chat_history:
-        player = ""
-        
-        if num_move % 2 == 0:
-            player = "Player Black"
-        else:
-            player = "Player White"
+    writer = autogen.ConversableAgent(
+        name="Writer",
+        llm_config=llm_config,
+        system_message="Writer."
+        "Please write blogs in markdown format (with relevant titles)"
+        " and put the content in pseudo ```md``` code block. "
+        "You take feedback from the admin and refine your blog.",
+        description="Writer."
+        "Write blogs based on the code execution results and take "
+        "feedback from the admin to refine the blog."
+    )
 
-        if num_move > 0:
-            result += f"**{player}, Move {num_move}**<br>{chat.get('content')}<br>{board_svgs[num_move - 1]}<br><br>"
-        
-        num_move += 1
+    groupchat = autogen.GroupChat(
+        agents=[user_proxy, engineer, writer, executor, planner],
+        messages=[],
+        max_round=10,
+    )
 
-        if num_moves % 2 == 0 and num_move == num_moves + 1:
-            break
+    manager = autogen.GroupChatManager(
+        groupchat=groupchat, llm_config=llm_config
+    )
 
-    #print("===")
-    #print(result)
-    #print("===")
+    groupchat_result = user_proxy.initiate_chat(
+        manager,
+        message=task,
+    )
     
-    return result
+    return groupchat_result