Update multi_agent.py

multi_agent.py

@@ -15,52 +15,6 @@ class AgentState(TypedDict):
     black_moves: Annotated[list[chess.Move], operator.add]
     board_svgs: Annotated[list[str], operator.add]
 
-@tool
-def get_legal_moves() -> Annotated[str, "A list of legal moves in UCI format"]:
-    """Returns a list of legal moves in UCI format. 
-       The input should always be an empty string, 
-       and this function will always return legal moves in UCI format."""
-    try:
-        return "Possible moves are: " + ",".join(
-            [str(move) for move in board.legal_moves]
-        )
-    except Exception as e:
-        print(f"An error occurred in get_legal_moves: {e}")
-        return "Error: unable to get legal moves"
-
-@tool
-def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "Result of the move."]:
-    """Makes a move. 
-       The input should always be a move in UCI format, 
-       and this function will always return the result of the move in UCI format."""
-    try:
-        move = chess.Move.from_uci(move)
-        board.push_uci(str(move))
-        global made_move
-        made_move = True
-        
-        board_svgs.append(chess.svg.board(
-            board,
-            arrows=[(move.from_square, move.to_square)],
-            fill={move.from_square: "gray"},
-            size=250
-        ))
-
-        piece = board.piece_at(move.to_square)
-        piece_symbol = piece.unicode_symbol()
-        piece_name = (
-            chess.piece_name(piece.piece_type).capitalize()
-            if piece_symbol.isupper()
-            else chess.piece_name(piece.piece_type)
-        )
-
-        return f"Moved {piece_name} ({piece_symbol}) from "\
-               f"{chess.SQUARE_NAMES[move.from_square]} to "\
-               f"{chess.SQUARE_NAMES[move.to_square]}."
-    except Exception as e:
-        print(f"An error occurred in make_move: {e}")
-        return f"Error: unable to make move {move}"
-
 def create_graph(llm_board, llm_white, llm_black):
     llm_board = ChatOpenAI(model=llm_board)
     llm_white = ChatOpenAI(model=llm_white)
@@ -146,7 +100,50 @@ def black_node(state: AgentState):
     return {
         "black_moves": response.content
     }
-    
+
+@tool
+def get_legal_moves() -> Annotated[str, "A list of legal moves in UCI format"]:
+    """Returns a list of legal moves in UCI format. 
+       The input should always be an empty string, 
+       and this function will always return legal moves in UCI format."""
+    try:
+        return "Possible moves are: " + ",".join(
+            [str(move) for move in board.legal_moves]
+        )
+    except Exception as e:
+        print(f"An error occurred in get_legal_moves: {e}")
+        return "Error: unable to get legal moves"
+
+@tool
+def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "Result of the move."]:
+    """Makes a move. 
+       The input should always be a move in UCI format, 
+       and this function will always return the result of the move in UCI format."""
+    try:
+        move = chess.Move.from_uci(move)
+        board.push_uci(str(move))
+        piece = board.piece_at(move.to_square)
+        piece_symbol = piece.unicode_symbol()
+        piece_name = (
+            chess.piece_name(piece.piece_type).capitalize()
+            if piece_symbol.isupper()
+            else chess.piece_name(piece.piece_type)
+        )
+
+        board_svgs.append(chess.svg.board(
+            board,
+            arrows=[(move.from_square, move.to_square)],
+            fill={move.from_square: "gray"},
+            size=250
+        ))
+        
+        return f"Moved {piece_name} ({piece_symbol}) from "\
+               f"{chess.SQUARE_NAMES[move.from_square]} to "\
+               f"{chess.SQUARE_NAMES[move.to_square]}."
+    except Exception as e:
+        print(f"An error occurred in make_move: {e}")
+        return f"Error: unable to make move {move}"
+   
 def run_multi_agent(llm_board, llm_white, llm_black, max_moves):
     graph = create_graph(llm_board, llm_white, llm_black, max_moves)