From 7da37b4f66b9b2e821796b06b75e03cb0237e0a9 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Mon, 12 Jan 2026 21:41:23 +0400 Subject: [PATCH] refac --- backend/open_webui/routers/channels.py | 8 ++------ backend/open_webui/routers/models.py | 6 ++---- backend/open_webui/routers/users.py | 6 ++---- 3 files changed, 6 insertions(+), 14 deletions(-) diff --git a/backend/open_webui/routers/channels.py b/backend/open_webui/routers/channels.py index 4e697142b..7f4b347be 100644 --- a/backend/open_webui/routers/channels.py +++ b/backend/open_webui/routers/channels.py @@ -1874,13 +1874,9 @@ async def delete_message_by_id( @router.get("/webhooks/{webhook_id}/profile/image") -async def get_webhook_profile_image( - webhook_id: str, - user=Depends(get_verified_user), - db: Session = Depends(get_session), -): +def get_webhook_profile_image(webhook_id: str, user=Depends(get_verified_user)): """Get webhook profile image by webhook ID.""" - webhook = Channels.get_webhook_by_id(webhook_id, db=db) + webhook = Channels.get_webhook_by_id(webhook_id) if not webhook: # Return default favicon if webhook not found return FileResponse(f"{STATIC_DIR}/favicon.png") diff --git a/backend/open_webui/routers/models.py b/backend/open_webui/routers/models.py index a1f642bbc..85f0fb4f6 100644 --- a/backend/open_webui/routers/models.py +++ b/backend/open_webui/routers/models.py @@ -343,10 +343,8 @@ async def get_model_by_id( @router.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) +def get_model_profile_image(id: str, user=Depends(get_verified_user)): + model = Models.get_model_by_id(id) if model: etag = f'"{model.updated_at}"' if model.updated_at else None diff --git a/backend/open_webui/routers/users.py b/backend/open_webui/routers/users.py index 8eb4b5d19..20b69bcdf 100644 --- a/backend/open_webui/routers/users.py +++ b/backend/open_webui/routers/users.py @@ -498,10 +498,8 @@ async def get_user_oauth_sessions_by_id( @router.get("/{user_id}/profile/image") -async def get_user_profile_image_by_id( - user_id: str, user=Depends(get_verified_user), db: Session = Depends(get_session) -): - user = Users.get_user_by_id(user_id, db=db) +def get_user_profile_image_by_id(user_id: str, user=Depends(get_verified_user)): + user = Users.get_user_by_id(user_id) if user: if user.profile_image_url: # check if it's url or base64