ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-01-02 14:45:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix: extract correct MIME type from base64 profile images (#20171)

Browse source
This commit is contained in:
landerrosette 2025-12-26 22:27:18 +08:00 committed by GitHub
parent f5c07e317e
commit d29271184e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
backend/open_webui/routers/models.py
View file

@ -306,12 +306,13 @@ async def get_model_profile_image(id: str, user=Depends(get_verified_user)):
header, base64_data = model.meta.profile_image_url.split(",", 1) header, base64_data = model.meta.profile_image_url.split(",", 1)
image_data = base64.b64decode(base64_data) image_data = base64.b64decode(base64_data)
image_buffer = io.BytesIO(image_data) image_buffer = io.BytesIO(image_data)
media_type = header.split(";")[0].lstrip("data:")
return StreamingResponse( return StreamingResponse(
image_buffer, image_buffer,
media_type="image/png", media_type=media_type,
headers={ headers={
"Content-Disposition": "inline; filename=image.png", "Content-Disposition": "inline",
**cache_headers, **cache_headers,
}, },
) )

5
backend/open_webui/routers/users.py
View file

@ -461,11 +461,12 @@ async def get_user_profile_image_by_id(user_id: str, user=Depends(get_verified_u
header, base64_data = user.profile_image_url.split(",", 1) header, base64_data = user.profile_image_url.split(",", 1)
image_data = base64.b64decode(base64_data) image_data = base64.b64decode(base64_data)
image_buffer = io.BytesIO(image_data) image_buffer = io.BytesIO(image_data)
media_type = header.split(";")[0].lstrip("data:")
return StreamingResponse( return StreamingResponse(
image_buffer, image_buffer,
media_type="image/png", media_type=media_type,
headers={"Content-Disposition": "inline; filename=image.png"}, headers={"Content-Disposition": "inline"},
) )
except Exception as e: except Exception as e:
pass pass