23 lines
644 B
Rust
23 lines
644 B
Rust
use crate::{DateTimeUtc, UserId};
|
|
use sea_orm::entity::prelude::*;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize, DeriveEntityModel)]
|
|
#[sea_orm(table_name = "ai_token_usage")]
|
|
pub struct Model {
|
|
#[sea_orm(primary_key)]
|
|
pub id: Uuid,
|
|
pub user_id: UserId,
|
|
pub conversation_id: Option<Uuid>,
|
|
pub model: String,
|
|
pub input_tokens: i32,
|
|
pub output_tokens: i32,
|
|
pub cost_usd: Option<rust_decimal::Decimal>,
|
|
pub recorded_at: DateTimeUtc,
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
pub enum Relation {}
|
|
|
|
impl ActiveModelBehavior for ActiveModel {}
|