collab: Use new subscription usage tables (#29848)

Marshall Bowers created 7 months ago

This PR updates Collab to use the new subscription usage tables added in
#29847.

Release Notes:

- N/A

Change summary

crates/collab/src/llm/db/queries/subscription_usages.rs     | 2 +-
crates/collab/src/llm/db/tables/subscription_usage.rs       | 4 ++--
crates/collab/src/llm/db/tables/subscription_usage_meter.rs | 4 ++--
3 files changed, 5 insertions(+), 5 deletions(-)

Detailed changes

crates/collab/src/llm/db/queries/subscription_usages.rs

@@ -69,7 +69,7 @@ impl LlmDatabase {
 
         Ok(
             subscription_usage::Entity::insert(subscription_usage::ActiveModel {
-                id: ActiveValue::not_set(),
+                id: ActiveValue::set(Uuid::now_v7()),
                 user_id: ActiveValue::set(user_id),
                 period_start_at: ActiveValue::set(period_start_at),
                 period_end_at: ActiveValue::set(period_end_at),

crates/collab/src/llm/db/tables/subscription_usage.rs

@@ -4,10 +4,10 @@ use sea_orm::entity::prelude::*;
 use time::PrimitiveDateTime;
 
 #[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
-#[sea_orm(table_name = "subscription_usages")]
+#[sea_orm(table_name = "subscription_usages_v2")]
 pub struct Model {
     #[sea_orm(primary_key)]
-    pub id: i32,
+    pub id: Uuid,
     pub user_id: UserId,
     pub period_start_at: PrimitiveDateTime,
     pub period_end_at: PrimitiveDateTime,

crates/collab/src/llm/db/tables/subscription_usage_meter.rs

@@ -4,10 +4,10 @@ use serde::Serialize;
 use crate::llm::db::ModelId;
 
 #[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
-#[sea_orm(table_name = "subscription_usage_meters")]
+#[sea_orm(table_name = "subscription_usage_meters_v2")]
 pub struct Model {
     #[sea_orm(primary_key)]
-    pub id: i32,
+    pub id: Uuid,
     pub subscription_usage_id: i32,
     pub model_id: ModelId,
     pub mode: CompletionMode,