POST /utils/markdown rendered a markdown string to HTML server-side. Its only frontend wrapper, getHTMLFromMarkdown in src/lib/apis/utils/index.ts, is dead: nothing imports or calls it, the route is hit by no other code path, and the path string appears nowhere else in the repo (no direct fetch, no test, no docs). Markdown is rendered client-side in the UI, so this endpoint was redundant. Fully self-contained removal: the endpoint, its MarkdownForm model, the now-orphaned 'import markdown' in the utils router (used only here), and the dead getHTMLFromMarkdown wrapper. Nothing else depends on any of them. Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
114 lines
3.6 KiB
Python
114 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
import black
|
|
from fastapi import APIRouter, Depends, HTTPException, Request, Response, status
|
|
from open_webui.config import DATA_DIR, ENABLE_ADMIN_EXPORT
|
|
from open_webui.constants import ERROR_MESSAGES
|
|
from open_webui.models.chats import ChatTitleMessagesForm
|
|
from open_webui.utils.auth import get_admin_user, get_verified_user
|
|
from open_webui.utils.code_interpreter import execute_code_jupyter
|
|
from open_webui.utils.misc import get_gravatar_url
|
|
from open_webui.utils.pdf_generator import PDFGenerator
|
|
from pydantic import BaseModel
|
|
from starlette.responses import FileResponse
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get('/gravatar')
|
|
async def get_gravatar(email: str, user=Depends(get_verified_user)):
|
|
return get_gravatar_url(email)
|
|
|
|
|
|
class CodeForm(BaseModel):
|
|
code: str
|
|
|
|
|
|
@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}
|
|
except black.NothingChanged:
|
|
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 not request.app.state.config.ENABLE_CODE_EXECUTION:
|
|
raise HTTPException(
|
|
status_code=403,
|
|
detail=ERROR_MESSAGES.FEATURE_DISABLED('Code execution'),
|
|
)
|
|
|
|
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'
|
|
else None
|
|
),
|
|
(
|
|
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,
|
|
)
|
|
|
|
return output
|
|
else:
|
|
raise HTTPException(
|
|
status_code=400,
|
|
detail=ERROR_MESSAGES.DEFAULT('Code execution engine not supported'),
|
|
)
|
|
|
|
|
|
class ChatForm(BaseModel):
|
|
title: str
|
|
messages: list[dict]
|
|
|
|
|
|
@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'},
|
|
)
|
|
except Exception as e:
|
|
log.exception(f'Error generating PDF: {e}')
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.get('/db/download')
|
|
async def download_db(user=Depends(get_admin_user)):
|
|
if not ENABLE_ADMIN_EXPORT:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
from open_webui.internal.db import engine
|
|
|
|
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',
|
|
)
|