From a378b3f300b9fc432714190d00af2744329f5010 Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Tue, 6 May 2025 12:06:03 -0400 Subject: [PATCH] collab: Treat staff as having usage-based pricing enabled (#30020) This PR makes it so staff are treated as having opted-in to usage-based pricing. Release Notes: - N/A --- crates/collab/src/rpc.rs | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/crates/collab/src/rpc.rs b/crates/collab/src/rpc.rs index 37cab3986152de0b2e6faa4eba1cce26005c705f..61daff975778e793db7ea911459f821821b113b4 100644 --- a/crates/collab/src/rpc.rs +++ b/crates/collab/src/rpc.rs @@ -2737,8 +2737,12 @@ async fn update_user_plan(user_id: UserId, session: &Session) -> Result<()> { trial_started_at: billing_customer .and_then(|billing_customer| billing_customer.trial_started_at) .map(|trial_started_at| trial_started_at.and_utc().timestamp() as u64), - is_usage_based_billing_enabled: billing_preferences - .map(|preferences| preferences.model_request_overages_enabled), + is_usage_based_billing_enabled: if session.is_staff() { + Some(true) + } else { + billing_preferences + .map(|preferences| preferences.model_request_overages_enabled) + }, usage: usage.map(|usage| { let plan = match plan { proto::Plan::Free => zed_llm_client::Plan::Free,