From b377e5ff4ca7630c61eb47a81920e2d3a5317173 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Fri, 9 Jan 2026 02:46:04 +0400 Subject: [PATCH] chore: format --- backend/open_webui/models/channels.py | 12 +++--------- backend/open_webui/routers/channels.py | 1 + backend/open_webui/routers/files.py | 4 +++- backend/open_webui/routers/users.py | 3 ++- backend/open_webui/utils/chat.py | 14 ++++++++++---- backend/open_webui/utils/middleware.py | 10 ++++++++-- 6 files changed, 27 insertions(+), 17 deletions(-) diff --git a/backend/open_webui/models/channels.py b/backend/open_webui/models/channels.py index e95b6f6c9..8e70918e1 100644 --- a/backend/open_webui/models/channels.py +++ b/backend/open_webui/models/channels.py @@ -994,9 +994,7 @@ class ChannelTable: ) -> Optional[ChannelWebhookModel]: with get_db_context(db) as db: webhook = ( - db.query(ChannelWebhook) - .filter(ChannelWebhook.id == webhook_id) - .first() + db.query(ChannelWebhook).filter(ChannelWebhook.id == webhook_id).first() ) return ChannelWebhookModel.model_validate(webhook) if webhook else None @@ -1022,9 +1020,7 @@ class ChannelTable: ) -> Optional[ChannelWebhookModel]: with get_db_context(db) as db: webhook = ( - db.query(ChannelWebhook) - .filter(ChannelWebhook.id == webhook_id) - .first() + db.query(ChannelWebhook).filter(ChannelWebhook.id == webhook_id).first() ) if not webhook: return None @@ -1039,9 +1035,7 @@ class ChannelTable: ) -> bool: with get_db_context(db) as db: webhook = ( - db.query(ChannelWebhook) - .filter(ChannelWebhook.id == webhook_id) - .first() + db.query(ChannelWebhook).filter(ChannelWebhook.id == webhook_id).first() ) if not webhook: return False diff --git a/backend/open_webui/routers/channels.py b/backend/open_webui/routers/channels.py index d501818b7..777f5e74e 100644 --- a/backend/open_webui/routers/channels.py +++ b/backend/open_webui/routers/channels.py @@ -1908,6 +1908,7 @@ async def get_webhook_profile_image( # Return default favicon if no profile image return FileResponse(f"{STATIC_DIR}/favicon.png") + @router.get("/{id}/webhooks", response_model=list[ChannelWebhookModel]) async def get_channel_webhooks( request: Request, diff --git a/backend/open_webui/routers/files.py b/backend/open_webui/routers/files.py index 15d2a503c..d0e56075e 100644 --- a/backend/open_webui/routers/files.py +++ b/backend/open_webui/routers/files.py @@ -377,7 +377,9 @@ async def search_files( ), content: bool = Query(True), skip: int = Query(0, ge=0, description="Number of files to skip"), - limit: int = Query(100, ge=1, le=1000, description="Maximum number of files to return"), + limit: int = Query( + 100, ge=1, le=1000, description="Maximum number of files to return" + ), user=Depends(get_verified_user), db: Session = Depends(get_session), ): diff --git a/backend/open_webui/routers/users.py b/backend/open_webui/routers/users.py index ac53b73c8..8eb4b5d19 100644 --- a/backend/open_webui/routers/users.py +++ b/backend/open_webui/routers/users.py @@ -334,7 +334,8 @@ async def update_user_settings_by_session_user( @router.get("/user/status") async def get_user_status_by_session_user( request: Request, - user=Depends(get_verified_user), db: Session = Depends(get_session) + user=Depends(get_verified_user), + db: Session = Depends(get_session), ): if not request.app.state.config.ENABLE_USER_STATUS: raise HTTPException( diff --git a/backend/open_webui/utils/chat.py b/backend/open_webui/utils/chat.py index e270de51d..be700dda7 100644 --- a/backend/open_webui/utils/chat.py +++ b/backend/open_webui/utils/chat.py @@ -238,8 +238,11 @@ async def generate_chat_completion( yield chunk response = await generate_chat_completion( - request, form_data, user, bypass_filter=True, - bypass_system_prompt=bypass_system_prompt + request, + form_data, + user, + bypass_filter=True, + bypass_system_prompt=bypass_system_prompt, ) return StreamingResponse( stream_wrapper(response.body_iterator), @@ -250,8 +253,11 @@ async def generate_chat_completion( return { **( await generate_chat_completion( - request, form_data, user, bypass_filter=True, - bypass_system_prompt=bypass_system_prompt + request, + form_data, + user, + bypass_filter=True, + bypass_system_prompt=bypass_system_prompt, ) ), "selected_model_id": selected_model_id, diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 6f8f8d274..0cfd03d9a 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -93,7 +93,11 @@ from open_webui.utils.misc import ( convert_logit_bias_input_to_json, get_content_from_message, ) -from open_webui.utils.tools import get_tools, get_updated_tool_function, has_tool_server_access +from open_webui.utils.tools import ( + get_tools, + get_updated_tool_function, + has_tool_server_access, +) from open_webui.utils.plugin import load_function_module_by_id from open_webui.utils.filter import ( get_sorted_filter_ids, @@ -1665,7 +1669,9 @@ async def process_chat_payload(request, form_data, user, metadata, model): # Check access control for MCP server if not has_tool_server_access(user, mcp_server_connection): - log.warning(f"Access denied to MCP server {server_id} for user {user.id}") + log.warning( + f"Access denied to MCP server {server_id} for user {user.id}" + ) continue auth_type = mcp_server_connection.get("auth_type", "")