agent: Log errors from token counting APIs (#29626)

Release Notes:

- N/A

crates/agent/src/active_thread.rs

@@ -702,7 +702,7 @@ struct EditMessageState {
     editor: Entity<Editor>,
     last_estimated_token_count: Option<usize>,
     _subscription: Subscription,
-    _update_token_count_task: Option<Task<anyhow::Result<()>>>,
+    _update_token_count_task: Option<Task<()>>,
 }
 
 impl ActiveThread {
@@ -1268,7 +1268,8 @@ impl ActiveThread {
                     .await;
             }
 
-            let token_count = if let Some(task) = cx.update(|cx| {
+            let token_count = if let Some(task) = cx
+                .update(|cx| {
                     let Some(message) = thread.read(cx).message(message_id) else {
                         log::error!("Message that was being edited no longer exists");
                         return None;
@@ -1306,12 +1307,16 @@ impl ActiveThread {
                     };
 
                     Some(configured_model.model.count_tokens(request, cx))
-            })? {
+                })
-                task.await?
+                .ok()
+                .flatten()
+            {
+                task.await.log_err()
             } else {
-                0
+                Some(0)
             };
 
+            if let Some(token_count) = token_count {
                 this.update(cx, |this, cx| {
                     let Some((_message_id, state)) = this.editing_message.as_mut() else {
                         return;
@@ -1320,6 +1325,8 @@ impl ActiveThread {
                     state.last_estimated_token_count = Some(token_count);
                     cx.emit(ActiveThreadEvent::EditingMessageTokenCountChanged);
                 })
+                .ok();
+            };
         }));
     }
 

crates/agent/src/message_editor.rs

@@ -63,7 +63,7 @@ pub struct MessageEditor {
     edits_expanded: bool,
     editor_is_expanded: bool,
     last_estimated_token_count: Option<usize>,
-    update_token_count_task: Option<Task<anyhow::Result<()>>>,
+    update_token_count_task: Option<Task<()>>,
     _subscriptions: Vec<Subscription>,
 }
 
@@ -1088,7 +1088,8 @@ impl MessageEditor {
                     .await;
             }
 
-            let token_count = if let Some(task) = this.update(cx, |this, cx| {
+            let token_count = if let Some(task) = this
+                .update(cx, |this, cx| {
                     let loaded_context = this
                         .last_loaded_context
                         .as_ref()
@@ -1128,17 +1129,23 @@ impl MessageEditor {
                     };
 
                     Some(model.model.count_tokens(request, cx))
-            })? {
+                })
-                task.await?
+                .ok()
+                .flatten()
+            {
+                task.await.log_err()
             } else {
-                0
+                Some(0)
             };
 
+            if let Some(token_count) = token_count {
                 this.update(cx, |this, cx| {
                     this.last_estimated_token_count = Some(token_count);
                     cx.emit(MessageEditorEvent::EstimatedTokenCount);
                     this.update_token_count_task.take();
                 })
+                .ok();
+            }
         }));
     }
 }