feat: Add engine parameter to Server class and update depth setting logic

Author: miskibin

draughts/server/server.py

@@ -32,8 +32,10 @@ def __init__(
         get_best_move_method: Callable = lambda board: np.random.choice(
             list(board.legal_moves)
         ),
+        engine=None,
     ):
         self.get_best_move_method = get_best_move_method
+        self.engine = engine
         self.board = board
         self.engine_depth = 6
         self.play_with_computer = False
@@ -79,6 +81,8 @@ async def load_fen(self, request: Request) -> PositionResponse:
     def set_depth(self, depth: int) -> dict:
         depth = max(1, min(10, int(depth)))
         self.engine_depth = depth
+        if self.engine is not None and hasattr(self.engine, 'depth'):
+            self.engine.depth = depth
         return {"depth": self.engine_depth}
 
     def set_play_mode(self, mode: str) -> dict:
@@ -170,5 +174,5 @@ def run(self, **kwargs):
 
     engine = AlphaBetaEngine(depth=6)
     board = get_board("standard")
-    server = Server(board=board, get_best_move_method=engine.get_best_move)
+    server = Server(board=board, get_best_move_method=engine.get_best_move, engine=engine)
     server.run()

tools/depth_benchmark.py

@@ -74,6 +74,7 @@ def main():
     print()
     print(f"{'Depth':>6} | {'Avg Time':>12} | {'Avg Nodes':>12} | {'Total Time':>12}")
     print("-" * 70)
+    sys.stdout.flush()
 
     results = []
 
@@ -94,6 +95,7 @@ def main():
             total_str = f"{total_time/60:.1f} min"
 
         print(f"{depth:>6} | {time_str:>12} | {result['avg_nodes']:>12,} | {total_str:>12}")
+        sys.stdout.flush()
 
         # Stop if taking too long (> 30s per move on average)
         if avg_time > 30000: