mirror of
https://github.com/open-webui/open-webui
synced 2025-06-26 18:26:48 +00:00
enh: folder export
This commit is contained in:
parent
3c1afa97af
commit
7d322a7238
@ -561,6 +561,21 @@ class ChatTable:
|
|||||||
all_chats = query.all()
|
all_chats = query.all()
|
||||||
return [ChatModel.model_validate(chat) for chat in all_chats]
|
return [ChatModel.model_validate(chat) for chat in all_chats]
|
||||||
|
|
||||||
|
def get_chats_by_folder_ids_and_user_id(
|
||||||
|
self, folder_ids: list[str], user_id: str
|
||||||
|
) -> list[ChatModel]:
|
||||||
|
with get_db() as db:
|
||||||
|
query = db.query(Chat).filter(
|
||||||
|
Chat.folder_id.in_(folder_ids), Chat.user_id == user_id
|
||||||
|
)
|
||||||
|
query = query.filter(or_(Chat.pinned == False, Chat.pinned == None))
|
||||||
|
query = query.filter_by(archived=False)
|
||||||
|
|
||||||
|
query = query.order_by(Chat.updated_at.desc())
|
||||||
|
|
||||||
|
all_chats = query.all()
|
||||||
|
return [ChatModel.model_validate(chat) for chat in all_chats]
|
||||||
|
|
||||||
def update_chat_folder_id_by_id_and_user_id(
|
def update_chat_folder_id_by_id_and_user_id(
|
||||||
self, id: str, user_id: str, folder_id: str
|
self, id: str, user_id: str, folder_id: str
|
||||||
) -> Optional[ChatModel]:
|
) -> Optional[ChatModel]:
|
||||||
|
@ -99,6 +99,30 @@ class FolderTable:
|
|||||||
except Exception:
|
except Exception:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
def get_children_folders_by_id_and_user_id(
|
||||||
|
self, id: str, user_id: str
|
||||||
|
) -> Optional[FolderModel]:
|
||||||
|
try:
|
||||||
|
with get_db() as db:
|
||||||
|
folders = []
|
||||||
|
|
||||||
|
def get_children(folder):
|
||||||
|
children = self.get_folders_by_parent_id_and_user_id(
|
||||||
|
folder.id, user_id
|
||||||
|
)
|
||||||
|
for child in children:
|
||||||
|
get_children(child)
|
||||||
|
folders.append(child)
|
||||||
|
|
||||||
|
folder = db.query(Folder).filter_by(id=id, user_id=user_id).first()
|
||||||
|
if not folder:
|
||||||
|
return None
|
||||||
|
|
||||||
|
get_children(folder)
|
||||||
|
return folders
|
||||||
|
except Exception:
|
||||||
|
return None
|
||||||
|
|
||||||
def get_folders_by_user_id(self, user_id: str) -> list[FolderModel]:
|
def get_folders_by_user_id(self, user_id: str) -> list[FolderModel]:
|
||||||
with get_db() as db:
|
with get_db() as db:
|
||||||
return [
|
return [
|
||||||
|
@ -10,6 +10,7 @@ from open_webui.apps.webui.models.chats import (
|
|||||||
ChatTitleIdResponse,
|
ChatTitleIdResponse,
|
||||||
)
|
)
|
||||||
from open_webui.apps.webui.models.tags import TagModel, Tags
|
from open_webui.apps.webui.models.tags import TagModel, Tags
|
||||||
|
from open_webui.apps.webui.models.folders import Folders
|
||||||
|
|
||||||
from open_webui.config import ENABLE_ADMIN_CHAT_ACCESS, ENABLE_ADMIN_EXPORT
|
from open_webui.config import ENABLE_ADMIN_CHAT_ACCESS, ENABLE_ADMIN_EXPORT
|
||||||
from open_webui.constants import ERROR_MESSAGES
|
from open_webui.constants import ERROR_MESSAGES
|
||||||
@ -151,6 +152,26 @@ async def search_user_chats(
|
|||||||
return chat_list
|
return chat_list
|
||||||
|
|
||||||
|
|
||||||
|
############################
|
||||||
|
# GetChatsByFolderId
|
||||||
|
############################
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/folder/{folder_id}", response_model=list[ChatResponse])
|
||||||
|
async def get_chats_by_folder_id(folder_id: str, user=Depends(get_verified_user)):
|
||||||
|
folder_ids = [folder_id]
|
||||||
|
children_folders = Folders.get_children_folders_by_id_and_user_id(
|
||||||
|
folder_id, user.id
|
||||||
|
)
|
||||||
|
if children_folders:
|
||||||
|
folder_ids.extend([folder.id for folder in children_folders])
|
||||||
|
|
||||||
|
return [
|
||||||
|
ChatResponse(**chat.model_dump())
|
||||||
|
for chat in Chats.get_chats_by_folder_ids_and_user_id(folder_ids, user.id)
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
############################
|
############################
|
||||||
# GetPinnedChats
|
# GetPinnedChats
|
||||||
############################
|
############################
|
||||||
|
@ -243,6 +243,37 @@ export const getChatListBySearchText = async (token: string, text: string, page:
|
|||||||
}));
|
}));
|
||||||
};
|
};
|
||||||
|
|
||||||
|
export const getChatsByFolderId = async (token: string, folderId: string) => {
|
||||||
|
let error = null;
|
||||||
|
|
||||||
|
const res = await fetch(`${WEBUI_API_BASE_URL}/chats/folder/${folderId}`, {
|
||||||
|
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 getAllArchivedChats = async (token: string) => {
|
export const getAllArchivedChats = async (token: string) => {
|
||||||
let error = null;
|
let error = null;
|
||||||
|
|
||||||
|
@ -1,124 +0,0 @@
|
|||||||
<script lang="ts">
|
|
||||||
import { DropdownMenu } from 'bits-ui';
|
|
||||||
import { flyAndScale } from '$lib/utils/transitions';
|
|
||||||
import { getContext } from 'svelte';
|
|
||||||
|
|
||||||
import Dropdown from '$lib/components/common/Dropdown.svelte';
|
|
||||||
import GarbageBin from '$lib/components/icons/GarbageBin.svelte';
|
|
||||||
import Pencil from '$lib/components/icons/Pencil.svelte';
|
|
||||||
import Tooltip from '$lib/components/common/Tooltip.svelte';
|
|
||||||
import Tags from '$lib/components/chat/Tags.svelte';
|
|
||||||
import Share from '$lib/components/icons/Share.svelte';
|
|
||||||
import ArchiveBox from '$lib/components/icons/ArchiveBox.svelte';
|
|
||||||
import DocumentDuplicate from '$lib/components/icons/DocumentDuplicate.svelte';
|
|
||||||
import Star from '$lib/components/icons/Star.svelte';
|
|
||||||
|
|
||||||
const i18n = getContext('i18n');
|
|
||||||
|
|
||||||
export let pinHandler: Function;
|
|
||||||
export let shareHandler: Function;
|
|
||||||
export let cloneChatHandler: Function;
|
|
||||||
export let archiveChatHandler: Function;
|
|
||||||
export let renameHandler: Function;
|
|
||||||
export let deleteHandler: Function;
|
|
||||||
export let onClose: Function;
|
|
||||||
|
|
||||||
export let chatId = '';
|
|
||||||
|
|
||||||
let show = false;
|
|
||||||
</script>
|
|
||||||
|
|
||||||
<Dropdown
|
|
||||||
bind:show
|
|
||||||
on:change={(e) => {
|
|
||||||
if (e.detail === false) {
|
|
||||||
onClose();
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Tooltip content={$i18n.t('More')}>
|
|
||||||
<slot />
|
|
||||||
</Tooltip>
|
|
||||||
|
|
||||||
<div slot="content">
|
|
||||||
<DropdownMenu.Content
|
|
||||||
class="w-full max-w-[180px] rounded-xl px-1 py-1.5 border border-gray-300/30 dark:border-gray-700/50 z-50 bg-white dark:bg-gray-850 dark:text-white shadow"
|
|
||||||
sideOffset={-2}
|
|
||||||
side="bottom"
|
|
||||||
align="start"
|
|
||||||
transition={flyAndScale}
|
|
||||||
>
|
|
||||||
<DropdownMenu.Item
|
|
||||||
class="flex gap-2 items-center px-3 py-2 text-sm font-medium cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
|
||||||
on:click={() => {
|
|
||||||
pinHandler();
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Star strokeWidth="2" />
|
|
||||||
<div class="flex items-center">{$i18n.t('Pin')}</div>
|
|
||||||
</DropdownMenu.Item>
|
|
||||||
|
|
||||||
<DropdownMenu.Item
|
|
||||||
class="flex gap-2 items-center px-3 py-2 text-sm font-medium cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
|
||||||
on:click={() => {
|
|
||||||
renameHandler();
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Pencil strokeWidth="2" />
|
|
||||||
<div class="flex items-center">{$i18n.t('Rename')}</div>
|
|
||||||
</DropdownMenu.Item>
|
|
||||||
|
|
||||||
<DropdownMenu.Item
|
|
||||||
class="flex gap-2 items-center px-3 py-2 text-sm font-medium cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
|
||||||
on:click={() => {
|
|
||||||
cloneChatHandler();
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<DocumentDuplicate strokeWidth="2" />
|
|
||||||
<div class="flex items-center">{$i18n.t('Clone')}</div>
|
|
||||||
</DropdownMenu.Item>
|
|
||||||
|
|
||||||
<DropdownMenu.Item
|
|
||||||
class="flex gap-2 items-center px-3 py-2 text-sm font-medium cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
|
||||||
on:click={() => {
|
|
||||||
archiveChatHandler();
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<ArchiveBox strokeWidth="2" />
|
|
||||||
<div class="flex items-center">{$i18n.t('Archive')}</div>
|
|
||||||
</DropdownMenu.Item>
|
|
||||||
|
|
||||||
<DropdownMenu.Item
|
|
||||||
class="flex gap-2 items-center px-3 py-2 text-sm font-medium cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
|
||||||
on:click={() => {
|
|
||||||
shareHandler();
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<Share />
|
|
||||||
<div class="flex items-center">{$i18n.t('Share')}</div>
|
|
||||||
</DropdownMenu.Item>
|
|
||||||
|
|
||||||
<DropdownMenu.Item
|
|
||||||
class="flex gap-2 items-center px-3 py-2 text-sm font-medium cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
|
||||||
on:click={() => {
|
|
||||||
deleteHandler();
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<GarbageBin strokeWidth="2" />
|
|
||||||
<div class="flex items-center">{$i18n.t('Delete')}</div>
|
|
||||||
</DropdownMenu.Item>
|
|
||||||
|
|
||||||
<hr class="border-gray-100 dark:border-gray-800 mt-2.5 mb-1.5" />
|
|
||||||
|
|
||||||
<div class="flex p-1">
|
|
||||||
<Tags
|
|
||||||
{chatId}
|
|
||||||
on:close={() => {
|
|
||||||
show = false;
|
|
||||||
onClose();
|
|
||||||
}}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
</DropdownMenu.Content>
|
|
||||||
</div>
|
|
||||||
</Dropdown>
|
|
19
src/lib/components/icons/Download.svelte
Normal file
19
src/lib/components/icons/Download.svelte
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
<script lang="ts">
|
||||||
|
export let className = 'size-4';
|
||||||
|
export let strokeWidth = '1.5';
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<svg
|
||||||
|
xmlns="http://www.w3.org/2000/svg"
|
||||||
|
fill="none"
|
||||||
|
viewBox="0 0 24 24"
|
||||||
|
stroke-width={strokeWidth}
|
||||||
|
stroke="currentColor"
|
||||||
|
class={className}
|
||||||
|
>
|
||||||
|
<path
|
||||||
|
stroke-linecap="round"
|
||||||
|
stroke-linejoin="round"
|
||||||
|
d="M3 16.5v2.25A2.25 2.25 0 0 0 5.25 21h13.5A2.25 2.25 0 0 0 21 18.75V16.5M16.5 12 12 16.5m0 0L7.5 12m4.5 4.5V3"
|
||||||
|
/>
|
||||||
|
</svg>
|
@ -26,6 +26,7 @@
|
|||||||
import { chats } from '$lib/stores';
|
import { chats } from '$lib/stores';
|
||||||
import { createMessagesList } from '$lib/utils';
|
import { createMessagesList } from '$lib/utils';
|
||||||
import { downloadChatAsPDF } from '$lib/apis/utils';
|
import { downloadChatAsPDF } from '$lib/apis/utils';
|
||||||
|
import Download from '$lib/components/icons/Download.svelte';
|
||||||
|
|
||||||
const i18n = getContext('i18n');
|
const i18n = getContext('i18n');
|
||||||
|
|
||||||
@ -198,20 +199,7 @@
|
|||||||
<DropdownMenu.SubTrigger
|
<DropdownMenu.SubTrigger
|
||||||
class="flex gap-2 items-center px-3 py-2 text-sm cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
class="flex gap-2 items-center px-3 py-2 text-sm cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
||||||
>
|
>
|
||||||
<svg
|
<Download strokeWidth="2" />
|
||||||
xmlns="http://www.w3.org/2000/svg"
|
|
||||||
fill="none"
|
|
||||||
viewBox="0 0 24 24"
|
|
||||||
stroke-width="1.5"
|
|
||||||
stroke="currentColor"
|
|
||||||
class="size-4"
|
|
||||||
>
|
|
||||||
<path
|
|
||||||
stroke-linecap="round"
|
|
||||||
stroke-linejoin="round"
|
|
||||||
d="M3 16.5v2.25A2.25 2.25 0 0 0 5.25 21h13.5A2.25 2.25 0 0 0 21 18.75V16.5M16.5 12 12 16.5m0 0L7.5 12m4.5 4.5V3"
|
|
||||||
/>
|
|
||||||
</svg>
|
|
||||||
|
|
||||||
<div class="flex items-center">{$i18n.t('Download')}</div>
|
<div class="flex items-center">{$i18n.t('Download')}</div>
|
||||||
</DropdownMenu.SubTrigger>
|
</DropdownMenu.SubTrigger>
|
||||||
|
@ -10,6 +10,7 @@
|
|||||||
import GarbageBin from '$lib/components/icons/GarbageBin.svelte';
|
import GarbageBin from '$lib/components/icons/GarbageBin.svelte';
|
||||||
import Pencil from '$lib/components/icons/Pencil.svelte';
|
import Pencil from '$lib/components/icons/Pencil.svelte';
|
||||||
import Tooltip from '$lib/components/common/Tooltip.svelte';
|
import Tooltip from '$lib/components/common/Tooltip.svelte';
|
||||||
|
import Download from '$lib/components/icons/Download.svelte';
|
||||||
|
|
||||||
let show = false;
|
let show = false;
|
||||||
</script>
|
</script>
|
||||||
@ -44,6 +45,17 @@
|
|||||||
<div class="flex items-center">{$i18n.t('Rename')}</div>
|
<div class="flex items-center">{$i18n.t('Rename')}</div>
|
||||||
</DropdownMenu.Item>
|
</DropdownMenu.Item>
|
||||||
|
|
||||||
|
<DropdownMenu.Item
|
||||||
|
class="flex gap-2 items-center px-3 py-1.5 text-sm cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
||||||
|
on:click={() => {
|
||||||
|
dispatch('export');
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<Download strokeWidth="2" />
|
||||||
|
|
||||||
|
<div class="flex items-center">{$i18n.t('Export')}</div>
|
||||||
|
</DropdownMenu.Item>
|
||||||
|
|
||||||
<DropdownMenu.Item
|
<DropdownMenu.Item
|
||||||
class="flex gap-2 items-center px-3 py-1.5 text-sm cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
class="flex gap-2 items-center px-3 py-1.5 text-sm cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-md"
|
||||||
on:click={() => {
|
on:click={() => {
|
||||||
|
@ -1,10 +1,13 @@
|
|||||||
<script>
|
<script>
|
||||||
import { getContext, createEventDispatcher, onMount, onDestroy, tick } from 'svelte';
|
import { getContext, createEventDispatcher, onMount, onDestroy, tick } from 'svelte';
|
||||||
import DOMPurify from 'dompurify';
|
|
||||||
|
|
||||||
const i18n = getContext('i18n');
|
const i18n = getContext('i18n');
|
||||||
const dispatch = createEventDispatcher();
|
const dispatch = createEventDispatcher();
|
||||||
|
|
||||||
|
import DOMPurify from 'dompurify';
|
||||||
|
import fileSaver from 'file-saver';
|
||||||
|
const { saveAs } = fileSaver;
|
||||||
|
|
||||||
import ChevronDown from '../../icons/ChevronDown.svelte';
|
import ChevronDown from '../../icons/ChevronDown.svelte';
|
||||||
import ChevronRight from '../../icons/ChevronRight.svelte';
|
import ChevronRight from '../../icons/ChevronRight.svelte';
|
||||||
import Collapsible from '../../common/Collapsible.svelte';
|
import Collapsible from '../../common/Collapsible.svelte';
|
||||||
@ -19,7 +22,7 @@
|
|||||||
updateFolderParentIdById
|
updateFolderParentIdById
|
||||||
} from '$lib/apis/folders';
|
} from '$lib/apis/folders';
|
||||||
import { toast } from 'svelte-sonner';
|
import { toast } from 'svelte-sonner';
|
||||||
import { updateChatFolderIdById } from '$lib/apis/chats';
|
import { getChatsByFolderId, updateChatFolderIdById } from '$lib/apis/chats';
|
||||||
import ChatItem from './ChatItem.svelte';
|
import ChatItem from './ChatItem.svelte';
|
||||||
import FolderMenu from './Folders/FolderMenu.svelte';
|
import FolderMenu from './Folders/FolderMenu.svelte';
|
||||||
import DeleteConfirmDialog from '$lib/components/common/ConfirmDialog.svelte';
|
import DeleteConfirmDialog from '$lib/components/common/ConfirmDialog.svelte';
|
||||||
@ -292,6 +295,22 @@
|
|||||||
input.focus();
|
input.focus();
|
||||||
}, 100);
|
}, 100);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
const exportHandler = async () => {
|
||||||
|
const chats = await getChatsByFolderId(localStorage.token, folderId).catch((error) => {
|
||||||
|
toast.error(error);
|
||||||
|
return null;
|
||||||
|
});
|
||||||
|
if (!chats) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const blob = new Blob([JSON.stringify(chats)], {
|
||||||
|
type: 'application/json'
|
||||||
|
});
|
||||||
|
|
||||||
|
saveAs(blob, `folder-${folders[folderId].name}-export-${Date.now()}.json`);
|
||||||
|
};
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<DeleteConfirmDialog
|
<DeleteConfirmDialog
|
||||||
@ -400,6 +419,9 @@
|
|||||||
on:delete={() => {
|
on:delete={() => {
|
||||||
showDeleteConfirm = true;
|
showDeleteConfirm = true;
|
||||||
}}
|
}}
|
||||||
|
on:export={() => {
|
||||||
|
exportHandler();
|
||||||
|
}}
|
||||||
>
|
>
|
||||||
<button class="p-0.5 dark:hover:bg-gray-850 rounded-lg touch-auto" on:click={(e) => {}}>
|
<button class="p-0.5 dark:hover:bg-gray-850 rounded-lg touch-auto" on:click={(e) => {}}>
|
||||||
<EllipsisHorizontal className="size-4" strokeWidth="2.5" />
|
<EllipsisHorizontal className="size-4" strokeWidth="2.5" />
|
||||||
|
Loading…
Reference in New Issue
Block a user