From 73458a074cda816df2306e3895369f36ad5b13af Mon Sep 17 00:00:00 2001
From: Ben Kunkle <ben@zed.dev>
Date: Fri, 16 Jan 2026 09:55:39 -0600
Subject: [PATCH] Fix context enabled race condition in edit prediction cli
 (#46997)

Closes #ISSUE

Release Notes:

- N/A *or* Added/Fixed/Improved ...
---
 crates/edit_prediction/src/edit_prediction.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/crates/edit_prediction/src/edit_prediction.rs b/crates/edit_prediction/src/edit_prediction.rs
index 1b1834b39f79ee5f71bcd240a8df54d249406fe4..406fd51bac5679429805f07dec6d7b288668bf15 100644
--- a/crates/edit_prediction/src/edit_prediction.rs
+++ b/crates/edit_prediction/src/edit_prediction.rs
@@ -2198,6 +2198,9 @@ impl EditPredictionStore {
     }
 
     fn configure_context_retrieval(&mut self, cx: &mut Context<'_, EditPredictionStore>) {
+        if cfg!(feature = "cli-support") {
+            return;
+        }
         self.use_context = cx.has_flag::<Zeta2FeatureFlag>()
             && all_language_settings(None, cx).edit_predictions.use_context;
     }