address comments, rename to ENABLE_ADMIN_EXPORT

This commit is contained in:
Jun Siang Cheah 2024-04-22 19:55:46 +01:00
parent 190b934ab5
commit e2a8ad5fca
6 changed files with 11 additions and 14 deletions

View file

@ -28,10 +28,7 @@ from apps.web.models.tags import (
from constants import ERROR_MESSAGES
from config import (
SRC_LOG_LEVELS,
ALLOW_ADMIN_EXPORT
)
from config import SRC_LOG_LEVELS, ENABLE_ADMIN_EXPORT
log = logging.getLogger(__name__)
log.setLevel(SRC_LOG_LEVELS["MODELS"])
@ -82,7 +79,7 @@ async def get_all_user_chats(user=Depends(get_current_user)):
@router.get("/all/db", response_model=List[ChatResponse])
async def get_all_user_chats_in_db(user=Depends(get_admin_user)):
if not ALLOW_ADMIN_EXPORT:
if not ENABLE_ADMIN_EXPORT:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,

View file

@ -91,7 +91,7 @@ async def download_chat_as_pdf(
@router.get("/db/download")
async def download_db(user=Depends(get_admin_user)):
if not ALLOW_ADMIN_EXPORT:
if not ENABLE_ADMIN_EXPORT:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,

View file

@ -382,7 +382,7 @@ MODEL_FILTER_LIST = [model.strip() for model in MODEL_FILTER_LIST.split(";")]
WEBHOOK_URL = os.environ.get("WEBHOOK_URL", "")
ALLOW_ADMIN_EXPORT = os.environ.get("ALLOW_ADMIN_EXPORT", "True").lower() == "true"
ENABLE_ADMIN_EXPORT = os.environ.get("ENABLE_ADMIN_EXPORT", "True").lower() == "true"
####################################
# WEBUI_VERSION

View file

@ -52,7 +52,7 @@ from config import (
GLOBAL_LOG_LEVEL,
SRC_LOG_LEVELS,
WEBHOOK_URL,
ALLOW_ADMIN_EXPORT,
ENABLE_ADMIN_EXPORT,
)
from constants import ERROR_MESSAGES
@ -208,7 +208,7 @@ async def get_app_config():
"default_models": webui_app.state.DEFAULT_MODELS,
"default_prompt_suggestions": webui_app.state.DEFAULT_PROMPT_SUGGESTIONS,
"trusted_header_auth": bool(webui_app.state.AUTH_TRUSTED_EMAIL_HEADER),
"allow_admin_export": ALLOW_ADMIN_EXPORT,
"admin_export_enabled": ENABLE_ADMIN_EXPORT,
}

View file

@ -25,15 +25,15 @@
<div class=" flex w-full justify-between">
<!-- <div class=" self-center text-xs font-medium">{$i18n.t('Allow Chat Deletion')}</div> -->
{#if $config?.allow_admin_export ?? true}
{#if $config?.admin_export_enabled ?? true}
<button
class=" flex rounded-md py-1.5 px-3 w-full hover:bg-gray-200 dark:hover:bg-gray-800 transition"
type="button"
on:click={() => {
// exportAllUserChats();
// exportAllUserChats();
downloadDatabase(localStorage.token);
}}
downloadDatabase(localStorage.token);
}}
>
<div class=" self-center mr-3">
<svg

View file

@ -301,7 +301,7 @@
</button>
{/if}
{#if $user?.role === 'admin' && ($config?.allow_admin_export ?? true)}
{#if $user?.role === 'admin' && ($config?.admin_export_enabled ?? true)}
<hr class=" dark:border-gray-700" />
<button