Add ZED_ZETA2 env var

Michael Sloan created 2 months ago

Change summary

crates/feature_flags/src/flags.rs              | 6 ------
crates/zed/src/zed/edit_prediction_registry.rs | 3 +--
2 files changed, 1 insertion(+), 8 deletions(-)

Detailed changes

crates/feature_flags/src/flags.rs

@@ -29,9 +29,3 @@ pub struct JjUiFeatureFlag {}
 impl FeatureFlag for JjUiFeatureFlag {
     const NAME: &'static str = "jj-ui";
 }
-
-pub struct EditPredictionV2FeatureFlag;
-
-impl FeatureFlag for EditPredictionV2FeatureFlag {
-    const NAME: &'static str = "edit-prediction-v2";
-}

crates/zed/src/zed/edit_prediction_registry.rs

@@ -2,7 +2,6 @@ use client::{Client, UserStore};
 use collections::HashMap;
 use copilot::{Copilot, CopilotCompletionProvider};
 use editor::Editor;
-use feature_flags::{EditPredictionV2FeatureFlag, FeatureFlagAppExt};
 use gpui::{AnyWindowHandle, App, AppContext as _, Context, Entity, WeakEntity};
 use language::language_settings::{EditPredictionProvider, all_language_settings};
 use settings::SettingsStore;
@@ -204,7 +203,7 @@ fn assign_edit_prediction_provider(
                     }
                 }
 
-                if cx.has_flag::<EditPredictionV2FeatureFlag>() {
+                if std::env::var("ZED_ZETA2").is_ok() {
                     let provider = cx.new(|cx| {
                         zeta2::ZetaEditPredictionProvider::new(
                             editor.project(),