ecemumutlu commited on
Commit
7ec6be5
·
1 Parent(s): e2ebff5

Fix incorrect oauth request form

Browse files
Files changed (1) hide show
  1. svc/router.py +4 -3
svc/router.py CHANGED
@@ -1,8 +1,9 @@
1
  from fastapi import APIRouter, HTTPException, Depends
2
  import logging
3
 
4
- from lm_eval import evaluator, utils
5
- from svc.schemas import LMHarnessTaskRequest, LMHarnessTaskResponse, OAuth2PasswordRequestForm, User
 
6
  from auth.authentication import get_current_user, create_access_token
7
  from dotenv import load_dotenv
8
  import os
@@ -17,7 +18,7 @@ logger = logging.getLogger(__name__)
17
  load_dotenv()
18
  @router.post("/token")
19
  async def login_for_access_token(form_data: OAuth2PasswordRequestForm = Depends()):
20
- if os.getenv("HF_TOKEN") != form_data.hf_token:
21
  raise HTTPException(status_code=400, detail="Incorrect username or password")
22
  access_token = create_access_token(data={"sub": form_data.username})
23
  return {"access_token": access_token, "token_type": "bearer"}
 
1
  from fastapi import APIRouter, HTTPException, Depends
2
  import logging
3
 
4
+ from lm_eval import evaluator
5
+ from svc.schemas import LMHarnessTaskRequest, LMHarnessTaskResponse
6
+ from fastapi.security import OAuth2PasswordRequestForm
7
  from auth.authentication import get_current_user, create_access_token
8
  from dotenv import load_dotenv
9
  import os
 
18
  load_dotenv()
19
  @router.post("/token")
20
  async def login_for_access_token(form_data: OAuth2PasswordRequestForm = Depends()):
21
+ if os.getenv("HF_TOKEN") != form_data.password:
22
  raise HTTPException(status_code=400, detail="Incorrect username or password")
23
  access_token = create_access_token(data={"sub": form_data.username})
24
  return {"access_token": access_token, "token_type": "bearer"}