lutum-protocol 0.1.0

Core traits and request/response types for lutum
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

use std::error::Error as StdError;

use thiserror::Error;

pub type BoxError = Box<dyn StdError + Send + Sync + 'static>;

#[derive(Debug, Error)]
pub enum AgentError {
    #[error("invalid model input: {0}")]
    InvalidModelInput(#[from] crate::conversation::ModelInputValidationError),
    #[error("budget error: {0}")]
    Budget(#[source] BoxError),
    #[error("backend error: {0}")]
    Backend(#[source] BoxError),
    #[error("failed to parse tool call: {0}")]
    ToolCall(#[from] crate::toolset::ToolCallError),
    #[error("failed to decode JSON: {0}")]
    Json(#[from] serde_json::Error),
    #[error("failed to decode structured output: {0}")]
    StructuredOutput(#[source] serde_json::Error),
    #[error("other error: {0}")]
    Other(#[source] BoxError),
}

impl AgentError {
    pub fn budget(source: impl StdError + Send + Sync + 'static) -> Self {
        Self::Budget(Box::new(source))
    }

    pub fn backend(source: impl StdError + Send + Sync + 'static) -> Self {
        Self::Backend(Box::new(source))
    }

    pub fn structured_output(source: serde_json::Error) -> Self {
        Self::StructuredOutput(source)
    }

    pub fn other(source: impl StdError + Send + Sync + 'static) -> Self {
        Self::Other(Box::new(source))
    }
}