Update main.py

main.py

@@ -38,11 +38,11 @@ rate_limit_store = defaultdict(lambda: {"count": 0, "timestamp": time.time()})
 async def get_api_key(authorization: str = Header(...)) -> str:
     if not authorization.startswith('Bearer '):
         logger.warning("Invalid authorization header format.")
-        raise HTTPException(status_code=401, detail='Invalid authorization header format')
+        raise HTTPException(status_code=401, detail='Invalid authorization header format | NiansuhAI')
     api_key = authorization[7:]
     if api_key not in API_KEYS:
         logger.warning(f"Invalid API key attempted: {api_key}")
-        raise HTTPException(status_code=401, detail='Invalid API key')
+        raise HTTPException(status_code=401, detail='Invalid API key | NiansuhAI')
     return api_key
 
 async def rate_limiter(api_key: str = Depends(get_api_key)):
@@ -53,7 +53,7 @@ async def rate_limiter(api_key: str = Depends(get_api_key)):
     else:
         if rate_limit_store[api_key]["count"] >= RATE_LIMIT:
             logger.warning(f"Rate limit exceeded for API key: {api_key}")
-            raise HTTPException(status_code=429, detail='Rate limit exceeded')
+            raise HTTPException(status_code=429, detail='Rate limit exceeded | NiansuhAI')
         rate_limit_store[api_key]["count"] += 1
 
 # Custom exception for model not working