Update app.py

app.py

@@ -15,7 +15,7 @@ import aiohttp
 import asyncio
 import json
 from agent import MagAgent  
-from token_bucket.core import TokenBucket
+from token_bucket import TokenBucket
 
 # (Keep Constants as is)
 # --- Constants ---