Update multi_agent.py

multi_agent.py

@@ -12,8 +12,6 @@ def get_legal_moves() -> Annotated[str, "A list of legal moves in UCI format"]:
         [str(move) for move in board.legal_moves]
     )
 
-    return moves
-
 def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "Result of the move."]:
     move = chess.Move.from_uci(move)
     board.push_uci(str(move))
@@ -40,11 +38,6 @@ def make_move(move: Annotated[str, "A move in UCI format."]) -> Annotated[str, "
            f"{chess.SQUARE_NAMES[move.to_square]}."
 
 def check_made_move(msg):
-    print("###")
-    print(msg)
-    print(str(len(board_svgs)))
-    print("###")
-    
     global made_move
     
     if made_move:
@@ -81,7 +74,8 @@ def run_multi_agent(llm_white, llm_black, num_moves):
         name="Player White",
         system_message="You are a chess Grandmaster and you play as white. "
         "First call get_legal_moves(), to get a list of legal moves. "
-        "If a legal move is possible, call make_move(move) to make a move, else reply 'TERMINATE'.",
+        "Then call make_move(move) to make a move. "
+        "Analyze the move in 3 bullet points. Respond in format **Analysis:** move in UCI format, unordered list.",
         llm_config=llm_config_white,
     )
     
@@ -89,7 +83,8 @@ def run_multi_agent(llm_white, llm_black, num_moves):
         name="Player Black",
         system_message="You are a chess Grandmaster and you play as black. "
         "First call get_legal_moves(), to get a list of legal moves. "
-        "If a legal move is possible, call make_move(move) to make a move, else reply 'TERMINATE'.",
+        "Then call make_move(move) to make a move. "
+        "Analyze the move in 3 bullet points. Respond in format **Analysis:** move in UCI format, unordered list.",
         llm_config=llm_config_black,
     )
     
@@ -169,8 +164,8 @@ def run_multi_agent(llm_white, llm_black, num_moves):
         if num_moves % 2 == 0 and num_move == num_moves + 1:
             break
 
-    print("===")
-    print(result)
-    print("===")
+    #print("===")
+    #print(result)
+    #print("===")
     
-    return result
+    return result