Update multi_agent.py

multi_agent.py

@@ -5,6 +5,7 @@ from typing_extensions import Annotated
 board = None
 board_svgs = None
 made_move = None
+gaem_over = False
 
 def get_legal_moves() -> Annotated[str, "A list of legal moves in UCI format"]:
     return "Possible moves are: " + ",".join(
@@ -36,6 +37,9 @@ 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 game_over():
+    game_over = True
+    
 def check_made_move(msg):
     global made_move
     
@@ -45,6 +49,14 @@ def check_made_move(msg):
     else:
         return False
 
+def check_game_over(msg):
+    global 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
@@ -80,19 +92,23 @@ def run_multi_agent(llm_white, llm_black, num_moves):
     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 no legal moves, EXIT. "
-        "Then call make_move(move) to make a legal move. "
+        "First call get_legal_moves(), to get a list of legal moves. "
+        "If there is no legal move, call game_over() "
+        "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, unordered list.",
-        llm_config=llm_config_white
+        llm_config=llm_config_white,
+        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 no legal moves, EXIT. "
-        "Then call make_move(move) to make a legal move. "
+        "First call get_legal_moves(), to get a list of legal moves. "
+        "If there is no legal move, call game_over() "
+        "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, unordered list.",
-        llm_config=llm_config_black
+        llm_config=llm_config_black,
+        is_termination_msg=check_game_over
     )
     
     for caller in [player_white, player_black]: