Hugging Face's logo

Spaces:
ronghua
/
cursor
Running

App Files Community
cursor
File size: 2,703 Bytes 
3c2af29
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
 
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 email_verified: bool,
    pub name: String,
    #[serde(rename(serialize = "id"))]
    pub sub: String,
    pub updated_at: DateTime<Local>,
    // Image link, rendered in /logs?
    // pub picture: Option<String>,
}