UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-07 00:59:52 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix logic

Browse Source
This commit is contained in:
Patrick Willnow 2024-10-04 10:14:20 +02:00
parent 79b9c8a677
commit 6ddd8c7241
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/main.py
View File

@ -2252,13 +2252,15 @@ async def oauth_callback(provider: str, request: Request, response: Response):
role = "admin" role = "admin"
elif webui_app.state.config.ENABLE_OAUTH_ROLE_MAPPING: elif webui_app.state.config.ENABLE_OAUTH_ROLE_MAPPING:
oauth_claim = webui_app.state.config.OAUTH_ROLES_CLAIM oauth_claim = webui_app.state.config.OAUTH_ROLES_CLAIM
oauth_roles = user_data.get(oauth_claim) # Works for simple claims with no nesting oauth_roles = user_data.get(oauth_claim) # Works for simple claims with no nesting
if oauth_roles and "." in oauth_claim: if oauth_roles and "." in oauth_claim:
# Implementation to handle nested claims of arbitrary depth # Implementation to handle nested claims of arbitrary depth
nested_claims = oauth_claim.split(".") nested_claims = oauth_claim.split(".")
claim_data = user_data.get(nested_claims[0]) claim_data = user_data
for nested_claim in nested_claims[1:]: for nested_claim in nested_claims:
claim_data = claim_data.get(nested_claim) claim_data = claim_data.get(nested_claim)
if claim_data is None:
break
oauth_roles = claim_data oauth_roles = claim_data
log.info(f"User {user.name} has OAuth roles: {oauth_roles}") log.info(f"User {user.name} has OAuth roles: {oauth_roles}")