From 26f77032a2f7e87cbc44c886f185883930d508b3 Mon Sep 17 00:00:00 2001
From: Agus Zubiaga <agus@zed.dev>
Date: Mon, 1 Dec 2025 18:14:00 -0300
Subject: [PATCH] edit prediction: Do not attempt to gather context for
 non-zeta2 models (#43943)

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

Release Notes:

- N/A
---
 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 b54016eb8350a813dcb9ab324b192c11d6e5003f..909f21200cc7c055adb80b1e510e6f13e7fc9784 100644
--- a/crates/zeta/src/zeta.rs
+++ b/crates/zeta/src/zeta.rs
@@ -2063,6 +2063,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;
         }