From 1a4d7249f6cf59dc37f64c8f745f0dee47a4e9ce Mon Sep 17 00:00:00 2001
From: Michael Sloan <michael@zed.dev>
Date: Tue, 29 Apr 2025 17:03:18 -0600
Subject: [PATCH] agent: Make token indicator not flash if last count tokens
 request failed (#29628)

Release Notes:

- N/A
---
 crates/agent/src/message_editor.rs | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/crates/agent/src/message_editor.rs b/crates/agent/src/message_editor.rs
index e223b5821f..3116bec5a3 100644
--- a/crates/agent/src/message_editor.rs
+++ b/crates/agent/src/message_editor.rs
@@ -1138,14 +1138,14 @@ impl MessageEditor {
                 Some(0)
             };
 
-            if let Some(token_count) = token_count {
-                this.update(cx, |this, cx| {
+            this.update(cx, |this, cx| {
+                if let Some(token_count) = token_count {
                     this.last_estimated_token_count = Some(token_count);
                     cx.emit(MessageEditorEvent::EstimatedTokenCount);
-                    this.update_token_count_task.take();
-                })
-                .ok();
-            }
+                }
+                this.update_token_count_task.take();
+            })
+            .ok();
         }));
     }
 }