lean_ctx/core/config/
memory.rs

1//! RAM cleanup profile and memory-footprint presets (`config.toml`).
2
3use serde::{Deserialize, Serialize};
4
5use super::Config;
6
7/// Controls how aggressively lean-ctx frees memory when idle.
8/// - `aggressive`: (Default) Cache cleared after short idle period (5 min). Best for single-IDE use.
9/// - `shared`: Cache retained longer (30 min). Best when multiple IDEs/models share lean-ctx context.
10#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
11#[serde(rename_all = "lowercase")]
12pub enum MemoryCleanup {
13    #[default]
14    Aggressive,
15    Shared,
16}
17
18impl MemoryCleanup {
19    pub fn from_env() -> Option<Self> {
20        std::env::var("LEAN_CTX_MEMORY_CLEANUP").ok().and_then(|v| {
21            match v.trim().to_lowercase().as_str() {
22                "aggressive" => Some(Self::Aggressive),
23                "shared" => Some(Self::Shared),
24                _ => None,
25            }
26        })
27    }
28
29    pub fn effective(config: &Config) -> Self {
30        if let Some(env_val) = Self::from_env() {
31            return env_val;
32        }
33        config.memory_cleanup.clone()
34    }
35
36    /// Idle TTL in seconds before cache is auto-cleared.
37    pub fn idle_ttl_secs(&self) -> u64 {
38        match self {
39            Self::Aggressive => 300,
40            Self::Shared => 1800,
41        }
42    }
43
44    /// BM25 index eviction age multiplier (shared mode retains longer).
45    pub fn index_retention_multiplier(&self) -> f64 {
46        match self {
47            Self::Aggressive => 1.0,
48            Self::Shared => 3.0,
49        }
50    }
51}
52
53/// Controls RAM usage vs. feature richness trade-off.
54/// - `low`: Minimal RAM footprint, disables optional caches and embedding features
55/// - `balanced`: Default — moderate caches, single embedding engine
56/// - `performance`: Maximum caches, all features enabled
57#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
58#[serde(rename_all = "lowercase")]
59pub enum MemoryProfile {
60    Low,
61    Balanced,
62    #[default]
63    Performance,
64}
65
66impl MemoryProfile {
67    pub fn from_env() -> Option<Self> {
68        std::env::var("LEAN_CTX_MEMORY_PROFILE").ok().and_then(|v| {
69            match v.trim().to_lowercase().as_str() {
70                "low" => Some(Self::Low),
71                "balanced" => Some(Self::Balanced),
72                "performance" => Some(Self::Performance),
73                _ => None,
74            }
75        })
76    }
77
78    pub fn effective(config: &Config) -> Self {
79        if let Some(env_val) = Self::from_env() {
80            return env_val;
81        }
82        config.memory_profile.clone()
83    }
84
85    pub fn bm25_max_cache_mb(&self) -> u64 {
86        match self {
87            Self::Low => 64,
88            Self::Balanced => 128,
89            Self::Performance => 512,
90        }
91    }
92
93    pub fn semantic_cache_enabled(&self) -> bool {
94        !matches!(self, Self::Low)
95    }
96
97    pub fn embeddings_enabled(&self) -> bool {
98        !matches!(self, Self::Low)
99    }
100}
101
102/// Controls visibility of token savings footers in tool output.
103///
104/// - `always` (default): shown on every compressed response
105/// - `never`: suppressed everywhere
106/// - `auto`: legacy compatibility mode; behavior is transport/context dependent
107///
108/// Also controllable via `LEAN_CTX_SHOW_SAVINGS=1|0` (overrides this setting).
109#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
110#[serde(rename_all = "lowercase")]
111pub enum SavingsFooter {
112    Auto,
113    Always,
114    #[default]
115    Never,
116}
117
118impl SavingsFooter {
119    pub fn from_env() -> Option<Self> {
120        std::env::var("LEAN_CTX_SAVINGS_FOOTER").ok().and_then(|v| {
121            match v.trim().to_lowercase().as_str() {
122                "auto" => Some(Self::Auto),
123                "always" => Some(Self::Always),
124                "never" => Some(Self::Never),
125                _ => None,
126            }
127        })
128    }
129
130    pub fn effective() -> Self {
131        if let Some(env_val) = Self::from_env() {
132            return env_val;
133        }
134        let cfg = super::Config::load();
135        cfg.savings_footer.clone()
136    }
137}
138
139/// RSS-based memory guardian configuration.
140pub struct MemoryGuardConfig {
141    pub max_ram_percent: u8,
142}
143
144impl MemoryGuardConfig {
145    pub fn effective(config: &Config) -> Self {
146        let pct = std::env::var("LEAN_CTX_MAX_RAM_PERCENT")
147            .ok()
148            .and_then(|v| v.parse::<u8>().ok())
149            .unwrap_or(config.max_ram_percent)
150            .clamp(1, 50);
151        Self {
152            max_ram_percent: pct,
153        }
154    }
155}
156
157#[cfg(test)]
158mod tests {
159    use super::*;
160
161    #[test]
162    fn savings_footer_defaults_to_never() {
163        assert_eq!(SavingsFooter::default(), SavingsFooter::Never);
164    }
165
166    #[test]
167    fn savings_footer_from_env_accepts_auto() {
168        let _guard = crate::core::data_dir::test_env_lock();
169        std::env::set_var("LEAN_CTX_SAVINGS_FOOTER", "auto");
170        assert_eq!(SavingsFooter::from_env(), Some(SavingsFooter::Auto));
171        std::env::remove_var("LEAN_CTX_SAVINGS_FOOTER");
172    }
173}
lean_ctx/core/config/memory.rs

lean_ctx/core/config/
memory.rs
