This commit is contained in:
Timothy Jaeryang Baek 2025-02-11 23:13:48 -08:00
parent 8daa549146
commit 689317c5b5

View File

@ -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