feat: stop continue regeneration

timgerkb · Feb 16, 2024 · 5bcbb78 · 5bcbb78
1 parent 30f0f34
commit 5bcbb78
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte
@@ -65,7 +65,7 @@
 		await tick();
 
 		if (tooltipInstance) {
-			tooltipInstance[0].destroy();
+			tooltipInstance[0]?.destroy();
 		}
 
 		renderLatex();

diff --git a/src/routes/(app)/+page.svelte b/src/routes/(app)/+page.svelte
@@ -709,6 +709,9 @@
 
 		if (messages.length != 0 && messages.at(-1).done == true) {
 			const responseMessage = history.messages[history.currentId];
+			responseMessage.done = false;
+			await tick();
+
 			const modelTag = $models.filter((m) => m.name === responseMessage.model).at(0);
 
 			if (modelTag?.external) {

diff --git a/src/routes/(app)/c/[id]/+page.svelte b/src/routes/(app)/c/[id]/+page.svelte
@@ -18,7 +18,7 @@
 	} from '$lib/stores';
 	import { copyToClipboard, splitStream, convertMessagesToHistory } from '$lib/utils';
 
-	import { generateChatCompletion, generateTitle } from '$lib/apis/ollama';
+	import { generateChatCompletion, generateTitle, cancelChatCompletion } from '$lib/apis/ollama';
 	import {
 		addTagById,
 		createNewChat,
@@ -711,6 +711,9 @@
 
 		if (messages.length != 0 && messages.at(-1).done == true) {
 			const responseMessage = history.messages[history.currentId];
+			responseMessage.done = false;
+			await tick();
+
 			const modelTag = $models.filter((m) => m.name === responseMessage.model).at(0);
 
 			if (modelTag?.external) {