From 1cc74ba885e059de01bede74addd56f5f5abaa28 Mon Sep 17 00:00:00 2001
From: Michael Sloan <michael@zed.dev>
Date: Fri, 19 Sep 2025 16:02:05 -0600
Subject: [PATCH] Add ZED_ZETA2 env var

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

diff --git a/crates/feature_flags/src/flags.rs b/crates/feature_flags/src/flags.rs
index 0e0b5d445c0dee453985344fe473ab63c37c728f..eaae9ce42519a8ebfd40d73d242536c5188fa3dd 100644
--- a/crates/feature_flags/src/flags.rs
+++ b/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";
-}
diff --git a/crates/zed/src/zed/edit_prediction_registry.rs b/crates/zed/src/zed/edit_prediction_registry.rs
index 2f824ab2ed24a72f1393c1d543fbb6ab950376f9..d13b93d28f2b1d43b30abd825952fb9a3548bc17 100644
--- a/crates/zed/src/zed/edit_prediction_registry.rs
+++ b/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(),