Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
landerrosette
2025-12-26 18:27:18 +04:00
committed by GitHub
parent f5c07e317e
commit d29271184e
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/routers/models.py
+3 -2
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,
}, },
) )
backend/open_webui/routers/users.py
+3 -2
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