mem7_config/
lib.rs

1use serde::{Deserialize, Serialize};
2
3#[derive(Debug, Clone, Serialize, Deserialize)]
4pub struct LlmConfig {
5    #[serde(default = "default_llm_provider")]
6    pub provider: String,
7    pub base_url: String,
8    pub api_key: String,
9    pub model: String,
10    #[serde(default = "default_temperature")]
11    pub temperature: f32,
12    #[serde(default = "default_max_tokens")]
13    pub max_tokens: u32,
14    /// When true, image messages are sent to the LLM to produce text
15    /// descriptions that are then stored as memory content.
16    #[serde(default)]
17    pub enable_vision: bool,
18}
19
20fn default_llm_provider() -> String {
21    "openai".into()
22}
23
24fn default_temperature() -> f32 {
25    0.0
26}
27
28fn default_max_tokens() -> u32 {
29    1000
30}
31
32impl Default for LlmConfig {
33    fn default() -> Self {
34        Self {
35            provider: default_llm_provider(),
36            base_url: "https://api.openai.com/v1".into(),
37            api_key: String::new(),
38            model: "gpt-4.1-nano".into(),
39            temperature: default_temperature(),
40            max_tokens: default_max_tokens(),
41            enable_vision: false,
42        }
43    }
44}
45
46#[derive(Debug, Clone, Serialize, Deserialize)]
47pub struct EmbeddingConfig {
48    #[serde(default = "default_embedding_provider")]
49    pub provider: String,
50    pub base_url: String,
51    pub api_key: String,
52    pub model: String,
53    #[serde(default = "default_embedding_dims")]
54    pub dims: usize,
55    pub cache_dir: Option<String>,
56}
57
58fn default_embedding_provider() -> String {
59    "openai".into()
60}
61
62fn default_embedding_dims() -> usize {
63    1536
64}
65
66impl Default for EmbeddingConfig {
67    fn default() -> Self {
68        Self {
69            provider: default_embedding_provider(),
70            base_url: "https://api.openai.com/v1".into(),
71            api_key: String::new(),
72            model: "text-embedding-3-small".into(),
73            dims: default_embedding_dims(),
74            cache_dir: None,
75        }
76    }
77}
78
79#[derive(Debug, Clone, Serialize, Deserialize)]
80pub struct VectorConfig {
81    #[serde(default = "default_vector_provider")]
82    pub provider: String,
83    #[serde(default = "default_collection")]
84    pub collection_name: String,
85    #[serde(default = "default_embedding_dims")]
86    pub dims: usize,
87    pub upstash_url: Option<String>,
88    pub upstash_token: Option<String>,
89}
90
91fn default_vector_provider() -> String {
92    "flat".into()
93}
94
95fn default_collection() -> String {
96    "mem7_memories".into()
97}
98
99impl Default for VectorConfig {
100    fn default() -> Self {
101        Self {
102            provider: default_vector_provider(),
103            collection_name: default_collection(),
104            dims: default_embedding_dims(),
105            upstash_url: None,
106            upstash_token: None,
107        }
108    }
109}
110
111#[derive(Debug, Clone, Serialize, Deserialize)]
112pub struct HistoryConfig {
113    #[serde(default = "default_history_path")]
114    pub db_path: String,
115}
116
117fn default_history_path() -> String {
118    "mem7_history.db".into()
119}
120
121impl Default for HistoryConfig {
122    fn default() -> Self {
123        Self {
124            db_path: default_history_path(),
125        }
126    }
127}
128
129#[derive(Debug, Clone, Serialize, Deserialize)]
130pub struct RerankerConfig {
131    pub provider: String,
132    pub model: Option<String>,
133    pub api_key: Option<String>,
134    pub base_url: Option<String>,
135    #[serde(default = "default_top_k_multiplier")]
136    pub top_k_multiplier: usize,
137}
138
139fn default_top_k_multiplier() -> usize {
140    3
141}
142
143impl Default for RerankerConfig {
144    fn default() -> Self {
145        Self {
146            provider: "cohere".into(),
147            model: None,
148            api_key: None,
149            base_url: None,
150            top_k_multiplier: default_top_k_multiplier(),
151        }
152    }
153}
154
155#[derive(Debug, Clone, Serialize, Deserialize)]
156pub struct GraphConfig {
157    #[serde(default = "default_graph_provider")]
158    pub provider: String,
159    pub kuzu_db_path: Option<String>,
160    pub neo4j_url: Option<String>,
161    pub neo4j_username: Option<String>,
162    pub neo4j_password: Option<String>,
163    pub neo4j_database: Option<String>,
164    pub custom_prompt: Option<String>,
165    pub llm: Option<LlmConfig>,
166}
167
168fn default_graph_provider() -> String {
169    "flat".into()
170}
171
172fn default_kuzu_db_path() -> String {
173    "mem7_graph.kuzu".into()
174}
175
176impl Default for GraphConfig {
177    fn default() -> Self {
178        Self {
179            provider: default_graph_provider(),
180            kuzu_db_path: Some(default_kuzu_db_path()),
181            neo4j_url: None,
182            neo4j_username: None,
183            neo4j_password: None,
184            neo4j_database: None,
185            custom_prompt: None,
186            llm: None,
187        }
188    }
189}
190
191#[derive(Debug, Clone, Serialize, Deserialize)]
192pub struct TelemetryConfig {
193    #[serde(default = "default_otlp_endpoint")]
194    pub otlp_endpoint: String,
195    #[serde(default = "default_service_name")]
196    pub service_name: String,
197}
198
199fn default_otlp_endpoint() -> String {
200    "http://localhost:4317".into()
201}
202
203fn default_service_name() -> String {
204    "mem7".into()
205}
206
207impl Default for TelemetryConfig {
208    fn default() -> Self {
209        Self {
210            otlp_endpoint: default_otlp_endpoint(),
211            service_name: default_service_name(),
212        }
213    }
214}
215
216/// Configuration for the Ebbinghaus-inspired memory decay / forgetting curve.
217///
218/// When enabled, older memories are deprioritized during search and dedup via a
219/// stretched-exponential retention factor. Memories that are accessed frequently
220/// decay more slowly (spaced-repetition effect).
221#[derive(Debug, Clone, Serialize, Deserialize)]
222pub struct DecayConfig {
223    /// Master switch. When `false` (the default), all scoring is unmodified.
224    #[serde(default)]
225    pub enabled: bool,
226    /// Base half-life in seconds before any rehearsal bonus. Default: 7 days.
227    #[serde(default = "default_base_half_life")]
228    pub base_half_life_secs: f64,
229    /// Stretched-exponential shape parameter (0 < gamma <= 1).
230    /// Lower values produce slower initial decay with a steeper tail.
231    #[serde(default = "default_decay_shape")]
232    pub decay_shape: f64,
233    /// Minimum retention floor so no memory ever fully vanishes.
234    #[serde(default = "default_min_retention")]
235    pub min_retention: f64,
236    /// How much each access (rehearsal) increases memory stability.
237    #[serde(default = "default_rehearsal_factor")]
238    pub rehearsal_factor: f64,
239}
240
241fn default_base_half_life() -> f64 {
242    604800.0 // 7 days
243}
244
245fn default_decay_shape() -> f64 {
246    0.8
247}
248
249fn default_min_retention() -> f64 {
250    0.1
251}
252
253fn default_rehearsal_factor() -> f64 {
254    0.5
255}
256
257impl Default for DecayConfig {
258    fn default() -> Self {
259        Self {
260            enabled: false,
261            base_half_life_secs: default_base_half_life(),
262            decay_shape: default_decay_shape(),
263            min_retention: default_min_retention(),
264            rehearsal_factor: default_rehearsal_factor(),
265        }
266    }
267}
268
269#[derive(Debug, Clone, Serialize, Deserialize, Default)]
270pub struct MemoryEngineConfig {
271    #[serde(default)]
272    pub llm: LlmConfig,
273    #[serde(default)]
274    pub embedding: EmbeddingConfig,
275    #[serde(default)]
276    pub vector: VectorConfig,
277    #[serde(default)]
278    pub history: HistoryConfig,
279    pub reranker: Option<RerankerConfig>,
280    pub graph: Option<GraphConfig>,
281    pub telemetry: Option<TelemetryConfig>,
282    pub decay: Option<DecayConfig>,
283    pub custom_fact_extraction_prompt: Option<String>,
284    pub custom_update_memory_prompt: Option<String>,
285}
mem7_config/lib.rs

mem7_config/
lib.rs
