From 2be92c028d324f56a16d4a7e6fc3e4dc82db32b1 Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Sat, 1 Mar 2025 22:33:20 +0100
Subject: [PATCH] mod

---
 backend/open_webui/models/auths.py | 4 ++--
 backend/open_webui/models/users.py | 5 +++--
 backend/open_webui/utils/oauth.py  | 3 ++-
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/backend/open_webui/models/auths.py b/backend/open_webui/models/auths.py
index cceb4b1..962c57c 100644
--- a/backend/open_webui/models/auths.py
+++ b/backend/open_webui/models/auths.py
@@ -202,8 +202,8 @@ class AuthsTable:
         except Exception:
             return False
         
-    def update_user_profile(self, id: str, name: str, profile_image_url: str, role: str) -> bool: 
-        result = Users.update_user_profile(id, name, profile_image_url, role)
+    def update_user_profile(self, id: str, name: str, profile_image_url: str, role: str, permissions: dict) -> bool: 
+        result = Users.update_user_profile(id, name, profile_image_url, role, permissions)
         return True if result == 1 else False
 
 Auths = AuthsTable()
diff --git a/backend/open_webui/models/users.py b/backend/open_webui/models/users.py
index 38a695a..62193e8 100644
--- a/backend/open_webui/models/users.py
+++ b/backend/open_webui/models/users.py
@@ -335,13 +335,14 @@ 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, profile_image_url: str, role: str) -> bool:
+    def update_user_profile(self, id: str, name: str, profile_image_url: str, role: str, permissions: dict) -> bool:
         try:
             with get_db() as db:
                 result = db.query(User).filter_by(id=id).update({
                     "name": name,
                     "profile_image_url": profile_image_url,
-                    "role": role
+                    "role": role,
+                    "permissions": permissions
                 })
                 db.commit()
                 return result
diff --git a/backend/open_webui/utils/oauth.py b/backend/open_webui/utils/oauth.py
index 49e8aa4..c91bd61 100644
--- a/backend/open_webui/utils/oauth.py
+++ b/backend/open_webui/utils/oauth.py
@@ -312,7 +312,8 @@ class OAuthManager:
                 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("role", self.get_user_role(None, user_data)),
+                user_data.get("permissions", None)
             )
 
         if not user: