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

use cognee_vector::SearchResult;

use crate::types::{SearchContext, SearchError, SearchItem};

pub(crate) fn search_results_to_context(
    results: Vec<SearchResult>,
) -> Result<SearchContext, SearchError> {
    results
        .into_iter()
        .map(|result| {
            let payload = serde_json::to_value(result.metadata)?;
            Ok(SearchItem {
                id: Some(result.id),
                score: Some(result.score),
                payload,
            })
        })
        .collect()
}