espionox 0.1.42

A library for creating multi-agent workflows in Rust
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

use tracing::warn;

use crate::errors::error_chain_fmt;
use std::fmt::{Debug, Display, Formatter, Result as FmtResult};

pub type CompletionResult<T> = Result<T, CompletionError>;

#[derive(thiserror::Error)]
pub enum CompletionError {
    #[error(transparent)]
    Undefined(#[from] anyhow::Error),
    Json(#[from] serde_json::Error),
    Request(#[from] reqwest::Error),
    Provider(String),
    FunctionNotImplemented,
    StreamTimeout,
    CouldNotCoerce,
}

pub trait ProviderResponseError: Debug {
    fn into_error(&self) -> CompletionError {
        warn!("Coercing to completion error: {:?}", self);
        CompletionError::Provider(format!("Provider error: {:?}", self))
    }
}
impl Debug for CompletionError {
    fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
        error_chain_fmt(self, f)
    }
}

impl Display for CompletionError {
    fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
        let display = match self {
            Self::Json(err) => err.to_string(),
            Self::Undefined(err) => err.to_string(),
            Self::Request(err) => err.to_string(),
            Self::StreamTimeout => "Stream Timeout".to_string(),
            Self::Provider(err) => err.to_string(),
            Self::CouldNotCoerce => "Could Not Coerce".to_string(),
            Self::FunctionNotImplemented => "Function Not Implemented".to_string(),
        };
        write!(f, "{}", display)
    }
}