scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-09-08 22:21:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2026-01-05 22:12:49 +04:00
parent b1d30673b6
commit ab7b97e21f
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -320,12 +320,14 @@ async def get_model_by_id(
@router.get("/model/profile/image")
async def get_model_profile_image(
def get_model_profile_image(
id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)
):
model = Models.get_model_by_id(id, db=db)
if model:
etag = f'"{model.updated_at}"' if model.updated_at else None
if model.meta.profile_image_url:
if model.meta.profile_image_url.startswith("http"):
return Response(
@ -339,12 +341,14 @@ async def get_model_profile_image(
image_buffer = io.BytesIO(image_data)
media_type = header.split(";")[0].lstrip("data:")
headers = {"Content-Disposition": "inline"}
if etag:
headers["ETag"] = etag
return StreamingResponse(
image_buffer,
media_type=media_type,
headers={
"Content-Disposition": "inline",
},
headers=headers,
)
except Exception as e:
pass