From 530868f5a124e9d11dcc194370f89ec39f79256c Mon Sep 17 00:00:00 2001
From: Aleksander Grygier <aleksander.grygier@gmail.com>
Date: Mon, 9 Feb 2026 15:01:05 +0100
Subject: [PATCH] feat: Enable MCP prompt button in chat message edit form

---
 .../src/lib/components/app/chat/ChatForm/ChatForm.svelte   | 7 +++++--
 .../ChatFormPromptPicker/ChatFormPromptPicker.svelte       | 3 +--
 .../app/chat/ChatMessages/ChatMessageEditForm.svelte       | 1 +
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/tools/server/webui/src/lib/components/app/chat/ChatForm/ChatForm.svelte b/tools/server/webui/src/lib/components/app/chat/ChatForm/ChatForm.svelte
index 74a943af22..46d5729190 100644
--- a/tools/server/webui/src/lib/components/app/chat/ChatForm/ChatForm.svelte
+++ b/tools/server/webui/src/lib/components/app/chat/ChatForm/ChatForm.svelte
@@ -365,8 +365,11 @@
 		promptInfo: MCPPromptInfo,
 		args?: Record<string, string>
 	) {
-		value = '';
-		onValueChange?.('');
+		// Only clear the value if the prompt was triggered by typing '/'
+		if (value.startsWith('/')) {
+			value = '';
+			onValueChange?.('');
+		}
 		isPromptPickerOpen = false;
 		promptSearchQuery = '';
 
diff --git a/tools/server/webui/src/lib/components/app/chat/ChatForm/ChatFormPromptPicker/ChatFormPromptPicker.svelte b/tools/server/webui/src/lib/components/app/chat/ChatForm/ChatFormPromptPicker/ChatFormPromptPicker.svelte
index 62453c8eda..1eb99aa161 100644
--- a/tools/server/webui/src/lib/components/app/chat/ChatForm/ChatFormPromptPicker/ChatFormPromptPicker.svelte
+++ b/tools/server/webui/src/lib/components/app/chat/ChatForm/ChatFormPromptPicker/ChatFormPromptPicker.svelte
@@ -350,7 +350,7 @@
 		}
 	}}
 >
-	<Popover.Trigger class="pointer-events-none absolute top-0 right-0 left-0 h-0 w-full opacity-0">
+	<Popover.Trigger class="pointer-events-none absolute inset-0 opacity-0">
 		<span class="sr-only">Open prompt picker</span>
 	</Popover.Trigger>
 
@@ -358,7 +358,6 @@
 		side="top"
 		align="start"
 		sideOffset={12}
-		avoidCollisions={false}
 		class="w-[var(--bits-popover-anchor-width)] max-w-none rounded-xl border-border/50 p-0 shadow-xl {className}"
 		onkeydown={handleKeydown}
 		onOpenAutoFocus={(e) => e.preventDefault()}
diff --git a/tools/server/webui/src/lib/components/app/chat/ChatMessages/ChatMessageEditForm.svelte b/tools/server/webui/src/lib/components/app/chat/ChatMessages/ChatMessageEditForm.svelte
index 5503ad1fab..0f0e53b81b 100644
--- a/tools/server/webui/src/lib/components/app/chat/ChatMessages/ChatMessageEditForm.svelte
+++ b/tools/server/webui/src/lib/components/app/chat/ChatMessages/ChatMessageEditForm.svelte
@@ -99,6 +99,7 @@
 		attachments={editCtx.editedExtras}
 		uploadedFiles={editCtx.editedUploadedFiles}
 		placeholder="Edit your message..."
+		showMcpPromptButton={true}
 		onValueChange={editCtx.setContent}
 		onAttachmentRemove={handleAttachmentRemove}
 		onUploadedFileRemove={handleUploadedFileRemove}