diff --git a/crates/agent/src/message_editor.rs b/crates/agent/src/message_editor.rs
index 43799af385a9ec32f26648e5cf529b8b867d5b7c..938ebf4cc05b0ab51c1192267c1e3296ddd98c73 100644
--- a/crates/agent/src/message_editor.rs
+++ b/crates/agent/src/message_editor.rs
@@ -26,7 +26,10 @@ use gpui::{
     Task, TextStyle, WeakEntity, linear_color_stop, linear_gradient, point, pulsating_between,
 };
 use language::{Buffer, Language};
-use language_model::{ConfiguredModel, LanguageModelRequestMessage, MessageContent, RequestUsage};
+use language_model::{
+    ConfiguredModel, LanguageModelRequestMessage, MessageContent, RequestUsage,
+    ZED_CLOUD_PROVIDER_ID,
+};
 use language_model_selector::ToggleModelSelector;
 use multi_buffer;
 use project::Project;
@@ -1071,6 +1074,17 @@ impl MessageEditor {
             return None;
         }
 
+        let is_using_zed_provider = self
+            .thread
+            .read(cx)
+            .configured_model()
+            .map_or(false, |model| {
+                model.provider.id().0 == ZED_CLOUD_PROVIDER_ID
+            });
+        if !is_using_zed_provider {
+            return None;
+        }
+
         let user_store = self.user_store.read(cx);
 
         let ubb_enable = user_store