cognee-lib 0.1.3

Cognee — an AI-memory pipeline that turns raw data into queryable knowledge graphs (umbrella crate).
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

//! Public session management API — thin wrappers over `SessionManager`.
//!
//! Mirrors the Python `cognee/api/v1/session/session.py` functions:
//! `get_session`, `add_feedback`, `delete_feedback`, `get_graph_context`,
//! `set_graph_context`.

pub use cognee_session::{
    SessionContext, SessionError, SessionManager, SessionQAEntry, SessionQAUpdate, SessionStore,
    UsedGraphElementIds,
};

/// Retrieve Q&A history from a session.
///
/// If `last_n` is `Some(n)`, only the most recent `n` entries are returned.
/// Otherwise all entries are returned. Delegates to the underlying store
/// via the manager, using the explicit `session_id` (not the manager's default).
pub async fn get_session(
    store: &dyn SessionStore,
    session_id: &str,
    user_id: Option<&str>,
    last_n: Option<usize>,
) -> Result<Vec<SessionQAEntry>, SessionError> {
    if let Some(n) = last_n {
        store.get_latest_qa_entries(session_id, user_id, n).await
    } else {
        store.get_all_qa_entries(session_id, user_id).await
    }
}

/// Add feedback (text and/or score) to a Q&A entry.
///
/// Returns `true` if the entry was found and updated.
pub async fn add_feedback(
    manager: &SessionManager,
    session_id: &str,
    qa_id: &str,
    user_id: Option<&str>,
    feedback_text: Option<&str>,
    feedback_score: Option<i32>,
) -> Result<bool, SessionError> {
    manager
        .add_feedback(
            Some(session_id),
            user_id,
            qa_id,
            feedback_text,
            feedback_score,
        )
        .await
}

/// Clear feedback from a Q&A entry.
///
/// Returns `true` if the entry was found and updated.
pub async fn delete_feedback(
    manager: &SessionManager,
    session_id: &str,
    qa_id: &str,
    user_id: Option<&str>,
) -> Result<bool, SessionError> {
    manager
        .delete_feedback(Some(session_id), user_id, qa_id)
        .await
}

/// Retrieve the graph knowledge snapshot for a session.
pub async fn get_graph_context(
    manager: &SessionManager,
    session_id: &str,
    user_id: Option<&str>,
) -> Result<Option<String>, SessionError> {
    manager.get_graph_context(Some(session_id), user_id).await
}

/// Store (or overwrite) the graph knowledge snapshot for a session.
pub async fn set_graph_context(
    manager: &SessionManager,
    session_id: &str,
    user_id: Option<&str>,
    context: &str,
) -> Result<(), SessionError> {
    manager
        .set_graph_context(Some(session_id), user_id, context)
        .await
}