Update multi_agent.py

multi_agent.py

@@ -47,7 +47,7 @@ def create_agent(llm: ChatOpenAI, tools: list, system_prompt: str):
 
 def agent_node(state, agent, name):
     try:
-        print("### node=" + name + " ###")
+        print(f"### {name} ###")
         result = agent.invoke(state)
         return {"messages": [HumanMessage(content=result["output"], name=name)]}
     except Exception as e:
@@ -60,7 +60,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")
+        #print("### get_legal_moves")
         global legal_moves
         legal_moves = ",".join([str(move) for move in board.legal_moves])
         return legal_moves
@@ -74,7 +74,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")
+        #print("### make_move")
         move = chess.Move.from_uci(move)
         board.push_uci(str(move))
 
@@ -209,7 +209,7 @@ def should_continue(state):
         print("### checkmate")
         return END # checkmate or stalemate
 
-    print("### chess_board_proxy")
+    print("### chess_board_proxy ###")
     return "chess_board_proxy"
 
 def initialize():
@@ -224,11 +224,11 @@ def initialize():
     legal_moves = ""
 
 def run_multi_agent(moves_num):
+    initialize()
+    
     global num_moves
     num_moves = moves_num
 
-    initialize()
-
     graph = create_graph()
 
     result = ""