From 35fbe1ef3da79b09b1443e5901259a67e72f9e4c Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Thu, 30 Jan 2025 21:07:38 -0500 Subject: [PATCH] zeta: Send staff edit predictions through `llm.zed.dev` again (#23996) This PR changes the edit predictions URL for Zed Staff back to `llm.zed.dev/predict_edits`. This endpoint is now being routed to the Cloudflare Workers instead of the LLM service. Release Notes: - N/A --- crates/zeta/src/zeta.rs | 19 ++++++------------- 1 file changed, 6 insertions(+), 13 deletions(-) diff --git a/crates/zeta/src/zeta.rs b/crates/zeta/src/zeta.rs index ad53dcc757..17bc519eab 100644 --- a/crates/zeta/src/zeta.rs +++ b/crates/zeta/src/zeta.rs @@ -602,7 +602,7 @@ and then another fn perform_predict_edits( client: Arc, llm_token: LlmApiToken, - is_staff: bool, + _is_staff: bool, body: PredictEditsParams, ) -> impl Future> { async move { @@ -611,18 +611,11 @@ and then another let mut did_retry = false; loop { - let request_builder = http_client::Request::builder().method(Method::POST); - let request_builder = if is_staff { - request_builder.uri( - "https://llm-worker-production.zed-industries.workers.dev/predict_edits", - ) - } else { - request_builder.uri( - http_client - .build_zed_llm_url("/predict_edits", &[])? - .as_ref(), - ) - }; + let request_builder = http_client::Request::builder().method(Method::POST).uri( + http_client + .build_zed_llm_url("/predict_edits", &[])? + .as_ref(), + ); let request = request_builder .header("Content-Type", "application/json") .header("Authorization", format!("Bearer {}", token))