浏览代码

add missing env mapping

Willnow, Patrick 6 月之前
父节点
当前提交
1c5b6987e2
共有 1 个文件被更改,包括 5 次插入1 次删除
  1. 5 1
      backend/open_webui/apps/webui/main.py

+ 5 - 1
backend/open_webui/apps/webui/main.py

@@ -37,12 +37,14 @@ from open_webui.config import (
     OAUTH_EMAIL_CLAIM,
     OAUTH_EMAIL_CLAIM,
     OAUTH_PICTURE_CLAIM,
     OAUTH_PICTURE_CLAIM,
     OAUTH_USERNAME_CLAIM,
     OAUTH_USERNAME_CLAIM,
+    OAUTH_ALLOWED_ROLES,
+    OAUTH_ADMIN_ROLES,
     SHOW_ADMIN_DETAILS,
     SHOW_ADMIN_DETAILS,
     USER_PERMISSIONS,
     USER_PERMISSIONS,
     WEBHOOK_URL,
     WEBHOOK_URL,
     WEBUI_AUTH,
     WEBUI_AUTH,
     WEBUI_BANNERS,
     WEBUI_BANNERS,
-    AppConfig,
+    AppConfig, OAUTH_ALLOWED_ROLES, OAUTH_ADMIN_ROLES,
 )
 )
 from open_webui.env import (
 from open_webui.env import (
     WEBUI_AUTH_TRUSTED_EMAIL_HEADER,
     WEBUI_AUTH_TRUSTED_EMAIL_HEADER,
@@ -97,6 +99,8 @@ app.state.config.OAUTH_EMAIL_CLAIM = OAUTH_EMAIL_CLAIM
 
 
 app.state.config.ENABLE_OAUTH_ROLE_MANAGEMENT = ENABLE_OAUTH_ROLE_MANAGEMENT
 app.state.config.ENABLE_OAUTH_ROLE_MANAGEMENT = ENABLE_OAUTH_ROLE_MANAGEMENT
 app.state.config.OAUTH_ROLES_CLAIM = OAUTH_ROLES_CLAIM
 app.state.config.OAUTH_ROLES_CLAIM = OAUTH_ROLES_CLAIM
+app.state.config.OAUTH_ALLOWED_ROLES = OAUTH_ALLOWED_ROLES
+app.state.config.OAUTH_ADMIN_ROLES = OAUTH_ADMIN_ROLES
 
 
 app.state.MODELS = {}
 app.state.MODELS = {}
 app.state.TOOLS = {}
 app.state.TOOLS = {}