collab: Treat staff as having usage-based pricing enabled (#30020)

Marshall Bowers created 7 months ago

This PR makes it so staff are treated as having opted-in to usage-based
pricing.

Release Notes:

- N/A

Change summary

crates/collab/src/rpc.rs | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)

Detailed changes

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,