refac

backend/open_webui/routers/utils.py

@@ -20,7 +20,7 @@ log = logging.getLogger(__name__)
 router = APIRouter()
 
 
-@router.get("/gravatar")
+@router.get('/gravatar')
 async def get_gravatar(email: str, user=Depends(get_verified_user)):
     return get_gravatar_url(email)
 
@@ -29,33 +29,31 @@ class CodeForm(BaseModel):
     code: str
 
 
-@router.post("/code/format")
+@router.post('/code/format')
 async def format_code(form_data: CodeForm, user=Depends(get_admin_user)):
     try:
         formatted_code = black.format_str(form_data.code, mode=black.Mode())
-        return {"code": formatted_code}
+        return {'code': formatted_code}
     except black.NothingChanged:
-        return {"code": form_data.code}
+        return {'code': form_data.code}
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
 
 
-@router.post("/code/execute")
-async def execute_code(
-    request: Request, form_data: CodeForm, user=Depends(get_verified_user)
-):
-    if request.app.state.config.CODE_EXECUTION_ENGINE == "jupyter":
+@router.post('/code/execute')
+async def execute_code(request: Request, form_data: CodeForm, user=Depends(get_verified_user)):
+    if request.app.state.config.CODE_EXECUTION_ENGINE == 'jupyter':
         output = await execute_code_jupyter(
             request.app.state.config.CODE_EXECUTION_JUPYTER_URL,
             form_data.code,
             (
                 request.app.state.config.CODE_EXECUTION_JUPYTER_AUTH_TOKEN
-                if request.app.state.config.CODE_EXECUTION_JUPYTER_AUTH == "token"
+                if request.app.state.config.CODE_EXECUTION_JUPYTER_AUTH == 'token'
                 else None
             ),
             (
                 request.app.state.config.CODE_EXECUTION_JUPYTER_AUTH_PASSWORD
-                if request.app.state.config.CODE_EXECUTION_JUPYTER_AUTH == "password"
+                if request.app.state.config.CODE_EXECUTION_JUPYTER_AUTH == 'password'
                 else None
             ),
             request.app.state.config.CODE_EXECUTION_JUPYTER_TIMEOUT,
@@ -65,7 +63,7 @@ async def execute_code(
     else:
         raise HTTPException(
             status_code=400,
-            detail="Code execution engine not supported",
+            detail='Code execution engine not supported',
         )
 
 
@@ -73,11 +71,9 @@ class MarkdownForm(BaseModel):
     md: str
 
 
-@router.post("/markdown")
-async def get_html_from_markdown(
-    form_data: MarkdownForm, user=Depends(get_verified_user)
-):
-    return {"html": markdown.markdown(form_data.md)}
+@router.post('/markdown')
+async def get_html_from_markdown(form_data: MarkdownForm, user=Depends(get_verified_user)):
+    return {'html': markdown.markdown(form_data.md)}
 
 
 class ChatForm(BaseModel):
@@ -85,24 +81,22 @@ class ChatForm(BaseModel):
     messages: list[dict]
 
 
-@router.post("/pdf")
-async def download_chat_as_pdf(
-    form_data: ChatTitleMessagesForm, user=Depends(get_verified_user)
-):
+@router.post('/pdf')
+async def download_chat_as_pdf(form_data: ChatTitleMessagesForm, user=Depends(get_verified_user)):
     try:
         pdf_bytes = PDFGenerator(form_data).generate_chat_pdf()
 
         return Response(
             content=pdf_bytes,
-            media_type="application/pdf",
-            headers={"Content-Disposition": "attachment;filename=chat.pdf"},
+            media_type='application/pdf',
+            headers={'Content-Disposition': 'attachment;filename=chat.pdf'},
         )
     except Exception as e:
-        log.exception(f"Error generating PDF: {e}")
+        log.exception(f'Error generating PDF: {e}')
         raise HTTPException(status_code=400, detail=str(e))
 
 
-@router.get("/db/download")
+@router.get('/db/download')
 async def download_db(user=Depends(get_admin_user)):
     if not ENABLE_ADMIN_EXPORT:
         raise HTTPException(
@@ -111,13 +105,13 @@ async def download_db(user=Depends(get_admin_user)):
         )
     from open_webui.internal.db import engine
 
-    if engine.name != "sqlite":
+    if engine.name != 'sqlite':
         raise HTTPException(
             status_code=status.HTTP_400_BAD_REQUEST,
             detail=ERROR_MESSAGES.DB_NOT_SQLITE,
         )
     return FileResponse(
         engine.url.database,
-        media_type="application/octet-stream",
-        filename="webui.db",
+        media_type='application/octet-stream',
+        filename='webui.db',
     )