zeph_core/agent/
session_config.rs

1// SPDX-FileCopyrightText: 2026 Andrei G <bug-ops>
2// SPDX-License-Identifier: MIT OR Apache-2.0
3
4use std::sync::Arc;
5
6use crate::config::{
7    Config, DebugConfig, DocumentConfig, GraphConfig, LearningConfig, OrchestrationConfig,
8    SecurityConfig, TimeoutConfig,
9};
10use crate::vault::Secret;
11
12/// Reserve ratio for `with_context_budget`: fraction of budget reserved for LLM reply.
13///
14/// Extracted from the hardcoded `0.20` literal used in both `spawn_acp_agent` and `runner.rs`.
15pub const CONTEXT_BUDGET_RESERVE_RATIO: f32 = 0.20;
16
17/// All config-derived fields needed to configure an `Agent` session.
18///
19/// This is the single source of truth for config → agent wiring.
20/// Adding a new config field requires exactly three changes:
21///
22/// 1. Add the field here.
23/// 2. Map it in [`AgentSessionConfig::from_config`].
24/// 3. Apply it in [`super::Agent::apply_session_config`] (destructure triggers a compile error if
25///    you forget step 3 — see the S4 note in the critic handoff).
26///
27/// ## What is NOT here
28///
29/// - **Shared runtime objects** (`provider`, `registry`, `memory`, `mcp_manager`, etc.) — these
30///   are expensive to create and shared across sessions; they stay in `SharedAgentDeps`.
31/// - **ACP-specific fields** (`acp_max_sessions`, bearer token, etc.) — transport-level, not
32///   agent-level.
33/// - **Optional runtime providers** (`summary_provider`, `judge_provider`,
34///   `quarantine_provider`) — these contain HTTP client pools (`AnyProvider`) that carry runtime
35///   state; callers wire them separately via `with_summary_provider` / `with_judge_provider` /
36///   `apply_quarantine_provider`.
37/// - **`mcp_config`** — passed alongside runtime MCP objects in `with_mcp()`; separating it
38///   from `mcp_tools` / `mcp_manager` would make the call site awkward.
39/// - **Runner-only fields** (`compression`, `routing`, `autosave`, `hybrid_search`, `trust_config`,
40///   `disambiguation_threshold`, `logging_config`, `subagent`, `experiment`, `instruction`,
41///   `lsp_hooks`, `response_cache`, `cost_tracker`) — not used in ACP sessions; keeping them out
42///   avoids unused-field noise and prevents inadvertent ACP behavior changes.
43/// - **Scheduler runtime objects** (`scheduler_executor`, broadcast senders) — runtime state,
44///   not config-derived values.
45#[derive(Clone)]
46#[allow(clippy::struct_excessive_bools)]
47pub struct AgentSessionConfig {
48    // Tool behavior
49    pub max_tool_iterations: usize,
50    pub max_tool_retries: usize,
51    pub max_retry_duration_secs: u64,
52    pub retry_base_ms: u64,
53    pub retry_max_ms: u64,
54    pub parameter_reformat_provider: String,
55    pub tool_repeat_threshold: usize,
56    pub tool_summarization: bool,
57    pub tool_call_cutoff: usize,
58    pub overflow_config: zeph_tools::OverflowConfig,
59    pub permission_policy: zeph_tools::PermissionPolicy,
60
61    // Model
62    pub model_name: String,
63    pub embed_model: String,
64
65    // Semantic cache
66    pub semantic_cache_enabled: bool,
67    pub semantic_cache_threshold: f32,
68    pub semantic_cache_max_candidates: u32,
69
70    // Memory / compaction
71    pub budget_tokens: usize,
72    pub soft_compaction_threshold: f32,
73    pub hard_compaction_threshold: f32,
74    pub compaction_preserve_tail: usize,
75    pub compaction_cooldown_turns: u8,
76    pub prune_protect_tokens: usize,
77    pub redact_credentials: bool,
78
79    // Security
80    pub security: SecurityConfig,
81    pub timeouts: TimeoutConfig,
82
83    // Feature configs
84    pub learning: LearningConfig,
85    pub document_config: DocumentConfig,
86    pub graph_config: GraphConfig,
87    pub anomaly_config: zeph_tools::AnomalyConfig,
88    pub result_cache_config: zeph_tools::ResultCacheConfig,
89    pub utility_config: zeph_tools::UtilityScoringConfig,
90    pub orchestration_config: OrchestrationConfig,
91    pub debug_config: DebugConfig,
92    pub server_compaction: bool,
93
94    /// Custom secrets from config.
95    ///
96    /// Stored as `Arc` because `Secret` intentionally does not implement `Clone` —
97    /// the wrapper prevents accidental duplication. Iteration produces new `Secret`
98    /// values via `Secret::new(v.expose())` on the consumption side.
99    pub secrets: Arc<[(String, Secret)]>,
100}
101
102impl AgentSessionConfig {
103    /// Build from a resolved [`Config`] snapshot and a pre-computed `budget_tokens`.
104    ///
105    /// `budget_tokens` is passed as a parameter because its computation (`auto_budget_tokens`)
106    /// depends on the active provider and must happen before `AgentSessionConfig` is constructed.
107    #[must_use]
108    pub fn from_config(config: &Config, budget_tokens: usize) -> Self {
109        Self {
110            max_tool_iterations: config.agent.max_tool_iterations,
111            max_tool_retries: config.tools.retry.max_attempts,
112            max_retry_duration_secs: config.tools.retry.budget_secs,
113            retry_base_ms: config.tools.retry.base_ms,
114            retry_max_ms: config.tools.retry.max_ms,
115            parameter_reformat_provider: config.tools.retry.parameter_reformat_provider.clone(),
116            tool_repeat_threshold: config.agent.tool_repeat_threshold,
117            tool_summarization: config.tools.summarize_output,
118            tool_call_cutoff: config.memory.tool_call_cutoff,
119            overflow_config: config.tools.overflow.clone(),
120            permission_policy: config
121                .tools
122                .permission_policy(config.security.autonomy_level),
123            model_name: config.llm.effective_model().to_owned(),
124            embed_model: crate::bootstrap::effective_embedding_model(config),
125            semantic_cache_enabled: config.llm.semantic_cache_enabled,
126            semantic_cache_threshold: config.llm.semantic_cache_threshold,
127            semantic_cache_max_candidates: config.llm.semantic_cache_max_candidates,
128            budget_tokens,
129            soft_compaction_threshold: config.memory.soft_compaction_threshold,
130            hard_compaction_threshold: config.memory.hard_compaction_threshold,
131            compaction_preserve_tail: config.memory.compaction_preserve_tail,
132            compaction_cooldown_turns: config.memory.compaction_cooldown_turns,
133            prune_protect_tokens: config.memory.prune_protect_tokens,
134            redact_credentials: config.memory.redact_credentials,
135            security: config.security.clone(),
136            timeouts: config.timeouts,
137            learning: config.skills.learning.clone(),
138            document_config: config.memory.documents.clone(),
139            graph_config: config.memory.graph.clone(),
140            anomaly_config: config.tools.anomaly.clone(),
141            result_cache_config: config.tools.result_cache.clone(),
142            utility_config: config.tools.utility.clone(),
143            orchestration_config: config.orchestration.clone(),
144            debug_config: config.debug.clone(),
145            server_compaction: config.llm.providers.iter().any(|e| e.server_compaction),
146            secrets: config
147                .secrets
148                .custom
149                .iter()
150                .map(|(k, v)| (k.clone(), Secret::new(v.expose().to_owned())))
151                .collect::<Vec<_>>()
152                .into(),
153        }
154    }
155}
156
157#[cfg(test)]
158mod tests {
159    use super::*;
160
161    #[test]
162    fn from_config_maps_all_fields() {
163        let config = Config::default();
164        let budget = 100_000;
165        let sc = AgentSessionConfig::from_config(&config, budget);
166
167        assert_eq!(sc.max_tool_iterations, config.agent.max_tool_iterations);
168        assert_eq!(sc.max_tool_retries, config.tools.retry.max_attempts);
169        assert_eq!(sc.max_retry_duration_secs, config.tools.retry.budget_secs);
170        assert_eq!(sc.retry_base_ms, config.tools.retry.base_ms);
171        assert_eq!(sc.retry_max_ms, config.tools.retry.max_ms);
172        assert_eq!(
173            sc.parameter_reformat_provider,
174            config.tools.retry.parameter_reformat_provider
175        );
176        assert_eq!(sc.tool_repeat_threshold, config.agent.tool_repeat_threshold);
177        assert_eq!(sc.tool_summarization, config.tools.summarize_output);
178        assert_eq!(sc.tool_call_cutoff, config.memory.tool_call_cutoff);
179        assert_eq!(sc.model_name, config.llm.effective_model());
180        assert_eq!(
181            sc.embed_model,
182            crate::bootstrap::effective_embedding_model(&config)
183        );
184        assert_eq!(sc.semantic_cache_enabled, config.llm.semantic_cache_enabled);
185        assert!(
186            (sc.semantic_cache_threshold - config.llm.semantic_cache_threshold).abs()
187                < f32::EPSILON
188        );
189        assert_eq!(
190            sc.semantic_cache_max_candidates,
191            config.llm.semantic_cache_max_candidates
192        );
193        assert_eq!(sc.budget_tokens, budget);
194        assert!(
195            (sc.soft_compaction_threshold - config.memory.soft_compaction_threshold).abs()
196                < f32::EPSILON
197        );
198        assert!(
199            (sc.hard_compaction_threshold - config.memory.hard_compaction_threshold).abs()
200                < f32::EPSILON
201        );
202        assert_eq!(
203            sc.compaction_preserve_tail,
204            config.memory.compaction_preserve_tail
205        );
206        assert_eq!(
207            sc.compaction_cooldown_turns,
208            config.memory.compaction_cooldown_turns
209        );
210        assert_eq!(sc.prune_protect_tokens, config.memory.prune_protect_tokens);
211        assert_eq!(sc.redact_credentials, config.memory.redact_credentials);
212        assert_eq!(sc.graph_config.enabled, config.memory.graph.enabled);
213        assert_eq!(
214            sc.orchestration_config.enabled,
215            config.orchestration.enabled
216        );
217        assert_eq!(
218            sc.orchestration_config.max_tasks,
219            config.orchestration.max_tasks
220        );
221        assert_eq!(sc.anomaly_config.enabled, config.tools.anomaly.enabled);
222        assert_eq!(
223            sc.result_cache_config.enabled,
224            config.tools.result_cache.enabled
225        );
226        assert_eq!(
227            sc.result_cache_config.ttl_secs,
228            config.tools.result_cache.ttl_secs
229        );
230        assert_eq!(sc.debug_config.enabled, config.debug.enabled);
231        assert_eq!(
232            sc.document_config.rag_enabled,
233            config.memory.documents.rag_enabled
234        );
235        assert_eq!(
236            sc.overflow_config.threshold,
237            config.tools.overflow.threshold
238        );
239        assert_eq!(
240            sc.permission_policy.autonomy_level(),
241            config.security.autonomy_level
242        );
243        assert_eq!(sc.security.autonomy_level, config.security.autonomy_level);
244        assert_eq!(sc.timeouts.llm_seconds, config.timeouts.llm_seconds);
245        assert_eq!(sc.learning.enabled, config.skills.learning.enabled);
246        assert_eq!(
247            sc.server_compaction,
248            config.llm.providers.iter().any(|e| e.server_compaction)
249        );
250        assert_eq!(sc.secrets.len(), config.secrets.custom.len());
251    }
252}
zeph_core/agent/session_config.rs

zeph_core/agent/
session_config.rs
