Update apis/chat_api.py

apis/chat_api.py

@@ -106,24 +106,21 @@ class ChatAPIApp:
         return self.available_models
 
     def extract_api_key(
-        credentials: HTTPAuthorizationCredentials = Depends(
-            HTTPBearer(auto_error=False)
-        ),
+        credentials: HTTPAuthorizationCredentials = Depends(HTTPBearer(auto_error=False)),
     ):
         api_key = None
         if credentials:
             api_key = credentials.credentials
+            if not api_key.startswith("hf_"):
+                logger.error("Invalid HF Token format!")
+                raise HTTPException(status_code=403, detail="Invalid API Token format.")
         else:
             api_key = os.getenv("HF_TOKEN")
-
-        if api_key:
-            if api_key.startswith("hf_"):
-                return api_key
-            else:
-                logger.warn(f"Invalid HF Token!")
-        else:
-            logger.warn("Not provide HF Token!")
-        return None
+            if not api_key:
+                logger.error("HF Token not provided in request or environment.")
+                raise HTTPException(status_code=403, detail="API token not provided.")
+    
+        return api_key
 
     class QueryRequest(BaseModel):
         input: str