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 0000000000000000000000000000000000000000..8d54c8b87ca820bd8aa46c7bf18ccd50ccf52807 --- /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 33311d22f6d2956299797e26c7cf3cdf4b87b7f9..f9640416f6830b174399750968311f0a5a20a813 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, }