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