From cceb13b7cdcf0a9343c38870c6f23fc0855e7780 Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Sun, 4 May 2025 12:08:43 -0400 Subject: [PATCH] collab: Add `use_llm_request_queue` to `LlmTokenClaims` (#29877) This PR adds a `use_llm_request_queue` field to the LLM token claims, based on the `llm-request-queue` feature flag. Release Notes: - N/A --- crates/collab/src/llm/token.rs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/crates/collab/src/llm/token.rs b/crates/collab/src/llm/token.rs index 9387c4d3f86c5632fdacd6b25b6d75b80ce54935..bf510c9c0d7e41a92239bc66aed258928bb48a99 100644 --- a/crates/collab/src/llm/token.rs +++ b/crates/collab/src/llm/token.rs @@ -30,6 +30,8 @@ pub struct LlmTokenClaims { pub has_llm_closed_beta_feature_flag: bool, pub bypass_account_age_check: bool, pub has_llm_subscription: bool, + #[serde(default)] + pub use_llm_request_queue: bool, pub max_monthly_spend_in_cents: u32, pub custom_llm_monthly_allowance_in_cents: Option, #[serde(default)] @@ -93,6 +95,7 @@ impl LlmTokenClaims { .custom_llm_monthly_allowance_in_cents .map(|allowance| allowance as u32), use_new_billing: feature_flags.iter().any(|flag| flag == "new-billing"), + use_llm_request_queue: feature_flags.iter().any(|flag| flag == "llm-request-queue"), plan: subscription .as_ref() .and_then(|subscription| subscription.kind)