monthly_usage.rs

 1use crate::{db::UserId, llm::db::ModelId};
 2use sea_orm::entity::prelude::*;
 3
 4#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
 5#[sea_orm(table_name = "monthly_usages")]
 6pub struct Model {
 7    #[sea_orm(primary_key)]
 8    pub id: i32,
 9    pub user_id: UserId,
10    pub model_id: ModelId,
11    pub month: i32,
12    pub year: i32,
13    pub input_tokens: i64,
14    pub cache_creation_input_tokens: i64,
15    pub cache_read_input_tokens: i64,
16    pub output_tokens: i64,
17}
18
19#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
20pub enum Relation {}
21
22impl ActiveModelBehavior for ActiveModel {}