rollback changes because wait tool will not work

app.py

@@ -6,7 +6,6 @@ import requests
 import inspect
 import pandas as pd
 from smolagents import (
-    tool,
     Tool,
     CodeAgent, 
     LiteLLMModel, 
@@ -40,18 +39,7 @@ CUSTOM_PROMPT = """
     When downloading files, try to download them in python code before visiting the website.
 """
 
-# rate_limiter = RateLimiter()
-
-@tool
-def wait(seconds: int = 60) -> None:
-    """
-    This tool is used to wait for a certain amount of time.
-    Use this tool if you encountered RPM(Requests per minute) or TMP(Tokens per minute) when calling the API.
-
-    Args:
-        seconds (int): How many seconds to wait. Default value is 60.
-    """
-    time.sleep(seconds)
+rate_limiter = RateLimiter()
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -69,10 +57,9 @@ class BasicAgent:
                 WikipediaTool(),
                 YoutubeTranscriptTool(),
                 AssemblyAIAudioTranscriptTool(),
-                wait,
             ], 
             model=model,
-            # step_callbacks=[rate_limiter.increment_and_sleep_if_needed],
+            step_callbacks=[rate_limiter.increment_and_sleep_if_needed],
             additional_authorized_imports=["pandas", "openpyxl", "requests"],
             instructions=CUSTOM_PROMPT
         )