Fix incorrect usages of `Zeta2FeatureFlag` (#46462)

Ben Kunkle created 2 days ago

Closes #ISSUE

Release Notes:

- N/A *or* Added/Fixed/Improved ...

Change summary

crates/settings_ui/src/pages/edit_prediction_provider_setup.rs | 6 ++--
crates/zed/src/zed/edit_prediction_registry.rs                 | 6 ++-
2 files changed, 7 insertions(+), 5 deletions(-)

Detailed changes

crates/settings_ui/src/pages/edit_prediction_provider_setup.rs

@@ -1,5 +1,5 @@
 use edit_prediction::{
-    ApiKeyState, Zeta2FeatureFlag,
+    ApiKeyState, MercuryFeatureFlag, SweepFeatureFlag,
     mercury::{MERCURY_CREDENTIALS_URL, mercury_api_token},
     sweep_ai::{SWEEP_CREDENTIALS_URL, sweep_api_token},
 };
@@ -33,7 +33,7 @@ impl Render for EditPredictionSetupPage {
 
         let providers = [
             Some(render_github_copilot_provider(window, cx).into_any_element()),
-            cx.has_flag::<Zeta2FeatureFlag>().then(|| {
+            cx.has_flag::<MercuryFeatureFlag>().then(|| {
                 render_api_key_provider(
                     IconName::Inception,
                     "Mercury",
@@ -46,7 +46,7 @@ impl Render for EditPredictionSetupPage {
                 )
                 .into_any_element()
             }),
-            cx.has_flag::<Zeta2FeatureFlag>().then(|| {
+            cx.has_flag::<SweepFeatureFlag>().then(|| {
                 render_api_key_provider(
                     IconName::SweepAi,
                     "Sweep",

crates/zed/src/zed/edit_prediction_registry.rs

@@ -2,7 +2,9 @@ use client::{Client, UserStore};
 use codestral::CodestralEditPredictionDelegate;
 use collections::HashMap;
 use copilot::{Copilot, CopilotEditPredictionDelegate};
-use edit_prediction::{SweepFeatureFlag, ZedEditPredictionDelegate, Zeta2FeatureFlag};
+use edit_prediction::{
+    MercuryFeatureFlag, SweepFeatureFlag, ZedEditPredictionDelegate, Zeta2FeatureFlag,
+};
 use editor::Editor;
 use feature_flags::FeatureFlagAppExt;
 use gpui::{AnyWindowHandle, App, AppContext as _, Context, Entity, WeakEntity};
@@ -204,7 +206,7 @@ fn assign_edit_prediction_provider(
                         {
                             edit_prediction::EditPredictionModel::Zeta2
                         } else if name == EXPERIMENTAL_MERCURY_EDIT_PREDICTION_PROVIDER_NAME
-                            && cx.has_flag::<Zeta2FeatureFlag>()
+                            && cx.has_flag::<MercuryFeatureFlag>()
                         {
                             edit_prediction::EditPredictionModel::Mercury
                         } else {