Added random move opening option

chess-gpt-eval/main.py

@@ -313,7 +313,7 @@ def get_player_titles_and_time(
 
 
 used_openings = []
-def initialize_game_with_opening(
+def random_book_opening(
     game_state: str, board: chess.Board
 ) -> Tuple[str, chess.Board]:
     global used_openings
@@ -342,6 +342,26 @@ def initialize_game_with_opening(
     return game_state.rstrip(), board, len(tokens) // 2
 
 
+def add_random_moves(
+    game_state: str, board: chess.Board, num_moves: int = 20
+) -> Tuple[str, chess.Board, int]:
+    for i in range(num_moves * 2):  # Full moves to half moves
+        legal_moves = list(board.legal_moves)
+        if not legal_moves:
+            break
+
+        move = board.san(random.choice(legal_moves))
+        board.push(board.parse_san(move))
+
+        if board.turn == chess.BLACK:
+            game_state += f" {i//2 + 1}.{move}" if move_num_in_gamestate else f" .{move}"
+        else:
+            game_state += f" {move}"
+
+    game_state = game_state.strip()
+    return game_state, board, num_moves
+    
+
 # Return is (move_san, move_uci, attempts, is_resignation, is_illegal_move)
 def get_legal_move(
     player: Player,
@@ -428,7 +448,9 @@ def play_game(
     player_one: Player,
     player_two: Player,
     max_games: int = 10,
-    random_opening_seed: bool = False,
+    book_opening: bool = False,
+    random_opening: bool = False,
+    random_opening_moves: int = 20,
 ):
     for z in range(max_games):
         print(f"\nGame {z} of {max_games}\n")
@@ -437,8 +459,10 @@ def play_game(
             game_state = f.read()
         board = chess.Board()
 
-        if random_opening_seed:
-            game_state, board, opening_moves = initialize_game_with_opening(game_state, board)
+        if book_opening:
+            game_state, board, opening_moves = random_book_opening(game_state, board)
+        elif random_opening:
+            game_state, board, opening_moves = add_random_moves(game_state, board, random_opening_moves)
         else:
             opening_moves = 0
         player_one_illegal_moves = 0
@@ -557,10 +581,13 @@ recording_file = "logs/determine.csv" # default recording file. Because we are u
 player_ones = ["50M/ckpt_9120050b.pt"]
 player_two_recording_name = "lc0_sweep" #"stockfish_sweep"
 move_num_in_gamestate = False
+book_opening = True
+random_opening = False
+random_opening_moves = 20
 if __name__ == "__main__":
     for nanogpt_player in player_ones:
         for i in range(1): #range(11):
-            num_games = 522 #265 #265 instead of 250 for duplicates (for lc0, stockfish doesn't need it)
+            num_games = 540
             # player_one = GPTPlayer(model="gpt-3.5-turbo-instruct")
             # player_one = LocalLlamaPlayer(model_name="meta-llama/Llama-2-7b-hf")
             # player_one = LocalLoraLlamaPlayer("meta-llama/Llama-2-7b-hf", "/workspace/axolotl/lora2-out")
@@ -580,6 +607,6 @@ if __name__ == "__main__":
             #print(f"\n\nSTARTING GAMES AGAINST STOCKFISH LEVEL {i}\n\n")
             print(f"\n\nSTARTING GAMES AGAINST LC0 LEVEL {i}\n\n")
 
-            play_game(player_one, player_two, num_games, random_opening_seed=True)
+            play_game(player_one, player_two, num_games, book_opening=book_opening, random_opening=random_opening,random_opening_moves=random_opening_moves)
             
 print("\n\n\n********\nDONE!\n********\n\n\n")