collab: Add `plan` column to `subscription_usages` (#28889)

Marshall Bowers created 9 months ago

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

Change summary

crates/collab/migrations_llm/20250416181354_add_plan_to_subscription_usages.sql | 4 
crates/collab/src/llm/db/tables/subscription_usage.rs                           | 2 
2 files changed, 6 insertions(+)

Detailed changes

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);

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,
 }