Timothy Jaeryang Baek 2 月之前
父節點
當前提交
689317c5b5
共有 1 個文件被更改,包括 6 次插入4 次删除
  1. 6 4
      backend/open_webui/routers/configs.py

+ 6 - 4
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