tinyagents 0.1.1

A Rust LLM orchestration library inspired by LangChain and LangGraph.
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

//! Types for run-scoped limit enforcement.
//!
//! [`RunLimits`] carries the configured policy; [`LimitTracker`] holds the live
//! counters and checks them against the policy.

/// Configures the hard limits applied across a single harness run.
///
/// All limits are checked fail-closed: the first call that exceeds a cap
/// returns an error and the run should be stopped.
///
/// # Examples
///
/// ```
/// use tinyagents::harness::limits::RunLimits;
///
/// let limits = RunLimits::default()
///     .with_max_model_calls(10)
///     .with_max_tool_calls(20);
/// ```
#[derive(Clone, Debug, PartialEq)]
pub struct RunLimits {
    /// Maximum number of model API calls permitted for this run.
    pub max_model_calls: usize,
    /// Maximum number of tool invocations permitted for this run.
    pub max_tool_calls: usize,
    /// Maximum elapsed wall-clock time in milliseconds. `None` means no limit.
    pub max_wall_clock_ms: Option<u64>,
    /// Maximum number of retry attempts per individual call.
    pub max_retries_per_call: usize,
    /// Maximum number of concurrent in-flight calls. `None` means no limit.
    pub max_concurrency: Option<usize>,
    /// Maximum sub-agent / recursion depth allowed for the run tree rooted at
    /// this run. A top-level run is depth `0`; each nested child run increments
    /// the depth. A sub-agent invocation whose child depth would exceed this cap
    /// fails fast (see [`crate::harness::subagent`]). Defaults to
    /// [`RunLimits::DEFAULT_MAX_DEPTH`].
    pub max_depth: usize,
}

impl RunLimits {
    /// Default sub-agent / recursion depth cap when none is configured.
    pub const DEFAULT_MAX_DEPTH: usize = 8;
}

impl Default for RunLimits {
    fn default() -> Self {
        Self {
            max_model_calls: 25,
            max_tool_calls: 50,
            max_wall_clock_ms: None,
            max_retries_per_call: 3,
            max_concurrency: None,
            max_depth: Self::DEFAULT_MAX_DEPTH,
        }
    }
}