Update multi_agent.py

multi_agent.py

@@ -6,8 +6,10 @@ from langchain_openai import ChatOpenAI
 from langgraph.graph import StateGraph, END
 from typing import TypedDict, Annotated, List
 
+board = None
+board_svgs = None
+
 class AgentState(TypedDict):
-    board: chess.Board
     player: str
     made_move: bool
     move_num: int
@@ -15,11 +17,11 @@ class AgentState(TypedDict):
     legal_moves: list[chess.Move]
     white_moves: Annotated[list[chess.Move], operator.add]
     black_moves: Annotated[list[chess.Move], operator.add]
-    board_svgs: Annotated[list[str], operator.add]
 
 def create_graph():
+    print("#### create_graph")
     graph = StateGraph(AgentState)
-
+  
     graph.add_node("chess_board", board_node)
     graph.add_node("player_white", white_node)
     graph.add_node("player_black", black_node)
@@ -44,6 +46,8 @@ def create_graph():
     return graph.compile()
 
 def should_continue_white(state):
+    print("#### should_continue_white")
+    print("#### state=" + str(state))
     print("## move_num=" + str(state["move_num"]))
     print("## max_moves=" + str(state["max_moves"]))
     if state["move_num"] > state["max_moves"]:
@@ -53,6 +57,8 @@ def should_continue_white(state):
     return "player_white"
 
 def should_continue_black(state):
+    print("#### should_continue_black")
+    print("#### state=" + str(state))
     print("## move_num=" + str(state["move_num"]))
     print("## max_moves=" + str(state["max_moves"]))
     if state["move_num"] > state["max_moves"]:
@@ -62,6 +68,8 @@ def should_continue_black(state):
     return "player_black"
     
 def board_node(state: AgentState):
+    print("#### board_node")
+    print("#### state=" + str(state))
     player = "player_black" if state["player"] == "player_white" else "player_white"
     print("## player=" + player)
     messages = [
@@ -76,6 +84,8 @@ def board_node(state: AgentState):
     }
     
 def white_node(state: AgentState):
+    print("#### white_node")
+    print("#### state=" + str(state))
     legal_moves = str(state["legal_moves"][-1])
     print("## legal_moves=" + legal_moves)
     messages = [
@@ -90,6 +100,8 @@ def white_node(state: AgentState):
     }
 
 def black_node(state: AgentState):
+    print("#### black_node")
+    print("#### state=" + str(state))
     legal_moves = str(state["legal_moves"][-1])
     print("## legal_moves=" + legal_moves)
     messages = [
@@ -109,6 +121,7 @@ def get_legal_moves() -> Annotated[str, "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:
+        print("#### get_legal_moves")
         return "Possible moves are: " + ",".join(
             [str(move) for move in board.legal_moves]
         )
@@ -122,6 +135,7 @@ 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:
+        print("#### make_move")
         move = chess.Move.from_uci(move)
         board.push_uci(str(move))
         piece = board.piece_at(move.to_square)
@@ -145,10 +159,19 @@ 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 initialize():
+    global board, board_svgs
+    board = chess.Board()
+    board_svgs = []
+    
 def run_multi_agent(max_moves):
+    initialize()
+    
     graph = create_graph()
 
+    print("#### Start")
+    
     for s in graph.stream({
         "board": chess.Board(),
         "move_num": 0,