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

//! Unified search orchestration across 15 retrieval strategies (GraphCompletion, RagCompletion, Chunks, and more).

/// Graph retrieval strategies and helpers.
pub mod graph_retrieval;
/// Observability hooks for search pipeline tracing.
pub mod observability;
/// Search orchestration and pipeline assembly.
pub mod orchestration;
/// Query router that maps natural-language queries to `SearchType`s.
pub mod query_router;
/// Query router override-count statistics.
pub mod query_router_stats;
/// `RecallScope` and related types for scoping search results.
pub mod recall_scope;
/// Individual retriever implementations for each `SearchType`.
pub mod retrievers;
/// Core search types: `SearchType`, `SearchParams`, `SearchResult`, etc.
pub mod types;
/// Shared search utilities (completion helpers, etc.).
pub mod utils;

pub use cognee_session::{SeaOrmSessionStore, SessionContext, SessionManager, SessionStore};
pub use orchestration::{SearchBuilder, SearchOrchestrator, SearchTypeRegistry};
pub use query_router::{RouteResult, route_query};
pub use query_router_stats::{clear_override_counts, override_counts_snapshot, record_override};
pub use recall_scope::{RecallItem, RecallScope, RecallSource, ScopeInput, normalize_scope};
pub use retrievers::{
    ChunksRetriever, CodingRulesRetriever, CompletionRetriever, CypherSearchRetriever,
    FeedbackRetriever, FeelingLuckyRetriever, GraphCompletionContextExtensionRetriever,
    GraphCompletionCotRetriever, GraphCompletionRetriever, GraphSummaryCompletionRetriever,
    LexicalRetriever, NaturalLanguageRetriever, SearchRetriever, SearchRetrieverRef,
    SummariesRetriever, TemporalRetriever, TripletRetriever,
};
pub use types::{
    FeedbackDetectionResult, Rule, SearchContext, SearchError, SearchGraph, SearchGraphEdge,
    SearchGraphNode, SearchItem, SearchOutput, SearchParams, SearchRequest, SearchResponse,
    SearchType,
};