From eaab7da2d83acd87016447f1121cc2da3f0fa6f3 Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Tue, 11 Feb 2025 12:28:13 -0500 Subject: [PATCH] zeta: Add ability to change predict edits URL via environment variable (#24668) This PR adds the ability to change the predict edits URL using the `ZED_PREDICT_EDITS_URL` environment variable. This allows for easily pointing Zed to a development version of the Cloudflare Worker. Release Notes: - N/A --- crates/zeta/src/zeta.rs | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/crates/zeta/src/zeta.rs b/crates/zeta/src/zeta.rs index 9365f1945d4520f3e1d1c7c5080d4b44dc4db011..f679bb6827242f0b9253c421e3377b78c88c6551 100644 --- a/crates/zeta/src/zeta.rs +++ b/crates/zeta/src/zeta.rs @@ -664,11 +664,17 @@ and then another let mut did_retry = false; loop { - let request_builder = 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))