diff --git a/src/lib/components/chat/Chat.svelte b/src/lib/components/chat/Chat.svelte
index 8d64ecbad..fab6f3c68 100644
--- a/src/lib/components/chat/Chat.svelte
+++ b/src/lib/components/chat/Chat.svelte
@@ -1569,6 +1569,15 @@
 
 			if (res) {
 				taskId = null;
+
+				const responseMessage = history.messages[history.currentId];
+				responseMessage.done = true;
+
+				history.messages[history.currentId] = responseMessage;
+
+				if (autoScroll) {
+					scrollToBottom();
+				}
 			}
 		}
 	};