From 8b910e1cd9dfe7bf4148390a982fb8a1d65f35e1 Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Fri, 25 Apr 2025 16:43:43 -0400 Subject: [PATCH] collab: Add LLM request overage columns to `billing_preferences` (#29446) This PR adds two new columns to the `billing_preferences` table to allow users to opt in to overages on LLM requests. Release Notes: - N/A --- ...01930_add_model_request_overages_to_billing_preferences.sql | 3 +++ crates/collab/src/db/tables/billing_preference.rs | 2 ++ 2 files changed, 5 insertions(+) create mode 100644 crates/collab/migrations/20250425201930_add_model_request_overages_to_billing_preferences.sql diff --git a/crates/collab/migrations/20250425201930_add_model_request_overages_to_billing_preferences.sql b/crates/collab/migrations/20250425201930_add_model_request_overages_to_billing_preferences.sql new file mode 100644 index 0000000000..86e35c9202 --- /dev/null +++ b/crates/collab/migrations/20250425201930_add_model_request_overages_to_billing_preferences.sql @@ -0,0 +1,3 @@ +alter table billing_preferences + add column model_request_overages_enabled bool not null default false, + add column model_request_overages_spend_limit_in_cents integer not null default 0; diff --git a/crates/collab/src/db/tables/billing_preference.rs b/crates/collab/src/db/tables/billing_preference.rs index 0ad92c25d6..c1888d3b2f 100644 --- a/crates/collab/src/db/tables/billing_preference.rs +++ b/crates/collab/src/db/tables/billing_preference.rs @@ -9,6 +9,8 @@ pub struct Model { pub created_at: DateTime, pub user_id: UserId, pub max_monthly_llm_usage_spending_in_cents: i32, + pub model_request_overages_enabled: bool, + pub model_request_overages_spend_limit_in_cents: i32, } #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]