From 689317c5b51a872aa3cff68e94962145fba559a0 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Tue, 11 Feb 2025 23:13:48 -0800 Subject: [PATCH] refac --- backend/open_webui/routers/configs.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/backend/open_webui/routers/configs.py b/backend/open_webui/routers/configs.py index 1759c9f30..016075234 100644 --- a/backend/open_webui/routers/configs.py +++ b/backend/open_webui/routers/configs.py @@ -41,20 +41,22 @@ async def export_config(user=Depends(get_admin_user)): ############################ -class DirectAPIConfigForm(BaseModel): +class DirectConnectionsConfigForm(BaseModel): ENABLE_DIRECT_CONNECTIONS: bool -@router.get("/direct_connections", response_model=DirectAPIConfigForm) +@router.get("/direct_connections", response_model=DirectConnectionsConfigForm) async def get_direct_connections_config(request: Request, user=Depends(get_admin_user)): return { "ENABLE_DIRECT_CONNECTIONS": request.app.state.config.ENABLE_DIRECT_CONNECTIONS, } -@router.post("/direct_connections", response_model=DirectAPIConfigForm) +@router.post("/direct_connections", response_model=DirectConnectionsConfigForm) async def set_direct_connections_config( - request: Request, form_data: DirectAPIConfigForm, user=Depends(get_admin_user) + request: Request, + form_data: DirectConnectionsConfigForm, + user=Depends(get_admin_user), ): request.app.state.config.ENABLE_DIRECT_CONNECTIONS = ( form_data.ENABLE_DIRECT_CONNECTIONS