diff --git a/src/lib/components/chat/Messages.svelte b/src/lib/components/chat/Messages.svelte index a4a5ffa2..9efab1e5 100644 --- a/src/lib/components/chat/Messages.svelte +++ b/src/lib/components/chat/Messages.svelte @@ -16,6 +16,7 @@ export let sendPrompt: Function; export let continueGeneration: Function; export let regenerateResponse: Function; + export let stopResponse: Function; export let processing = ''; export let bottomPadding = false; @@ -228,6 +229,7 @@ await updateChatById(localStorage.token, chatId, { history }); await chats.set(await getChatList(localStorage.token)); + stopResponse(); }; diff --git a/src/routes/(app)/+page.svelte b/src/routes/(app)/+page.svelte index 33409ec3..b93dc860 100644 --- a/src/routes/(app)/+page.svelte +++ b/src/routes/(app)/+page.svelte @@ -828,6 +828,7 @@ {sendPrompt} {continueGeneration} {regenerateResponse} + {stopResponse} /> diff --git a/src/routes/(app)/c/[id]/+page.svelte b/src/routes/(app)/c/[id]/+page.svelte index be84467d..737a13e0 100644 --- a/src/routes/(app)/c/[id]/+page.svelte +++ b/src/routes/(app)/c/[id]/+page.svelte @@ -858,6 +858,7 @@ {sendPrompt} {continueGeneration} {regenerateResponse} + {stopResponse} />