From 10507f9a4c4dcc0a8900e7dab618f0f34806d693 Mon Sep 17 00:00:00 2001 From: Marshall Bowers Date: Wed, 16 Apr 2025 14:36:49 -0400 Subject: [PATCH] collab: Add `plan` column to `subscription_usages` (#28889) This PR adds a `plan` column to the `subscription_usages` table. These tables don't have any records in them yet, so it's fine to make the column required without a default. Release Notes: - N/A --- .../20250416181354_add_plan_to_subscription_usages.sql | 4 ++++ crates/collab/src/llm/db/tables/subscription_usage.rs | 2 ++ 2 files changed, 6 insertions(+) create mode 100644 crates/collab/migrations_llm/20250416181354_add_plan_to_subscription_usages.sql diff --git a/crates/collab/migrations_llm/20250416181354_add_plan_to_subscription_usages.sql b/crates/collab/migrations_llm/20250416181354_add_plan_to_subscription_usages.sql new file mode 100644 index 0000000000..8d54c8b87c --- /dev/null +++ b/crates/collab/migrations_llm/20250416181354_add_plan_to_subscription_usages.sql @@ -0,0 +1,4 @@ +alter table subscription_usages + add column plan text not null; + +create index ix_subscription_usages_on_plan on subscription_usages (plan); diff --git a/crates/collab/src/llm/db/tables/subscription_usage.rs b/crates/collab/src/llm/db/tables/subscription_usage.rs index 33311d22f6..f9640416f6 100644 --- a/crates/collab/src/llm/db/tables/subscription_usage.rs +++ b/crates/collab/src/llm/db/tables/subscription_usage.rs @@ -1,4 +1,5 @@ use crate::db::UserId; +use crate::db::billing_subscription::SubscriptionKind; use sea_orm::entity::prelude::*; use time::PrimitiveDateTime; @@ -10,6 +11,7 @@ pub struct Model { pub user_id: UserId, pub period_start_at: PrimitiveDateTime, pub period_end_at: PrimitiveDateTime, + pub plan: SubscriptionKind, pub model_requests: i32, pub edit_predictions: i32, }