Update api/rpmlimits.py

api/rpmlimits.py

@@ -1,15 +1,17 @@
-from slowapi import Limiter
 from slowapi.util import get_remote_address
-from slowapi.errors import RateLimitExceeded
-from fastapi import FastAPI, Request
-from starlette.responses import JSONResponse
+from slowapi import Limiter
+from fastapi import Request
 
-# Initialize the limiter
+# Initialize a limiter instance
 limiter = Limiter(key_func=get_remote_address)
 
-# Exception handler for rate limit exceeded
-async def rate_limit_exceeded_handler(request: Request, exc: RateLimitExceeded):
-    return JSONResponse(
-        status_code=429,
-        content={"error": {"message": "Rate limit exceeded. Please wait and try again.", "type": "rate_limit"}}
-    )
+def check_rate_limit(request: Request):
+    """
+    Function to check if a user has exceeded their rate limit. 
+    This is managed by slowapi's limiter.
+    """
+    if not limiter.is_allowed(request):
+        raise HTTPException(
+            status_code=429,
+            detail="Rate limit exceeded. Please try again later."
+        )