From e396af3cc891a4271fbdb5703f12eaf9135ac35c Mon Sep 17 00:00:00 2001 From: Classic298 <27028174+Classic298@users.noreply.github.com> Date: Fri, 17 Apr 2026 04:28:19 +0200 Subject: [PATCH] perf: reuse request db session in get_model_profile_image (#23796) Pass the request-scoped AsyncSession into Models.get_model_by_id so the endpoint no longer opens a fresh DB session on every call, avoiding an extra connection acquisition per profile image request. Co-authored-by: Claude --- backend/open_webui/routers/models.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/routers/models.py b/backend/open_webui/routers/models.py index b7d31321ff..451622a219 100644 --- a/backend/open_webui/routers/models.py +++ b/backend/open_webui/routers/models.py @@ -384,8 +384,12 @@ async def get_model_by_id(id: str, user=Depends(get_verified_user), db: AsyncSes @router.get('/model/profile/image') -async def get_model_profile_image(id: str, user=Depends(get_verified_user)): - model = await Models.get_model_by_id(id) +async def get_model_profile_image( + id: str, + user=Depends(get_verified_user), + db: AsyncSession = Depends(get_async_session), +): + model = await Models.get_model_by_id(id, db=db) if model: etag = f'"{model.updated_at}"' if model.updated_at else None