From 7c2f297c849d7a2b84ed9e9e10bb2639fffe248a Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sun, 4 Feb 2024 01:07:18 -0800 Subject: [PATCH] feat: export all chats from db admin only --- backend/apps/web/models/chats.py | 12 +++-- backend/apps/web/routers/chats.py | 19 ++++++++ src/lib/apis/chats/index.ts | 31 ++++++++++++ src/lib/components/chat/SettingsModal.svelte | 50 +++++++++++++++++--- 4 files changed, 103 insertions(+), 9 deletions(-) diff --git a/backend/apps/web/models/chats.py b/backend/apps/web/models/chats.py index 4895740d..ca55c71d 100644 --- a/backend/apps/web/models/chats.py +++ b/backend/apps/web/models/chats.py @@ -70,9 +70,9 @@ class ChatTable: **{ "id": id, "user_id": user_id, - "title": form_data.chat["title"] - if "title" in form_data.chat - else "New Chat", + "title": ( + form_data.chat["title"] if "title" in form_data.chat else "New Chat" + ), "chat": json.dumps(form_data.chat), "timestamp": int(time.time()), } @@ -131,6 +131,12 @@ class ChatTable: .order_by(Chat.timestamp.desc()) ] + def get_all_chats(self) -> List[ChatModel]: + return [ + ChatModel(**model_to_dict(chat)) + for chat in Chat.select().order_by(Chat.timestamp.desc()) + ] + def get_all_chats_by_user_id(self, user_id: str) -> List[ChatModel]: return [ ChatModel(**model_to_dict(chat)) diff --git a/backend/apps/web/routers/chats.py b/backend/apps/web/routers/chats.py index d2830095..aa725409 100644 --- a/backend/apps/web/routers/chats.py +++ b/backend/apps/web/routers/chats.py @@ -54,6 +54,25 @@ async def get_all_user_chats(user=Depends(get_current_user)): ] +############################ +# GetAllChatsInDB +############################ + + +@router.get("/all/db", response_model=List[ChatResponse]) +async def get_all_user_chats_in_db(user=Depends(get_current_user)): + if user.role == "admin": + return [ + ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)}) + for chat in Chats.get_all_chats() + ] + else: + raise HTTPException( + status_code=status.HTTP_403_FORBIDDEN, + detail=ERROR_MESSAGES.ACCESS_PROHIBITED, + ) + + ############################ # CreateNewChat ############################ diff --git a/src/lib/apis/chats/index.ts b/src/lib/apis/chats/index.ts index 7c515f11..58c14bb9 100644 --- a/src/lib/apis/chats/index.ts +++ b/src/lib/apis/chats/index.ts @@ -93,6 +93,37 @@ export const getAllChats = async (token: string) => { return res; }; +export const getAllUserChats = async (token: string) => { + let error = null; + + const res = await fetch(`${WEBUI_API_BASE_URL}/chats/all/db`, { + method: 'GET', + headers: { + Accept: 'application/json', + 'Content-Type': 'application/json', + ...(token && { authorization: `Bearer ${token}` }) + } + }) + .then(async (res) => { + if (!res.ok) throw await res.json(); + return res.json(); + }) + .then((json) => { + return json; + }) + .catch((err) => { + error = err; + console.log(err); + return null; + }); + + if (error) { + throw error; + } + + return res; +}; + export const getAllChatTags = async (token: string) => { let error = null; diff --git a/src/lib/components/chat/SettingsModal.svelte b/src/lib/components/chat/SettingsModal.svelte index e348c807..82973864 100644 --- a/src/lib/components/chat/SettingsModal.svelte +++ b/src/lib/components/chat/SettingsModal.svelte @@ -17,7 +17,13 @@ deleteModel } from '$lib/apis/ollama'; import { updateUserPassword } from '$lib/apis/auths'; - import { createNewChat, deleteAllChats, getAllChats, getChatList } from '$lib/apis/chats'; + import { + createNewChat, + deleteAllChats, + getAllChats, + getAllUserChats, + getChatList + } from '$lib/apis/chats'; import { WEB_UI_VERSION, WEBUI_API_BASE_URL } from '$lib/constants'; import { config, models, voices, settings, user, chats } from '$lib/stores'; @@ -179,6 +185,13 @@ saveAs(blob, `chat-export-${Date.now()}.json`); }; + const exportAllUserChats = async () => { + let blob = new Blob([JSON.stringify(await getAllUserChats(localStorage.token))], { + type: 'application/json' + }); + saveAs(blob, `all-chats-export-${Date.now()}.json`); + }; + const deleteChats = async () => { await goto('/'); await deleteAllChats(localStorage.token); @@ -1962,23 +1975,48 @@ fill="currentColor" class="w-4 h-4" > - -
Delete All Chats
+
Delete Chats
{/if} {#if $user?.role === 'admin'}
+ + +
+