diff --git a/crates/inline_completion_button/src/inline_completion_button.rs b/crates/inline_completion_button/src/inline_completion_button.rs
index fd25823307bdf6..3c10b0d3503ade 100644
--- a/crates/inline_completion_button/src/inline_completion_button.rs
+++ b/crates/inline_completion_button/src/inline_completion_button.rs
@@ -405,7 +405,11 @@ impl InlineCompletionButton {
         let fs = self.fs.clone();
         let line_height = window.line_height();
 
-        if let Some(provider) = self.edit_prediction_provider.as_ref() {
+        if let Some(provider) = self
+            .edit_prediction_provider
+            .as_ref()
+            .filter(|p| p.name() == zeta::ZED_PREDICT_PROVIDER_NAME)
+        {
             let usage = provider.usage(cx).or_else(|| {
                 let user_store = self.user_store.read(cx);
 
diff --git a/crates/zeta/src/zeta.rs b/crates/zeta/src/zeta.rs
index eb82e97ec1c034..330072cfa1f3ff 100644
--- a/crates/zeta/src/zeta.rs
+++ b/crates/zeta/src/zeta.rs
@@ -64,6 +64,7 @@ const EDITABLE_REGION_START_MARKER: &'static str = "<|editable_region_start|>";
 const EDITABLE_REGION_END_MARKER: &'static str = "<|editable_region_end|>";
 const BUFFER_CHANGE_GROUPING_INTERVAL: Duration = Duration::from_secs(1);
 const ZED_PREDICT_DATA_COLLECTION_CHOICE: &str = "zed_predict_data_collection_choice";
+pub const ZED_PREDICT_PROVIDER_NAME: &str = "zed-predict";
 
 const MAX_CONTEXT_TOKENS: usize = 150;
 const MAX_REWRITE_TOKENS: usize = 350;
@@ -1383,7 +1384,7 @@ impl ZetaInlineCompletionProvider {
 
 impl inline_completion::EditPredictionProvider for ZetaInlineCompletionProvider {
     fn name() -> &'static str {
-        "zed-predict"
+        ZED_PREDICT_PROVIDER_NAME
     }
 
     fn display_name() -> &'static str {