reflect RateLimiter's changes

app.py

@@ -22,8 +22,6 @@ from youtube_transcript_tool import YoutubeTranscriptTool
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 rate_limiter = RateLimiter()
-litellm.success_callback = [rate_limiter.increment_successful_calls]
-litellm.failure_callback = [rate_limiter.increment_failed_calls]
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -45,6 +43,7 @@ class BasicAgent:
         )
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
+        limiter.sleep_if_needed()
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         answer = self.agent.run(question)
         print(f"Agent returning answer: {answer}")