Update agent.py

agent.py

@@ -327,8 +327,18 @@ class MagAgent:
         import chess
         import chess.engine
         name = "stockfish_analysis"
-        description = "Analyzes chess position using Stockfish"
-        inputs = {'fen': {'type': 'string'}}
+        description = "Analyzes chess position using Stockfish and returns best move"
+        inputs = {
+            "fen": {
+                "type": "string",
+                "description": "FEN string of the current chess position"
+            },
+            "time_limit": {
+                "type": "number",
+                "description": "Analysis time in seconds",
+                "nullable": True
+            }
+        }
         output_type = "string"
 
         def forward(self, fen: str) -> str: