| use chrono::{DateTime, Local}; |
| use serde::{Deserialize, Serialize}; |
| use rkyv::{Archive, Deserialize as RkyvDeserialize, Serialize as RkyvSerialize}; |
|
|
| #[derive(Serialize)] |
| #[serde(untagged)] |
| pub enum GetUserInfo { |
| Usage(TokenProfile), |
| Error { error: String }, |
| } |
|
|
| #[derive(Serialize, Clone, Archive, RkyvDeserialize, RkyvSerialize)] |
| pub struct TokenProfile { |
| pub usage: UsageProfile, |
| pub user: UserProfile, |
| pub stripe: StripeProfile, |
| } |
|
|
| #[derive(Deserialize, Serialize, PartialEq, Clone, Archive, RkyvDeserialize, RkyvSerialize)] |
| pub enum MembershipType { |
| #[serde(rename = "free")] |
| Free, |
| #[serde(rename = "free_trial")] |
| FreeTrial, |
| #[serde(rename = "pro")] |
| Pro, |
| #[serde(rename = "enterprise")] |
| Enterprise, |
| } |
|
|
| #[derive(Deserialize, Serialize, Clone, Archive, RkyvDeserialize, RkyvSerialize)] |
| pub struct StripeProfile { |
| #[serde(rename(deserialize = "membershipType"))] |
| pub membership_type: MembershipType, |
| #[serde( |
| rename(deserialize = "paymentId"), |
| default, |
| skip_serializing_if = "Option::is_none" |
| )] |
| pub payment_id: Option<String>, |
| #[serde(rename(deserialize = "daysRemainingOnTrial"))] |
| pub days_remaining_on_trial: u32, |
| } |
|
|
| #[derive(Deserialize, Serialize, Clone, Archive, RkyvDeserialize, RkyvSerialize)] |
| pub struct ModelUsage { |
| #[serde(rename(deserialize = "numRequests", serialize = "requests"))] |
| pub num_requests: u32, |
| #[serde( |
| rename(deserialize = "numRequestsTotal"), |
| default, |
| skip_serializing_if = "Option::is_none" |
| )] |
| pub requests_total: Option<u32>, |
| #[serde(rename(deserialize = "numTokens", serialize = "tokens"))] |
| pub num_tokens: u32, |
| #[serde( |
| rename(deserialize = "maxRequestUsage"), |
| skip_serializing_if = "Option::is_none" |
| )] |
| pub max_requests: Option<u32>, |
| #[serde( |
| rename(deserialize = "maxTokenUsage"), |
| skip_serializing_if = "Option::is_none" |
| )] |
| pub max_tokens: Option<u32>, |
| } |
|
|
| #[derive(Deserialize, Serialize, Clone, Archive, RkyvDeserialize, RkyvSerialize)] |
| pub struct UsageProfile { |
| #[serde(rename(deserialize = "gpt-4"))] |
| pub premium: ModelUsage, |
| #[serde(rename(deserialize = "gpt-3.5-turbo"))] |
| pub standard: ModelUsage, |
| #[serde(rename(deserialize = "gpt-4-32k"))] |
| pub unknown: ModelUsage, |
| } |
|
|
| #[derive(Deserialize, Serialize, Clone, Archive, RkyvDeserialize, RkyvSerialize)] |
| pub struct UserProfile { |
| pub email: String, |
| |
| pub name: String, |
| #[serde(rename(serialize = "id"))] |
| pub sub: String, |
| pub updated_at: DateTime<Local>, |
| |
| |
| } |
|
|
