systemprompt-models 0.1.22

Shared data models and types for systemprompt.io OS
Documentation 
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

//! Rate limit configuration.

use crate::auth::RateLimitTier;
use crate::profile::{RateLimitsConfig, TierMultipliers};

#[derive(Debug, Clone, Copy)]
pub struct RateLimitConfig {
    pub oauth_public_per_second: u64,
    pub oauth_auth_per_second: u64,
    pub contexts_per_second: u64,
    pub tasks_per_second: u64,
    pub artifacts_per_second: u64,
    pub agent_registry_per_second: u64,
    pub agents_per_second: u64,
    pub mcp_registry_per_second: u64,
    pub mcp_per_second: u64,
    pub stream_per_second: u64,
    pub content_per_second: u64,
    pub burst_multiplier: u64,
    pub disabled: bool,
    pub tier_multipliers: TierMultipliers,
}

impl Default for RateLimitConfig {
    fn default() -> Self {
        Self {
            oauth_public_per_second: 10,
            oauth_auth_per_second: 10,
            contexts_per_second: 100,
            tasks_per_second: 50,
            artifacts_per_second: 50,
            agent_registry_per_second: 50,
            agents_per_second: 20,
            mcp_registry_per_second: 50,
            mcp_per_second: 200,
            stream_per_second: 100,
            content_per_second: 50,
            burst_multiplier: 3,
            disabled: false,
            tier_multipliers: TierMultipliers::default(),
        }
    }
}

impl RateLimitConfig {
    pub fn production() -> Self {
        Self::default()
    }

    pub fn testing() -> Self {
        Self {
            oauth_public_per_second: 10000,
            oauth_auth_per_second: 10000,
            contexts_per_second: 10000,
            tasks_per_second: 10000,
            artifacts_per_second: 10000,
            agent_registry_per_second: 10000,
            agents_per_second: 10000,
            mcp_registry_per_second: 10000,
            mcp_per_second: 10000,
            stream_per_second: 10000,
            content_per_second: 10000,
            burst_multiplier: 100,
            disabled: false,
            tier_multipliers: TierMultipliers::default(),
        }
    }

    pub fn disabled() -> Self {
        let mut config = Self::testing();
        config.disabled = true;
        config
    }

    #[must_use]
    pub fn effective_limit(&self, base_rate: u64, tier: RateLimitTier) -> u64 {
        let multiplier = self.tier_multiplier(tier);
        let base_capped = u32::try_from(base_rate).unwrap_or(u32::MAX);
        let scaled = f64::from(base_capped) * multiplier;
        let clamped = scaled.clamp(1.0, f64::from(u32::MAX));
        #[allow(clippy::cast_possible_truncation, clippy::cast_sign_loss)]
        let result = clamped as u64;
        result
    }

    pub const fn tier_multiplier(&self, tier: RateLimitTier) -> f64 {
        match tier {
            RateLimitTier::Admin => self.tier_multipliers.admin,
            RateLimitTier::User => self.tier_multipliers.user,
            RateLimitTier::A2a => self.tier_multipliers.a2a,
            RateLimitTier::Mcp => self.tier_multipliers.mcp,
            RateLimitTier::Service => self.tier_multipliers.service,
            RateLimitTier::Anon => self.tier_multipliers.anon,
        }
    }
}

impl From<&RateLimitsConfig> for RateLimitConfig {
    fn from(config: &RateLimitsConfig) -> Self {
        Self {
            oauth_public_per_second: config.oauth_public_per_second,
            oauth_auth_per_second: config.oauth_auth_per_second,
            contexts_per_second: config.contexts_per_second,
            tasks_per_second: config.tasks_per_second,
            artifacts_per_second: config.artifacts_per_second,
            agent_registry_per_second: config.agent_registry_per_second,
            agents_per_second: config.agents_per_second,
            mcp_registry_per_second: config.mcp_registry_per_second,
            mcp_per_second: config.mcp_per_second,
            stream_per_second: config.stream_per_second,
            content_per_second: config.content_per_second,
            burst_multiplier: config.burst_multiplier,
            disabled: config.disabled,
            tier_multipliers: config.tier_multipliers,
        }
    }
}