From 74cafce4d26e9383cc7011c6fee0692bd3c18584 Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Tue, 8 Jul 2025 16:43:26 +0400
Subject: [PATCH] refac: note revision history

---
 src/lib/components/notes/NoteEditor.svelte      | 4 ++++
 src/lib/components/notes/NoteEditor/Chat.svelte | 7 +++++++
 2 files changed, 11 insertions(+)

diff --git a/src/lib/components/notes/NoteEditor.svelte b/src/lib/components/notes/NoteEditor.svelte
index 4f790e8517..8755056655 100644
--- a/src/lib/components/notes/NoteEditor.svelte
+++ b/src/lib/components/notes/NoteEditor.svelte
@@ -665,6 +665,7 @@ Provide the enhanced notes in markdown format. Use markdown syntax for headings,
 	};
 
 	const insertHandler = (content) => {
+		insertNoteVersion(note);
 		inputElement?.insertContent(content);
 	};
 
@@ -1075,6 +1076,9 @@ Provide the enhanced notes in markdown format. Use markdown syntax for headings,
 				{files}
 				onInsert={insertHandler}
 				onStop={stopResponseHandler}
+				insertNoteHandler={() => {
+					insertNoteVersion(note);
+				}}
 				scrollToBottomHandler={scrollToBottom}
 			/>
 		{:else if selectedPanel === 'settings'}
diff --git a/src/lib/components/notes/NoteEditor/Chat.svelte b/src/lib/components/notes/NoteEditor/Chat.svelte
index 6fb0a3893b..40a047c2b5 100644
--- a/src/lib/components/notes/NoteEditor/Chat.svelte
+++ b/src/lib/components/notes/NoteEditor/Chat.svelte
@@ -40,6 +40,8 @@
 
 	export let onInsert = (content) => {};
 	export let onStop = () => {};
+
+	export let insertNoteHandler = () => {};
 	export let scrollToBottomHandler = () => {};
 
 	let loaded = false;
@@ -253,6 +255,11 @@ Based on the user's instruction, update and enhance the existing notes by incorp
 			scrollToBottom();
 
 			loading = true;
+
+			if (editorEnabled) {
+				insertNoteHandler();
+			}
+
 			await chatCompletionHandler();
 			messages = messages.map((message) => {
 				message.done = true;