From 09afb7f3ee5f64b5a4a8d60e4768c066fcbee11c Mon Sep 17 00:00:00 2001
From: "zed-zippy[bot]" <234243425+zed-zippy[bot]@users.noreply.github.com>
Date: Mon, 1 Dec 2025 23:53:52 +0000
Subject: [PATCH] edit prediction: Do not attempt to gather context for
 non-zeta2 models (#43943) (cherry-pick to preview) (#43951)

Cherry-pick of #43943 to preview

----
We were running the LLM-based context gathering for zeta1 and sweep
which don't use it.

Release Notes:

- N/A

Co-authored-by: Agus Zubiaga <agus@zed.dev>
---
 crates/zeta/src/zeta.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/crates/zeta/src/zeta.rs b/crates/zeta/src/zeta.rs
index bdbd2b730714daee7bfad3cd0926ac7e01471b30..fe2d00b87b6343a9d3972fd4ceed41c0abc23c94 100644
--- a/crates/zeta/src/zeta.rs
+++ b/crates/zeta/src/zeta.rs
@@ -1966,6 +1966,10 @@ impl Zeta {
         cursor_position: language::Anchor,
         cx: &mut Context<Self>,
     ) {
+        if !matches!(self.edit_prediction_model, ZetaEditPredictionModel::Zeta2) {
+            return;
+        }
+
         if !matches!(&self.options().context, ContextMode::Agentic { .. }) {
             return;
         }