Update multi_agent.py

multi_agent.py

@@ -25,8 +25,6 @@ def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "
         size=250
     ))
 
-    print("### " + str(len(board_svgs)))
-    
     piece = board.piece_at(move.to_square)
     piece_symbol = piece.unicode_symbol()
     piece_name = (
@@ -74,8 +72,8 @@ def run_multi_agent(llm_white, llm_black, num_moves):
     
     player_white = ConversableAgent(
         name="Player White",
-        system_message="You are a chess player and you play as white. "
-        "First call get_legal_moves(), to get a list of legal moves. "
+        system_message="You are a chess Grandmaster and you play as white. "
+        "First call get_legal_moves(), to get a list of legal moves. If there are no legal moves, terminate the game. "
         "Then call make_move(move) to make a move. "
         "After a move is made, analyze the move in 3 bullet points. Respond in format **Analysis:** move from/to, unordered list.",
         llm_config=llm_config_white,
@@ -83,8 +81,8 @@ def run_multi_agent(llm_white, llm_black, num_moves):
     
     player_black = ConversableAgent(
         name="Player Black",
-        system_message="You are a chess player and you play as black. "
-        "First call get_legal_moves(), to get a list of legal moves. "
+        system_message="You are a chess Grandmaster and you play as black. "
+        "First call get_legal_moves(), to get a list of legal moves. If there are no legal moves, terminate the game. "
         "Then call make_move(move) to make a move. "
         "After a move is made, analyze the move in 3 bullet points. Respond in format **Analysis:** move from/to, unordered list.",
         llm_config=llm_config_black,