replace custom rate limiter with wait tool

app.py

@@ -1,10 +1,12 @@
 import os
+import time
 import gradio as gr
 import litellm
 import requests
 import inspect
 import pandas as pd
 from smolagents import (
+    tool,
     Tool,
     CodeAgent, 
     LiteLLMModel, 
@@ -38,7 +40,18 @@ CUSTOM_PROMPT = """
     When downloading files, try to download them in python code before visiting the website.
 """
 
-rate_limiter = RateLimiter()
+# 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)
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -56,9 +69,10 @@ 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
         )