diff --git a/crates/zeta/src/zeta.rs b/crates/zeta/src/zeta.rs
index 16a5e81793..eb82e97ec1 100644
--- a/crates/zeta/src/zeta.rs
+++ b/crates/zeta/src/zeta.rs
@@ -740,13 +740,18 @@ and then another
             let mut did_retry = false;
 
             loop {
-                let request = http_client::Request::builder()
-                    .method(Method::POST)
-                    .uri(
-                        http_client
-                            .build_zed_llm_url("/predict_edits/v2", &[])?
-                            .as_ref(),
-                    )
+                let request_builder = http_client::Request::builder().method(Method::POST);
+                let request_builder =
+                    if let Ok(predict_edits_url) = std::env::var("ZED_PREDICT_EDITS_URL") {
+                        request_builder.uri(predict_edits_url)
+                    } else {
+                        request_builder.uri(
+                            http_client
+                                .build_zed_llm_url("/predict_edits/v2", &[])?
+                                .as_ref(),
+                        )
+                    };
+                let request = request_builder
                     .header("Content-Type", "application/json")
                     .header("Authorization", format!("Bearer {}", token))
                     .header(ZED_VERSION_HEADER_NAME, app_version.to_string())