jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

use optional for user

Browse source
This commit is contained in:
Guangya Liu 2023-09-14 19:48:48 -04:00
commit 03a71b1fe8
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/services/database/models/user/crud.py
View file

@ -6,6 +6,7 @@ from langflow.services.database.models.user.user import User, UserUpdate
from langflow.services.utils import get_session from langflow.services.utils import get_session
from sqlalchemy.exc import IntegrityError from sqlalchemy.exc import IntegrityError
from sqlmodel import Session from sqlmodel import Session
from typing import Optional
from sqlalchemy.orm.attributes import flag_modified from sqlalchemy.orm.attributes import flag_modified
@ -19,7 +20,7 @@ def get_user_by_id(db: Session, id: UUID) -> Union[User, None]:
def update_user( def update_user(
user_db: Union[User, None], user: UserUpdate, db: Session = Depends(get_session) user_db: Optional[User], user: UserUpdate, db: Session = Depends(get_session)
) -> User: ) -> User:
if not user_db: if not user_db:
raise HTTPException(status_code=404, detail="User not found") raise HTTPException(status_code=404, detail="User not found")