From 6c702d86ee5a5574afbdc796f15ab7fa71aceea3 Mon Sep 17 00:00:00 2001
From: Saoud Rizwan <7799382+saoudrizwan@users.noreply.github.com>
Date: Sat, 28 Sep 2024 19:38:47 -0400
Subject: [PATCH] Bug fixes

---
 src/core/ClaudeDev.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/core/ClaudeDev.ts b/src/core/ClaudeDev.ts
index bdc60b5..4b43588 100644
--- a/src/core/ClaudeDev.ts
+++ b/src/core/ClaudeDev.ts
@@ -2179,6 +2179,7 @@ ${this.customInstructions.trim()}
 
 		*/
 
+		this.presentAssistantMessageLocked = false
 		if (!block.partial) {
 			// block is finished streaming and executing
 			if (this.currentStreamingContentIndex === this.assistantMessageContent.length - 1) {
@@ -2193,7 +2194,7 @@ ${this.customInstructions.trim()}
 			if (this.currentStreamingContentIndex < this.assistantMessageContent.length) {
 				// there are already more content blocks to stream, so we'll call this function ourselves
 				// await this.presentAssistantContent()
-				this.presentAssistantMessageLocked = false
+
 				this.presentAssistantMessage()
 				return
 			}
@@ -2202,7 +2203,6 @@ ${this.customInstructions.trim()}
 		if (this.presentAssistantMessageHasPendingUpdates) {
 			this.presentAssistantMessage()
 		}
-		this.presentAssistantMessageLocked = false
 	}
 
 	// streaming