cognis-core 0.1.0

Core traits and types for the Cognis LLM framework
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use thiserror::Error;

/// Error codes matching langchain ErrorCode enum.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ErrorCode {
    InvalidPromptInput,
    InvalidToolResults,
    MessageCoercionFailure,
    ModelAuthentication,
    ModelNotFound,
    ModelRateLimit,
    OutputParsingFailure,
}

impl ErrorCode {
    pub fn as_str(&self) -> &'static str {
        match self {
            Self::InvalidPromptInput => "INVALID_PROMPT_INPUT",
            Self::InvalidToolResults => "INVALID_TOOL_RESULTS",
            Self::MessageCoercionFailure => "MESSAGE_COERCION_FAILURE",
            Self::ModelAuthentication => "MODEL_AUTHENTICATION",
            Self::ModelNotFound => "MODEL_NOT_FOUND",
            Self::ModelRateLimit => "MODEL_RATE_LIMIT",
            Self::OutputParsingFailure => "OUTPUT_PARSING_FAILURE",
        }
    }
}

/// Core error type for Cognis.
#[derive(Debug, Error)]
pub enum CognisError {
    #[error("Output parser error: {message}")]
    OutputParserError {
        message: String,
        observation: Option<String>,
        llm_output: Option<String>,
    },

    #[error("Context overflow: {0}")]
    ContextOverflow(String),

    #[error("Tracer error: {0}")]
    TracerError(String),

    #[error("Invalid key: {0}")]
    InvalidKey(String),

    #[error("Serialization error: {0}")]
    SerializationError(#[from] serde_json::Error),

    #[error("Not implemented: {0}")]
    NotImplemented(String),

    #[error("Tool exception: {0}")]
    ToolException(String),

    #[error("Tool validation error: {0}")]
    ToolValidationError(String),

    #[error("Schema annotation error: {0}")]
    SchemaAnnotationError(String),

    #[error("Recursion limit exceeded: {0}")]
    RecursionLimitExceeded(String),

    #[error("Type mismatch: expected {expected}, got {got}")]
    TypeMismatch { expected: String, got: String },

    #[error("HTTP error {status}: {body}")]
    HttpError { status: u16, body: String },

    #[error("IO error: {0}")]
    IoError(#[from] std::io::Error),

    #[error("{0}")]
    Other(String),
}

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