Update multi_agent.py

multi_agent.py

@@ -5,19 +5,13 @@ from typing_extensions import Annotated
 board = None
 board_svgs = None
 made_move = None
-game_over = False
-move_num = 0
 
 def get_legal_moves() -> Annotated[str, "A list of legal moves in UCI format"]:
-    print("###### get_legal_moves")
     return "Possible moves are: " + ",".join(
         [str(move) for move in board.legal_moves]
     )
 
 def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "Result of the move."]:
-    global move_num
-    move_num += 1
-    print("###### make_move: " + str(move_num))
     move = chess.Move.from_uci(move)
     board.push_uci(str(move))
     global made_move
@@ -42,33 +36,15 @@ def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "
            f"{chess.SQUARE_NAMES[move.from_square]} to "\
            f"{chess.SQUARE_NAMES[move.to_square]}."
 
-def set_game_over():
-    print("###### set_game_over")
-    global game_over
-    
-    game_over = True
-    
 def check_made_move(msg):
     global made_move
-
-    print("###### check_made_move: " + str(made_move))
-
+    
     if made_move:
         made_move = False
         return True
     else:
         return False
 
-def check_game_over(msg):
-    global game_over
-
-    print("###### check_game_over: " + str(game_over))
-    
-    if game_over:
-        return True
-    else:
-        return False
-
 def get_num_turns(num_moves):
     # Each turn includes two moves (one by each player)
     # The first move by player black kicks off the chat
@@ -95,46 +71,30 @@ def run_multi_agent(llm_white, llm_black, num_moves):
     
     board_proxy = ConversableAgent(
         name="Board Proxy",
-        llm_config=llm_config_white,
+        llm_config=False,
         is_termination_msg=check_made_move,
         default_auto_reply="Please make a move.",
         human_input_mode="NEVER",
-        system_message = "Your loyal assistant! Reply 'TERMINATE' when no legal move is possible."
     )
     
     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. "
-        "If there is no legal move, call set_game_over() and reply 'TERMINATE'. "
-        "Else call make_move(move) to make a legal move. ",
-        #"Analyze the move in 3 bullet points. Respond in format **Analysis:** move in UCI format (emoji of piece), unordered list.",
+        "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 (piece emoji), unordered list.",
         llm_config=llm_config_white,
-        human_input_mode="NEVER",
-        is_termination_msg=check_game_over
     )
     
     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. "
-        "If there is no legal move, call set_game_over() and reply 'TERMINATE'. "
-        "Else call make_move(move) to make a legal move. ",
-        #"Analyze the move in 3 bullet points. Respond in format **Analysis:** move in UCI format (emoji of piece), unordered list.",
+        "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 (piece emoji), unordered list.",
         llm_config=llm_config_black,
-        human_input_mode="NEVER",
-        is_termination_msg=check_game_over
     )
-
-    msg1 = player_white.system_message
-    print("msg1=" + msg1)
-
-    msg2 = player_black.system_message
-    print("msg2=" + msg2)
-
-    msg3 = board_proxy.system_message
-    print("msg3=" + msg3)
-
+    
     for caller in [player_white, player_black]:
         register_function(
             get_legal_moves,
@@ -151,14 +111,6 @@ def run_multi_agent(llm_white, llm_black, num_moves):
             name="make_move",
             description="Call this tool to make a move.",
         )
-
-        register_function(
-            set_game_over,
-            caller=caller,
-            executor=board_proxy,
-            name="set_game_over",
-            description="Call this tool to end the game.",
-        )
     
     player_white.register_nested_chats(
         trigger=player_black,
@@ -219,8 +171,8 @@ def run_multi_agent(llm_white, llm_black, num_moves):
         if num_moves % 2 == 0 and num_move == num_moves + 1:
             break
 
-    print("===")
-    print(result)
-    print("===")
+    #print("===")
+    #print(result)
+    #print("===")
     
     return result