Send EP trigger as part of zeta2 prediction request (#47523)

Max Brunsfeld created 2 weeks ago

Release Notes:

- N/A

Change summary

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(+)

Detailed changes

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)]

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(

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