ruchat 0.1.2

ollama/chroma command-line AI chat tool
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

use thiserror::Error;

/// An enumeration of possible errors in the RuChat application.
///
/// This enum defines various error types that can occur within the
/// RuChat application, each with a descriptive error message.
#[derive(Error, Debug)]
pub enum RuChatError {
    /// Represents an error related to the model.
    #[error("Model Error: {0}")]
    ModelError(String),

    /// Represents an error due to invalid metadata.
    #[error("Invalid metadata: {0}")]
    InvalidMetadata(String),

    /// Represents an input/output error.
    #[error("Input/output error: {0}")]
    Io(#[from] std::io::Error),

    /// Represents an error during serialization or deserialization.
    #[error("Serde error: {0}")]
    SerdeError(#[from] serde_json::error::Error),

    /// Represents an error from the Ollama library.
    #[error("Ollama error: {0}")]
    OllamaError(#[from] ollama_rs::error::OllamaError),

    /// Represents an error when parsing the server argument.
    #[error("Unable to parse arg --server: '{0}'")]
    ArgServerError(String),

    /// Represents an error from the Chroma library.
    #[error("Chroma error: {0}")]
    ChromaError(#[from] anyhow::Error),

    /// Represents an error when converting from an integer.
    #[error("TryFromIntError: {0}")]
    TryFromIntError(#[from] std::num::TryFromIntError),

    /// Represents an error when the cursor's first index is out of bounds.
    #[error("Cursor.0 out of bounds")]
    Cursor0OutOfBounds,

    /// Represents an error when the cursor's second index is out of bounds.
    #[error("Cursor.1 out of bounds")]
    Cursor1OutOfBounds,

    /// Represents an error when a task join operation fails.
    #[error("Join error: {0}")]
    JoinError(#[from] tokio::task::JoinError),

    /// Represents an error when a request is not found.
    #[error("Request not found")]
    QuestionNotFound,

    /// Represents an error when an answer is not found.
    #[error("Answer not found")]
    AnswerNotFound,

    /// Represents an error when a question already exists.
    #[error("Question already exists")]
    QuestionAlreadyExists,

    /// Represents an error when an answer already exists.
    #[error("Answer already exists")]
    AnswerAlreadyExists,
}

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

    #[test]
    fn test_model_error() {
        let error = RuChatError::ModelError("Test model error".to_string());
        assert_eq!(format!("{}", error), "Model Error: Test model error");
    }

    #[test]
    fn test_invalid_metadata_error() {
        let error = RuChatError::InvalidMetadata("Test invalid metadata".to_string());
        assert_eq!(
            format!("{}", error),
            "Invalid metadata: Test invalid metadata"
        );
    }

    #[test]
    fn test_cursor0_out_of_bounds_error() {
        let error = RuChatError::Cursor0OutOfBounds;
        assert_eq!(format!("{}", error), "Cursor.0 out of bounds");
    }

    #[test]
    fn test_cursor1_out_of_bounds_error() {
        let error = RuChatError::Cursor1OutOfBounds;
        assert_eq!(format!("{}", error), "Cursor.1 out of bounds");
    }
}