fornix 0.3.0

Knowledge storage, retrieval, and graph infrastructure for cognitive systems
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

//! Error types for the agent layer.

use thiserror::Error;

#[derive(Debug, Error)]
pub enum Error {
    #[error("configuration error: {0}")]
    Configuration(String),
    #[error("tool error: {0}")]
    Tool(String),
    #[error("policy blocked: {0}")]
    PolicyBlocked(String),
    #[error("max depth exceeded at depth {0}")]
    MaxDepthExceeded(usize),
    #[error("step budget exhausted")]
    StepBudgetExhausted,
    #[error("token budget exhausted: used {used}, max {max}")]
    TokenBudgetExhausted { used: usize, max: usize },
    #[error("time limit exceeded after {steps} step(s)")]
    TimeLimitExceeded { steps: usize },
    #[error("cancelled")]
    Cancelled,
    #[error(transparent)]
    Other(#[from] Box<dyn std::error::Error + Send + Sync>),
}

impl Error {
    pub fn config(msg: impl Into<String>) -> Self { Self::Configuration(msg.into()) }
    pub fn tool(msg: impl Into<String>) -> Self { Self::Tool(msg.into()) }
    pub fn blocked(msg: impl Into<String>) -> Self { Self::PolicyBlocked(msg.into()) }
}

pub type Result<T> = std::result::Result<T, Error>;

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn config_error() {
        assert_eq!(Error::config("no model").to_string(), "configuration error: no model");
    }

    #[test]
    fn token_budget_exhausted() {
        let e = Error::TokenBudgetExhausted { used: 5000, max: 4096 };
        assert!(e.to_string().contains("5000"));
    }

    #[test]
    fn cancelled_message() {
        assert_eq!(Error::Cancelled.to_string(), "cancelled");
    }
}