From 0a656c276e529b73f4aa940da8eec346426a4648 Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Wed, 23 Apr 2025 14:02:03 -0400 Subject: [PATCH] language_models: Pass up `mode` for completion requests through Zed (#29294) This PR makes it so we pass up the `mode` for completion requests through the Zed provider. Release Notes: - N/A --- Cargo.lock | 4 ++-- Cargo.toml | 2 +- crates/language_models/src/provider/cloud.rs | 5 ++++- 3 files changed, 7 insertions(+), 4 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index 7177f82392c2e60c8701bcd5d7ddf8adf64c4130..04b4bc67b751b78bc80ddeb03e0025a32d4af3cd 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -18550,9 +18550,9 @@ dependencies = [ [[package]] name = "zed_llm_client" -version = "0.6.1" +version = "0.7.0" source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "ad17428120f5ca776dc5195e2411a282f5150a26d5536671f8943c622c31274f" +checksum = "3c1666cd923c5eb4635f3743e69c6920d0ed71f29b26920616a5d220607df7c4" dependencies = [ "anyhow", "serde", diff --git a/Cargo.toml b/Cargo.toml index db780bbe87d3ad8acaee9ecb91e8474220ff3532..1f298aa75b68922df2c10adcece233985978d9bd 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -606,7 +606,7 @@ wasmtime-wasi = "29" which = "6.0.0" wit-component = "0.221" workspace-hack = "0.1.0" -zed_llm_client = "0.6.1" +zed_llm_client = "0.7.0" zstd = "0.11" metal = "0.29" diff --git a/crates/language_models/src/provider/cloud.rs b/crates/language_models/src/provider/cloud.rs index 23aa00849e1cc1aee50169c982d0286f3beed0e6..6aeb7862115f7879f82ecc253dccb5d0df0afd12 100644 --- a/crates/language_models/src/provider/cloud.rs +++ b/crates/language_models/src/provider/cloud.rs @@ -35,7 +35,7 @@ use strum::IntoEnumIterator; use thiserror::Error; use ui::{TintColor, prelude::*}; use zed_llm_client::{ - CURRENT_PLAN_HEADER_NAME, CompletionBody, EXPIRED_LLM_TOKEN_HEADER_NAME, + CURRENT_PLAN_HEADER_NAME, CompletionBody, CompletionMode, EXPIRED_LLM_TOKEN_HEADER_NAME, MAX_LLM_MONTHLY_SPEND_REACHED_HEADER_NAME, MODEL_REQUESTS_RESOURCE_HEADER_VALUE, SUBSCRIPTION_LIMIT_RESOURCE_HEADER_NAME, }; @@ -736,6 +736,7 @@ impl LanguageModel for CloudLanguageModel { CompletionBody { thread_id, prompt_id, + mode: Some(CompletionMode::Max), provider: zed_llm_client::LanguageModelProvider::Anthropic, model: request.model.clone(), provider_request: serde_json::to_value(&request)?, @@ -782,6 +783,7 @@ impl LanguageModel for CloudLanguageModel { CompletionBody { thread_id, prompt_id, + mode: Some(CompletionMode::Max), provider: zed_llm_client::LanguageModelProvider::OpenAi, model: request.model.clone(), provider_request: serde_json::to_value(&request)?, @@ -812,6 +814,7 @@ impl LanguageModel for CloudLanguageModel { CompletionBody { thread_id, prompt_id, + mode: Some(CompletionMode::Max), provider: zed_llm_client::LanguageModelProvider::Google, model: request.model.clone(), provider_request: serde_json::to_value(&request)?,