diff --git a/src/backend/langflow/routers/login.py b/src/backend/langflow/routers/login.py index 48cdd11dc..3cb71a82e 100644 --- a/src/backend/langflow/routers/login.py +++ b/src/backend/langflow/routers/login.py @@ -11,15 +11,15 @@ from langflow.auth.auth import ( from sqlalchemy.orm import Session from langflow.services.utils import get_session +from langflow.database.models.user import User router = APIRouter() -def create_user_token(user: str) -> dict: +def create_user_token(user: User) -> dict: access_token_expires = timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES) access_token = create_access_token( - # type: ignore data={"sub": user.username}, expires_delta=access_token_expires, ) @@ -32,7 +32,7 @@ async def login_to_get_access_token( form_data: OAuth2PasswordRequestForm = Depends(), db: Session = Depends(get_session) ): if user := authenticate_user(db, form_data.username, form_data.password): - return create_user_token(user) # type: ignore + return create_user_token(user) else: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, diff --git a/src/backend/langflow/routers/users.py b/src/backend/langflow/routers/users.py index 98809f530..bcf1a7075 100644 --- a/src/backend/langflow/routers/users.py +++ b/src/backend/langflow/routers/users.py @@ -18,12 +18,12 @@ def get_password_hash(password): @router.get("/user", response_model=UserListModel) -async def read_current_user(current_user: User = Depends(get_current_active_user)): +def read_current_user(current_user: User = Depends(get_current_active_user)): return current_user @router.get("/users", response_model=List[UserListModel]) -async def read_all_users( +def read_all_users( skip: int = 0, limit: int = 10, _: Session = Depends(get_current_active_user), @@ -36,7 +36,7 @@ async def read_all_users( @router.post("/user", response_model=User) -async def add_user( +def add_user( user: UserAddModel, _: Session = Depends(get_current_active_user), db: Session = Depends(get_session), @@ -60,7 +60,7 @@ async def add_user( # TODO: Remove - Just for testing purposes @router.post("/super_user", response_model=User) -async def add_super_user_to_testing_purposes(db: Session = Depends(get_session)): +def add_super_user_to_testing_purposes(db: Session = Depends(get_session)): new_user = User(username="superuser", password="12345", is_superuser=True) try: