feat : log limits data before reseting

Tanzania-AI-Community · Jan 25, 2025 · 2b40201 · jurmy24 · Jan 27, 2025 · fredygerman
1 parent a1de7fd
commit 2b40201
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/app/scheduler.py b/app/scheduler.py
@@ -12,6 +12,12 @@
 
 async def reset_daily_limits():
     redis_client = get_redis_client()
+    # Note: Maybe before deleting the keys, we can send a report of the daily usage to the admin
+    # Log all the keys and their values before deleting them
+    keys = await redis_client.keys("rate_limit:*")
+    for key in keys:
+        value = await redis_client.get(key)
+        logger.info(f"Key: {key}, Value: {value}")
     keys = await redis_client.keys("rate_limit:*")
     for key in keys:
         await redis_client.delete(key)