diff --git a/backend/open_webui/models/auths.py b/backend/open_webui/models/auths.py index 52d8ebf..79bfa01 100644 --- a/backend/open_webui/models/auths.py +++ b/backend/open_webui/models/auths.py @@ -202,7 +202,7 @@ class AuthsTable: except Exception: return False - def update_user_profile(self, id: str, name: str) -> bool: - return Users.insert_new_user(id, name) + def update_user_profile(self, id: str, name: str, profile_image_url: str, role: str) -> bool: + return Users.insert_new_user(id, name, profile_image_url, role) Auths = AuthsTable() diff --git a/backend/open_webui/models/users.py b/backend/open_webui/models/users.py index 51a7d7d..65e336b 100644 --- a/backend/open_webui/models/users.py +++ b/backend/open_webui/models/users.py @@ -330,11 +330,13 @@ class UsersTable: users = db.query(User).filter(User.id.in_(user_ids)).all() return [user.id for user in users] - def update_user_profile(self, id: str, name: str) -> bool: + def update_user_profile(self, id: str, name: str, profile_image_url: str, role: str) -> bool: try: with get_db() as db: result = db.query(User).filter_by(id=id).update({ - "name": name + "name": name, + "profile_image_url": profile_image_url, + "role": role }) db.commit() if result == 1: diff --git a/backend/open_webui/utils/oauth.py b/backend/open_webui/utils/oauth.py index ff94af3..e9a6257 100644 --- a/backend/open_webui/utils/oauth.py +++ b/backend/open_webui/utils/oauth.py @@ -314,8 +314,8 @@ class OAuthManager: updated = Auths.update_user_profile( user.id, user_data.get("name", ""), - #user_data.get("profile_image_url", ""), - #user_data.get("role", self.get_user_role(None, user_data)) + user_data.get("profile_image_url", ""), + user_data.get("role", self.get_user_role(None, user_data)) ) if updated: return {"message": "Profile updated successfully."}