From 7674229e3af606bba258479e321f9b5163b7a289 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Fri, 31 May 2024 10:30:42 -0700
Subject: [PATCH] feat: chat clone
---
backend/apps/webui/routers/chats.py | 26 +++++++++++++
src/lib/apis/chats/index.ts | 38 +++++++++++++++++++
.../components/icons/DocumentDuplicate.svelte | 19 ++++++++++
src/lib/components/layout/Sidebar.svelte | 18 ++++++++-
.../components/layout/Sidebar/ChatMenu.svelte | 34 +++++++++++------
.../workspace/Models/ModelMenu.svelte | 16 +-------
6 files changed, 125 insertions(+), 26 deletions(-)
create mode 100644 src/lib/components/icons/DocumentDuplicate.svelte
diff --git a/backend/apps/webui/routers/chats.py b/backend/apps/webui/routers/chats.py
index 5d52f40c9..e7d176fd2 100644
--- a/backend/apps/webui/routers/chats.py
+++ b/backend/apps/webui/routers/chats.py
@@ -288,6 +288,32 @@ async def delete_chat_by_id(request: Request, id: str, user=Depends(get_current_
return result
+############################
+# CloneChat
+############################
+
+
+@router.get("/{id}/clone", response_model=Optional[ChatResponse])
+async def clone_chat_by_id(id: str, user=Depends(get_current_user)):
+ chat = Chats.get_chat_by_id_and_user_id(id, user.id)
+ if chat:
+
+ chat_body = json.loads(chat.chat)
+ updated_chat = {
+ **chat_body,
+ "originalChatId": chat.id,
+ "branchPointMessageId": chat_body["history"]["currentId"],
+ "title": f"Clone of {chat.title}",
+ }
+
+ chat = Chats.insert_new_chat(user.id, ChatForm(**{"chat": updated_chat}))
+ return ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
+ else:
+ raise HTTPException(
+ status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.DEFAULT()
+ )
+
+
############################
# ArchiveChat
############################
diff --git a/src/lib/apis/chats/index.ts b/src/lib/apis/chats/index.ts
index 834e29d29..648e3580e 100644
--- a/src/lib/apis/chats/index.ts
+++ b/src/lib/apis/chats/index.ts
@@ -325,6 +325,44 @@ export const getChatByShareId = async (token: string, share_id: string) => {
return res;
};
+export const cloneChatById = async (token: string, id: string) => {
+ let error = null;
+
+ const res = await fetch(`${WEBUI_API_BASE_URL}/chats/${id}/clone`, {
+ 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;
+
+ if ('detail' in err) {
+ error = err.detail;
+ } else {
+ error = err;
+ }
+
+ console.log(err);
+ return null;
+ });
+
+ if (error) {
+ throw error;
+ }
+
+ return res;
+};
+
export const shareChatById = async (token: string, id: string) => {
let error = null;
diff --git a/src/lib/components/icons/DocumentDuplicate.svelte b/src/lib/components/icons/DocumentDuplicate.svelte
new file mode 100644
index 000000000..a208fefc8
--- /dev/null
+++ b/src/lib/components/icons/DocumentDuplicate.svelte
@@ -0,0 +1,19 @@
+<script lang="ts">
+ export let className = 'w-4 h-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="M15.75 17.25v3.375c0 .621-.504 1.125-1.125 1.125h-9.75a1.125 1.125 0 0 1-1.125-1.125V7.875c0-.621.504-1.125 1.125-1.125H6.75a9.06 9.06 0 0 1 1.5.124m7.5 10.376h3.375c.621 0 1.125-.504 1.125-1.125V11.25c0-4.46-3.243-8.161-7.5-8.876a9.06 9.06 0 0 0-1.5-.124H9.375c-.621 0-1.125.504-1.125 1.125v3.5m7.5 10.375H9.375a1.125 1.125 0 0 1-1.125-1.125v-9.25m12 6.625v-1.875a3.375 3.375 0 0 0-3.375-3.375h-1.5a1.125 1.125 0 0 1-1.125-1.125v-1.5a3.375 3.375 0 0 0-3.375-3.375H9.75"
+ />
+</svg>
diff --git a/src/lib/components/layout/Sidebar.svelte b/src/lib/components/layout/Sidebar.svelte
index c746e343a..0bf00e472 100644
--- a/src/lib/components/layout/Sidebar.svelte
+++ b/src/lib/components/layout/Sidebar.svelte
@@ -22,7 +22,8 @@
getChatListByTagName,
updateChatById,
getAllChatTags,
- archiveChatById
+ archiveChatById,
+ cloneChatById
} from '$lib/apis/chats';
import { toast } from 'svelte-sonner';
import { fade, slide } from 'svelte/transition';
@@ -182,6 +183,18 @@
}
};
+ const cloneChatHandler = async (id) => {
+ const res = await cloneChatById(localStorage.token, id).catch((error) => {
+ toast.error(error);
+ return null;
+ });
+
+ if (res) {
+ goto(`/c/${res.id}`);
+ await chats.set(await getChatList(localStorage.token));
+ }
+ };
+
const saveSettings = async (updated) => {
await settings.set({ ...$settings, ...updated });
await updateUserSettings(localStorage.token, { ui: $settings });
@@ -601,6 +614,9 @@
<div class="flex self-center space-x-1 z-10">
<ChatMenu
chatId={chat.id}
+ cloneChatHandler={() => {
+ cloneChatHandler(chat.id);
+ }}
shareHandler={() => {
shareChatId = selectedChatId;
showShareChatModal = true;
diff --git a/src/lib/components/layout/Sidebar/ChatMenu.svelte b/src/lib/components/layout/Sidebar/ChatMenu.svelte
index a47502a41..db264f352 100644
--- a/src/lib/components/layout/Sidebar/ChatMenu.svelte
+++ b/src/lib/components/layout/Sidebar/ChatMenu.svelte
@@ -10,10 +10,12 @@
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';
const i18n = getContext('i18n');
export let shareHandler: Function;
+ export let cloneChatHandler: Function;
export let archiveChatHandler: Function;
export let renameHandler: Function;
export let deleteHandler: Function;
@@ -38,22 +40,12 @@
<div slot="content">
<DropdownMenu.Content
- class="w-full max-w-[160px] 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"
+ 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={() => {
- 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={() => {
@@ -64,6 +56,16 @@
<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={() => {
@@ -74,6 +76,16 @@
<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={() => {
diff --git a/src/lib/components/workspace/Models/ModelMenu.svelte b/src/lib/components/workspace/Models/ModelMenu.svelte
index 364893229..bde54e709 100644
--- a/src/lib/components/workspace/Models/ModelMenu.svelte
+++ b/src/lib/components/workspace/Models/ModelMenu.svelte
@@ -10,6 +10,7 @@
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';
const i18n = getContext('i18n');
@@ -60,20 +61,7 @@
cloneHandler();
}}
>
- <svg
- xmlns="http://www.w3.org/2000/svg"
- fill="none"
- viewBox="0 0 24 24"
- stroke-width="1.5"
- stroke="currentColor"
- class="w-4 h-4"
- >
- <path
- stroke-linecap="round"
- stroke-linejoin="round"
- d="M15.75 17.25v3.375c0 .621-.504 1.125-1.125 1.125h-9.75a1.125 1.125 0 0 1-1.125-1.125V7.875c0-.621.504-1.125 1.125-1.125H6.75a9.06 9.06 0 0 1 1.5.124m7.5 10.376h3.375c.621 0 1.125-.504 1.125-1.125V11.25c0-4.46-3.243-8.161-7.5-8.876a9.06 9.06 0 0 0-1.5-.124H9.375c-.621 0-1.125.504-1.125 1.125v3.5m7.5 10.375H9.375a1.125 1.125 0 0 1-1.125-1.125v-9.25m12 6.625v-1.875a3.375 3.375 0 0 0-3.375-3.375h-1.5a1.125 1.125 0 0 1-1.125-1.125v-1.5a3.375 3.375 0 0 0-3.375-3.375H9.75"
- />
- </svg>
+ <DocumentDuplicate />
<div class="flex items-center">{$i18n.t('Clone')}</div>
</DropdownMenu.Item>