From 2301c5f9f0af1b29bc35aa030e38be9b420c587d Mon Sep 17 00:00:00 2001
From: Max Brunsfeld <maxbrunsfeld@gmail.com>
Date: Fri, 23 Jan 2026 15:40:49 -0800
Subject: [PATCH] Send EP trigger as part of zeta2 prediction request (#47523)

Release Notes:

- N/A
---
 crates/cloud_llm_client/src/predict_edits_v3.rs | 3 +++
 crates/edit_prediction/src/edit_prediction.rs   | 2 ++
 crates/edit_prediction/src/zeta2.rs             | 2 ++
 3 files changed, 7 insertions(+)

diff --git a/crates/cloud_llm_client/src/predict_edits_v3.rs b/crates/cloud_llm_client/src/predict_edits_v3.rs
index 4b64813590d0dc00350d3fc5856540d378864eb9..c0bd3bc3204f06eb85fc1e6db16095a3d0af3f44 100644
--- a/crates/cloud_llm_client/src/predict_edits_v3.rs
+++ b/crates/cloud_llm_client/src/predict_edits_v3.rs
@@ -1,3 +1,4 @@
+use crate::PredictEditsRequestTrigger;
 use serde::{Deserialize, Serialize};
 use std::borrow::Cow;
 
@@ -20,6 +21,8 @@ pub struct PredictEditsV3Request {
     pub model: Option<String>,
     #[serde(default)]
     pub prompt_version: zeta_prompt::ZetaVersion,
+    #[serde(default)]
+    pub trigger: PredictEditsRequestTrigger,
 }
 
 #[derive(Debug, Deserialize, Serialize)]
diff --git a/crates/edit_prediction/src/edit_prediction.rs b/crates/edit_prediction/src/edit_prediction.rs
index 6a04ab62bc94257cbeb8618a7eadd85970f20a5b..ec080ae69d4c02f5914be5ce8f268299d1ce15a5 100644
--- a/crates/edit_prediction/src/edit_prediction.rs
+++ b/crates/edit_prediction/src/edit_prediction.rs
@@ -1923,6 +1923,7 @@ impl EditPredictionStore {
         client: Arc<Client>,
         llm_token: LlmApiToken,
         app_version: Version,
+        trigger: PredictEditsRequestTrigger,
     ) -> Result<(PredictEditsV3Response, Option<EditPredictionUsage>)> {
         let url = client
             .http_client()
@@ -1932,6 +1933,7 @@ impl EditPredictionStore {
             input,
             model: EDIT_PREDICTIONS_MODEL_ID.clone(),
             prompt_version,
+            trigger,
         };
 
         Self::send_api_request(
diff --git a/crates/edit_prediction/src/zeta2.rs b/crates/edit_prediction/src/zeta2.rs
index 4d307cd7c78f2dec9abfa9053410cdf04d10d1d7..9eefe1e38e1fcf17db575d92a62dc37bbcf5eb1e 100644
--- a/crates/edit_prediction/src/zeta2.rs
+++ b/crates/edit_prediction/src/zeta2.rs
@@ -34,6 +34,7 @@ pub fn request_prediction_with_zeta2(
         related_files,
         events,
         debug_tx,
+        trigger,
         ..
     }: EditPredictionModelInput,
     zeta_version: ZetaVersion,
@@ -110,6 +111,7 @@ pub fn request_prediction_with_zeta2(
                     client,
                     llm_token,
                     app_version,
+                    trigger,
                 )
                 .await?;