Check endpoint if it works

auth/authentication.py

@@ -8,7 +8,7 @@ SECRET_KEY = "llmbenchmark_tr" # your secret key
 ALGORITHM = "HS256"
 ACCESS_TOKEN_EXPIRE_MINUTES = 30
 
-oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="api/token")
 
 def create_access_token(data: dict):
     to_encode = data.copy()

requirements.txt

@@ -1,4 +1,5 @@
 fastapi
 uvicorn[standard]
 lm_eval
-python-jose
+python-jose
+python-multipart

svc/router.py

@@ -13,12 +13,12 @@ router = APIRouter()
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
 
-
-
 load_dotenv()
+
 @router.post("/token")
 async def login_for_access_token(form_data: OAuth2PasswordRequestForm = Depends()):
-    if os.getenv("HF_TOKEN") != form_data.password:
+    hf_token = os.getenv("HF_TOKEN")
+    if hf_token != form_data.password:
         raise HTTPException(status_code=400, detail="Incorrect username or password")
     access_token = create_access_token(data={"sub": form_data.username})
     return {"access_token": access_token, "token_type": "bearer"}
@@ -30,7 +30,7 @@ async def protected_route(username: str = Depends(get_current_user)):
 
 
 @router.post("/chat",  response_model=LMHarnessTaskResponse)
-def inference_model(request: LMHarnessTaskRequest = Depends(get_current_user)):
+def inference_model(request: LMHarnessTaskRequest, username: str = Depends(get_current_user)):
     try:
         results = evaluator.simple_evaluate(
             model=request.model,