UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-23 21:53:37 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Check is response is OK from retrieve the picture if not then default

Browse Source
This commit is contained in:
Rodrigo Agundez 2025-02-08 09:49:16 +09:00
parent 14398ab628
commit 546ef6ab42
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

@ -261,6 +261,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
@ -270,6 +271,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}"