From 690d0708052ba03a19500aebb4213b12bdbb0aa9 Mon Sep 17 00:00:00 2001 From: Willy Douhard Date: Tue, 16 Apr 2024 13:08:28 +0200 Subject: [PATCH] put back user creation --- backend/chainlit/auth.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/backend/chainlit/auth.py b/backend/chainlit/auth.py index 0a91ddae2c..981d33abe4 100644 --- a/backend/chainlit/auth.py +++ b/backend/chainlit/auth.py @@ -71,8 +71,17 @@ async def authenticate_user(token: str = Depends(reuseable_oauth)): user = User(**dict) except Exception as e: raise HTTPException(status_code=401, detail="Invalid authentication token") - - return user + if data_layer := get_data_layer(): + try: + persisted_user = await data_layer.get_user(user.identifier) + if persisted_user == None: + persisted_user = await data_layer.create_user(user) + except Exception as e: + return user + + return persisted_user + else: + return user async def get_current_user(token: str = Depends(reuseable_oauth)):