diff --git a/backend/open_webui/models/auths.py b/backend/open_webui/models/auths.py index fe14984..52d8ebf 100644 --- a/backend/open_webui/models/auths.py +++ b/backend/open_webui/models/auths.py @@ -203,21 +203,6 @@ class AuthsTable: return False def update_user_profile(self, id: str, name: str) -> bool: - try: - with get_db() as db: - result = db.query(Auth).filter_by(id=id).update({ - "name": name - }) - db.commit() - if result == 1: - print(f"User profile updated successfully: id={id}, name={name}") - return True - else: - print(f"Failed to update user profile: id={id}, name={name}") - return False - except Exception as e: - print(f"Error updating user profile: {e}") - return False - + return Users.insert_new_user(id, name) Auths = AuthsTable() diff --git a/backend/open_webui/models/users.py b/backend/open_webui/models/users.py index 6052995..51a7d7d 100644 --- a/backend/open_webui/models/users.py +++ b/backend/open_webui/models/users.py @@ -329,6 +329,22 @@ class UsersTable: with get_db() as db: 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: + try: + with get_db() as db: + result = db.query(User).filter_by(id=id).update({ + "name": name + }) + db.commit() + if result == 1: + print(f"User profile updated successfully: id={id}, name={name}") + return True + else: + print(f"Failed to update user profile: id={id}, name={name}") + return False + except Exception as e: + print(f"Error updating user profile: {e}") + return False Users = UsersTable()