Update agent.py

agent.py

@@ -324,8 +324,6 @@ class MagAgent:
 
     
     class ChessEngineTool(Tool):
-        import chess
-        import chess.engine
         name = "stockfish_analysis"
         description = "Analyzes chess position using Stockfish and returns best move"
         inputs = {
@@ -343,9 +341,11 @@ class MagAgent:
 
         def forward(self, fen: str) -> str:
             try:
+                import chess
+                import chess.engine
                 board = chess.Board(fen)
                 engine = chess.engine.SimpleEngine.popen_uci("stockfish")
-                result = engine.play(board, chess.engine.Limit(time=2.0))
+                result = engine.play(board, chess.engine.Limit(time=time_limit))
                 engine.quit()
                 return board.san(result.move)
             except Exception as e: