Update multi_agent.py

multi_agent.py

@@ -15,8 +15,6 @@ from langchain_openai import ChatOpenAI
 
 from langgraph.graph import StateGraph, END
 
-num_moves = 0
-
 class AgentState(TypedDict):
     messages: Annotated[Sequence[BaseMessage], operator.add]
     next: str
@@ -51,6 +49,8 @@ def agent_node(state, agent, name):
         
 board = None
 board_svgs = None
+num_moves = 0
+move_num = 0
 
 @tool
 def get_legal_moves() -> Annotated[str, "A list of legal moves in UCI format"]:
@@ -71,6 +71,12 @@ def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "
        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:
+        global move_num
+    
+        move_num += 1
+    
+        print("## move_num=" + str(move_num))
+    
         move = chess.Move.from_uci(move)
         board.push_uci(str(move))
 
@@ -171,7 +177,7 @@ def create_graph(llm_board, llm_white, llm_black):
                                      "You are a chess Grandmaster and you play as white. "
                                      "1. First call get_legal_moves(), to get a list of legal moves. "
                                      "2. If no legal move is possible, reply FINISH. "
-                                     "3. Else choose a move returned in step one and call make_move(move). "
+                                     "3. Else choose a move returned in step one and call make_move(move). IMPORTANT: Do NOT make illegal moves. "
                                      "4. Analyze the move in 3 bullet points. Respond in format **Analysis:** move in UCI format (piece emoji), unordered list.")
     player_white_node = functools.partial(agent_node, agent=player_white_agent, name="player_white")
 
@@ -179,7 +185,7 @@ def create_graph(llm_board, llm_white, llm_black):
                                      "You are a chess Grandmaster and you play as black. "
                                      "1. First call get_legal_moves(), to get a list of legal moves. "
                                      "2. If no legal move is possible, reply FINISH. "
-                                     "3. Else choose a move returned in step one and call make_move(move). "
+                                     "3. Else choose a move returned in step one and call make_move(move). IMPORTANT: Do NOT make illegal moves. "
                                      "4. Analyze the move in 3 bullet points. Respond in format **Analysis:** move in UCI format (piece emoji), unordered list.")
     player_black_node = functools.partial(agent_node, agent=player_black_agent, name="player_black")
     
@@ -204,6 +210,8 @@ def run_multi_agent(llm_board, llm_white, llm_black, moves_num):
 
     num_moves = moves_num
 
+    print("## num_moves=" + num_moves)
+    
     graph = create_graph(llm_board, llm_white, llm_black)
 
     result = ""