Просмотр исходного кода

Merge pull request #18572 from attilaolah/issue-18542

feat: OAUTH_ROLES_SEPARATOR env var
Tim Baek 3 месяцев назад
Родитель
Сommit
f83dc60666
1 измененных файлов с 5 добавлено и 2 удалено
  1. 5 2
      backend/open_webui/config.py

+ 5 - 2
backend/open_webui/config.py

@@ -576,19 +576,22 @@ OAUTH_ROLES_CLAIM = PersistentConfig(
     os.environ.get("OAUTH_ROLES_CLAIM", "roles"),
 )
 
+SEP = os.environ.get("OAUTH_ROLES_SEPARATOR", ",")
+
 OAUTH_ALLOWED_ROLES = PersistentConfig(
     "OAUTH_ALLOWED_ROLES",
     "oauth.allowed_roles",
     [
         role.strip()
-        for role in os.environ.get("OAUTH_ALLOWED_ROLES", "user,admin").split(",")
+        for role in os.environ.get("OAUTH_ALLOWED_ROLES", f"user{SEP}admin").split(SEP)
+        if role
     ],
 )
 
 OAUTH_ADMIN_ROLES = PersistentConfig(
     "OAUTH_ADMIN_ROLES",
     "oauth.admin_roles",
-    [role.strip() for role in os.environ.get("OAUTH_ADMIN_ROLES", "admin").split(",")],
+    [role.strip() for role in os.environ.get("OAUTH_ADMIN_ROLES", "admin").split(SEP) if role],
 )
 
 OAUTH_ALLOWED_DOMAINS = PersistentConfig(