diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte
index 1d99901481..2230a0bb7a 100644
--- a/src/lib/components/chat/Messages/ResponseMessage.svelte
+++ b/src/lib/components/chat/Messages/ResponseMessage.svelte
@@ -478,6 +478,7 @@
 									<ContentRenderer
 										id={message.id}
 										content={message.content}
+										floatingButtons={message?.done}
 										save={true}
 										{model}
 										on:update={(e) => {