2023-12-24 00:38:52 +01:00
|
|
|
from fastapi import APIRouter, UploadFile, File, BackgroundTasks
|
|
|
|
from fastapi import Depends, HTTPException, status
|
2024-03-02 09:33:20 +01:00
|
|
|
from starlette.responses import StreamingResponse, FileResponse
|
|
|
|
|
2023-12-24 00:38:52 +01:00
|
|
|
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
|
|
import requests
|
|
|
|
import os
|
2023-12-24 08:40:14 +01:00
|
|
|
import aiohttp
|
2023-12-24 00:38:52 +01:00
|
|
|
import json
|
2023-12-24 08:40:14 +01:00
|
|
|
|
2024-03-02 09:33:20 +01:00
|
|
|
|
|
|
|
from utils.utils import get_admin_user
|
2024-01-27 06:38:33 +01:00
|
|
|
from utils.misc import calculate_sha256, get_gravatar_url
|
2023-12-24 08:40:14 +01:00
|
|
|
|
2024-03-06 20:44:00 +01:00
|
|
|
from config import OLLAMA_BASE_URLS, DATA_DIR, UPLOAD_DIR
|
2024-01-09 22:25:42 +01:00
|
|
|
from constants import ERROR_MESSAGES
|
|
|
|
|
2023-12-24 00:38:52 +01:00
|
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
|
|
|
2024-01-27 06:38:33 +01:00
|
|
|
@router.get("/gravatar")
|
|
|
|
async def get_gravatar(
|
|
|
|
email: str,
|
|
|
|
):
|
|
|
|
return get_gravatar_url(email)
|
2024-03-02 09:33:20 +01:00
|
|
|
|
|
|
|
|
|
|
|
@router.get("/db/download")
|
|
|
|
async def download_db(user=Depends(get_admin_user)):
|
|
|
|
|
|
|
|
return FileResponse(
|
|
|
|
f"{DATA_DIR}/webui.db",
|
|
|
|
media_type="application/octet-stream",
|
|
|
|
filename="webui.db",
|
|
|
|
)
|