Update multi_agent.py

multi_agent.py

@@ -20,7 +20,7 @@ board_svgs = None
 
 num_moves = 0
 move_num = 0
-made_move = None
+legal_moves = ""
 
 class AgentState(TypedDict):
     messages: Annotated[Sequence[BaseMessage], operator.add]
@@ -45,7 +45,7 @@ def create_agent(llm: ChatOpenAI, tools: list, system_prompt: str):
                          handle_parsing_errors=True,
                          return_intermediate_steps=True,
                          verbose=True,
-                         max_iterations=3) #get_num_turns(num_moves))
+                         max_iterations=get_num_turns(num_moves))
 
 def agent_node(state, agent, name):
     try:
@@ -64,9 +64,9 @@ def get_legal_moves() -> Annotated[str, "A list of legal moves in UCI format"]:
        and this function will always return legal moves in UCI format."""
     try:
         print("## get_legal_moves")
-        return "Possible moves are: " + ",".join(
-            [str(move) for move in board.legal_moves]
-        )
+        global legal_moves
+        legal_moves = ",".join([str(move) for move in board.legal_moves])
+        return legal_moves
     except Exception as e:
         print(f"An error occurred in get_legal_moves: {e}")
         return "Error: unable to get legal moves"
@@ -80,8 +80,6 @@ def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "
         print("## make_move")
         move = chess.Move.from_uci(move)
         board.push_uci(str(move))
-        global made_move
-        made_move = True
         
         board_svgs.append(chess.svg.board(
             board,
@@ -108,16 +106,6 @@ def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "
     except Exception as e:
         print(f"An error occurred in make_move: {e}")
         return f"Error: unable to make move {move}"
-
-def check_made_move(msg):
-    print("## check_made_move")
-    global made_move
-    
-    if made_move:
-        made_move = False
-        return True
-    else:
-        return False
         
 def get_num_turns(num_moves):
     # Each turn includes two moves (one by each player)
@@ -214,9 +202,6 @@ def create_graph():
     graph.add_node("player_black", player_black_node)
     graph.add_node("manager", supervisor_chain)
 
-#    for player in players:
-#        graph.add_edge(player, "manager")
-
     graph.add_conditional_edges(
         "player_white", 
         should_continue, 
@@ -239,10 +224,13 @@ def create_graph():
 
 def should_continue(state):
     print("#### should_continue")
-    global move_num, num_moves
+    global move_num, num_moves, legal_moves
     if move_num == num_moves:
         print("False")
         return END # max moves reached
+    if not legal_moves:
+        print("False")
+        return END # checkmate or stalemate
     print("True")
     return "manager"