Update main.py

main.py

@@ -46,11 +46,11 @@ ip_rate_limit_store = defaultdict(lambda: {"count": 0, "timestamp": time.time()}
 async def get_api_key(authorization: str = Header(None)) -> str:
     if authorization is None or not authorization.startswith('Bearer '):
         logger.warning("Invalid or missing authorization header.")
-        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(req: Request, api_key: str = Depends(get_api_key)):
@@ -61,8 +61,8 @@ async def rate_limiter(req: Request, api_key: str = Depends(get_api_key)):
         rate_limit_store[api_key] = {"count": 1, "timestamp": current_time}
     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 for API key')
+            logger.warning(f"Rate limit exceeded for API key: {api_key} | NiansuhAI")
+            raise HTTPException(status_code=429, detail='Rate limit exceeded for API key | NiansuhAI')
         rate_limit_store[api_key]["count"] += 1
 
     # Rate limiting per IP address
@@ -73,7 +73,7 @@ async def rate_limiter(req: Request, api_key: str = Depends(get_api_key)):
     else:
         if ip_rate_limit_store[client_ip]["count"] >= RATE_LIMIT:
             logger.warning(f"Rate limit exceeded for IP address: {client_ip}")
-            raise HTTPException(status_code=429, detail='Rate limit exceeded for IP address')
+            raise HTTPException(status_code=429, detail='Rate limit exceeded for IP address | NiansuhAI')
         ip_rate_limit_store[client_ip]["count"] += 1
 
 # Custom exception for model not working
@@ -225,13 +225,13 @@ class Blackbox:
     ) -> AsyncGenerator[Any, None]:
         model = cls.get_model(model)
         if model is None:
-            logger.error(f"Model {model} is not available.")
+            logger.error(f"Model {model} is not available. | NiansuhAI")
             raise ModelNotWorkingException(model)
 
         logger.info(f"Selected model: {model}")
 
         if not cls.working or model not in cls.models:
-            logger.error(f"Model {model} is not working or not supported.")
+            logger.error(f"Model {model} is not working or not supported. | NiansuhAI")
             raise ModelNotWorkingException(model)
         
         headers = {
@@ -243,13 +243,13 @@ class Blackbox:
             "pragma": "no-cache",
             "priority": "u=1, i",
             "referer": cls.model_referers.get(model, cls.url),
-            "sec-ch-ua": '"Chromium";v="129", "Not=A?Brand";v="8"',
+            "sec-ch-ua": '"Chromium";v="130", "Not=A?Brand";v="99"',
             "sec-ch-ua-mobile": "?0",
             "sec-ch-ua-platform": '"Linux"',
             "sec-fetch-dest": "empty",
             "sec-fetch-mode": "cors",
             "sec-fetch-site": "same-origin",
-            "user-agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36",
+            "user-agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36",
         }
 
         if model in cls.model_prefixes:
@@ -324,7 +324,7 @@ class Blackbox:
                                 yield ImageResponse(image_url, alt=messages[-1]['content'])
                             else:
                                 logger.error("Image URL not found in the response.")
-                                raise Exception("Image URL not found in the response")
+                                raise Exception("Image URL not found in the response | NiansuhAI")
                         else:
                             full_response = ""
                             search_results_json = ""
@@ -360,11 +360,11 @@ class Blackbox:
             except ClientError as ce:
                 logger.error(f"Client error occurred: {ce}. Retrying attempt {attempt + 1}/{retry_attempts}")
                 if attempt == retry_attempts - 1:
-                    raise HTTPException(status_code=502, detail="Error communicating with the external API.")
+                    raise HTTPException(status_code=502, detail="Error communicating with the external API. | NiansuhAI")
             except asyncio.TimeoutError:
                 logger.error(f"Request timed out. Retrying attempt {attempt + 1}/{retry_attempts}")
                 if attempt == retry_attempts - 1:
-                    raise HTTPException(status_code=504, detail="External API request timed out.")
+                    raise HTTPException(status_code=504, detail="External API request timed out. | NiansuhAI")
             except Exception as e:
                 logger.error(f"Unexpected error: {e}. Retrying attempt {attempt + 1}/{retry_attempts}")
                 if attempt == retry_attempts - 1:
@@ -440,7 +440,7 @@ async def chat_completions(request: ChatRequest, req: Request, api_key: str = De
         # Validate that the requested model is available
         if request.model not in Blackbox.models and request.model not in Blackbox.model_aliases:
             logger.warning(f"Attempt to use unavailable model: {request.model}")
-            raise HTTPException(status_code=400, detail="Requested model is not available.")
+            raise HTTPException(status_code=400, detail="Requested model is not available. | NiansuhAI")
 
         # Process the request with actual message content, but don't log it
         async_generator = Blackbox.create_async_generator(
@@ -525,9 +525,9 @@ async def get_models():
     return {"data": [{"id": model, "object": "model"} for model in Blackbox.models]}
 
 # Additional endpoints for better functionality
-@app.get("/v1/health", dependencies=[Depends(rate_limiter)])
-async def health_check(req: Request, api_key: str = Depends(get_api_key)):
-    logger.info(f"Health check requested by API key: {api_key}")
+@app.get("/v1/health")
+async def health_check(req: Request):
+    logger.info("Health check requested")
     return {"status": "ok"}
 
 @app.get("/v1/models/{model}/status")
@@ -585,9 +585,9 @@ class CompletionRequest(BaseModel):
     logit_bias: Optional[Dict[str, float]] = None
     user: Optional[str] = None
 
-@app.post("/v1/completions", dependencies=[Depends(rate_limiter)])
-async def completions(request: CompletionRequest, req: Request, api_key: str = Depends(get_api_key)):
-    logger.info(f"Received completion request from API key: {api_key} | Model: {request.model}")
+@app.post("/v1/completions")
+async def completions(request: CompletionRequest, req: Request):
+    logger.info(f"Received completion request | Model: {request.model}")
 
     try:
         # Validate that the requested model is available