cognee-search 0.1.3

Context retrieval (search) over the cognee knowledge graph and vector store.
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

use thiserror::Error;

#[derive(Debug, Error)]
pub enum SearchError {
    #[error("invalid input: {0}")]
    InvalidInput(String),

    #[error("vector error: {0}")]
    VectorError(String),

    #[error("graph error: {0}")]
    GraphError(String),

    #[error("llm error: {0}")]
    LlmError(String),

    #[error("embedding error: {0}")]
    EmbeddingError(String),

    #[error("database error: {0}")]
    DatabaseError(String),

    #[error("not found: {0}")]
    NotFound(String),

    #[error("dataset not found: {0}")]
    DatasetNotFound(String),

    #[error("unsupported search type: {0:?}")]
    UnsupportedSearchType(crate::types::SearchType),

    #[error("serialization error: {0}")]
    SerializationError(String),
}

impl From<cognee_vector::VectorDBError> for SearchError {
    fn from(value: cognee_vector::VectorDBError) -> Self {
        Self::VectorError(value.to_string())
    }
}

impl From<cognee_graph::GraphDBError> for SearchError {
    fn from(value: cognee_graph::GraphDBError) -> Self {
        Self::GraphError(value.to_string())
    }
}

impl From<cognee_llm::LlmError> for SearchError {
    fn from(value: cognee_llm::LlmError) -> Self {
        Self::LlmError(value.to_string())
    }
}

impl From<cognee_embedding::EmbeddingError> for SearchError {
    fn from(value: cognee_embedding::EmbeddingError) -> Self {
        Self::EmbeddingError(value.to_string())
    }
}

impl From<serde_json::Error> for SearchError {
    fn from(value: serde_json::Error) -> Self {
        Self::SerializationError(value.to_string())
    }
}

impl From<cognee_database::DatabaseError> for SearchError {
    fn from(value: cognee_database::DatabaseError) -> Self {
        Self::DatabaseError(value.to_string())
    }
}

impl From<cognee_session::SessionError> for SearchError {
    fn from(value: cognee_session::SessionError) -> Self {
        Self::DatabaseError(value.to_string())
    }
}