Update api/routes.py

api/routes.py

@@ -1,22 +1,21 @@
 import json
 from fastapi import APIRouter, Depends, HTTPException, Request, Response
 from fastapi.responses import StreamingResponse
+from slowapi import Limiter
 from slowapi.util import get_remote_address
 from slowapi.errors import RateLimitExceeded
-from slowapi import Limiter
 from api.auth import verify_app_secret
 from api.config import ALLOWED_MODELS
 from api.models import ChatRequest
 from api.utils import process_non_streaming_response, process_streaming_response
 from api.logger import setup_logger
+from api.rpmlimits import check_rate_limit
 
 # Set up logging
 logger = setup_logger(__name__)
 
-# Initialize the router
+# Initialize the router and limiter
 router = APIRouter()
-
-# Set up limiter for rate limiting
 limiter = Limiter(key_func=get_remote_address)
 
 # Rate limit exception handler