From 03a71b1fe8cae5442b0fc18c6e0979b573bce5d3 Mon Sep 17 00:00:00 2001 From: Guangya Liu Date: Thu, 14 Sep 2023 19:48:48 -0400 Subject: [PATCH] use optional for user --- src/backend/langflow/services/database/models/user/crud.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/services/database/models/user/crud.py b/src/backend/langflow/services/database/models/user/crud.py index fc17d09be..f7f5958fe 100644 --- a/src/backend/langflow/services/database/models/user/crud.py +++ b/src/backend/langflow/services/database/models/user/crud.py @@ -6,6 +6,7 @@ from langflow.services.database.models.user.user import User, UserUpdate from langflow.services.utils import get_session from sqlalchemy.exc import IntegrityError from sqlmodel import Session +from typing import Optional 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( - user_db: Union[User, None], user: UserUpdate, db: Session = Depends(get_session) + user_db: Optional[User], user: UserUpdate, db: Session = Depends(get_session) ) -> User: if not user_db: raise HTTPException(status_code=404, detail="User not found")