Update agent.py

agent.py

@@ -74,6 +74,12 @@ class ChessboardToFENOnlineTool(Tool):
         except Exception as e:
             return f"API Call Failed: {str(e)}"
 
+from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception
+import requests
+
+def is_429_error(exception):
+    return isinstance(exception, requests.exceptions.HTTPError) and exception.response.status_code == 429
+
 class VisitWebpageTool(Tool):
     name = "visit_webpage"
     description = "Visits a webpage at the given url and reads its content as a markdown string. Use this to browse webpages."
@@ -190,6 +196,11 @@ class RetryDuckDuckGoSearchTool(DuckDuckGoSearchTool):
     def forward(self, query: str) -> str:
         return super().forward(query)
 
+
+##############################
+#   MAG Agent
+##############################
+        
 class MagAgent:
     def __init__(self, rate_limiter: Optional[Limiter] = None):
         """Initialize the MagAgent with search tools."""