Update agent.py

agent.py

@@ -338,20 +338,19 @@ class MagAgent:
             }
         }
         output_type = "string"
-
-        def forward(self, fen: str) -> str:
+    
+        def forward(self, fen: str, time_limit: float = 0.1) -> str:  # Add time_limit parameter
+            """Analyzes chess position using Stockfish engine"""
             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=time_limit))
+                result = engine.play(board, chess.engine.Limit(time=time_limit))  # Use parameter
                 engine.quit()
                 return board.san(result.move)
             except Exception as e:
-                return f"Engine error: {str(e)}"
-        
-    
+                return f"Engine error: {str(e)}"    
             
     class PythonCodeReaderTool(Tool):
         name = "read_python_code"