Update app.py

app.py

@@ -15,7 +15,7 @@ import aiohttp
 import asyncio
 import json
 from agent import MagAgent  
-from token_bucket import Limiter
+from token_bucket import Limiter, MemoryStorage 
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -26,8 +26,9 @@ RATE_LIMIT = 15  # Requests per minute
 TOKEN_BUCKET_CAPACITY = RATE_LIMIT
 TOKEN_BUCKET_REFILL_RATE = RATE_LIMIT / 60.0  # Tokens per second
 
-# Initialize global token bucket
-token_bucket = Limiter(rate=TOKEN_BUCKET_REFILL_RATE, capacity=TOKEN_BUCKET_CAPACITY)
+# Initialize global token bucket with MemoryStorage
+storage = MemoryStorage()
+token_bucket = Limiter(rate=TOKEN_BUCKET_REFILL_RATE, capacity=TOKEN_BUCKET_CAPACITY, storage=storage)
 
 async def fetch_questions(session: aiohttp.ClientSession, questions_url: str) -> list:
     """Fetch questions asynchronously."""
@@ -72,7 +73,8 @@ async def process_question(agent, question_text: str, task_id: str, results_log:
         await token_bucket.wait(1)
         submitted_answer = await agent(question_text)
         results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
-        return {"task_id": task_id, "submitted_answer": submitted_answer}
+        return {"task_id"
+: task_id, "submitted_answer": submitted_answer}
     except aiohttp.ClientResponseError as e:
         if e.status == 429:
             print(f"Rate limit hit for task {task_id}. Retrying after delay...")