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;
         }