Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

zeta: Send staff edit predictions through llm.zed.dev again (#23996)

Browse source 
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
This commit is contained in:
Marshall Bowers 2025-01-30 21:07:38 -05:00 • committed by GitHub
parent 517e519bdc
commit 35fbe1ef3d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

19
crates/zeta/src/zeta.rs
View file

@ -602,7 +602,7 @@ and then another
fn perform_predict_edits(
client: Arc<Client>,
llm_token: LlmApiToken,
is_staff: bool,
_is_staff: bool,
body: PredictEditsParams,
) -> impl Future<Output = Result<PredictEditsResponse>> {
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))