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