OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-20 05:07:59 +00:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Merge pull request #9629 from rragundez/handle-no-picture

Browse Source 
Handle no picture when retrieving from Microsoft EntraID
This commit is contained in:
Timothy Jaeryang Baek 2025-02-07 22:52:35 -08:00 committed by GitHub
commit dee07a6260
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/open_webui/utils/oauth.py
View File

@ -284,6 +284,7 @@ class OAuthManager:
} }
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.get(picture_url, **get_kwargs) as resp: async with session.get(picture_url, **get_kwargs) as resp:
if resp.ok:
picture = await resp.read() picture = await resp.read()
base64_encoded_picture = base64.b64encode( base64_encoded_picture = base64.b64encode(
picture picture
@ -293,6 +294,8 @@ class OAuthManager:
# assume JPG, browsers are tolerant enough of image formats # assume JPG, browsers are tolerant enough of image formats
guessed_mime_type = "image/jpeg" guessed_mime_type = "image/jpeg"
picture_url = f"data:{guessed_mime_type};base64,{base64_encoded_picture}" picture_url = f"data:{guessed_mime_type};base64,{base64_encoded_picture}"
else:
picture_url = "/user.png"
except Exception as e: except Exception as e:
log.error( log.error(
f"Error downloading profile image '{picture_url}': {e}" f"Error downloading profile image '{picture_url}': {e}"