cognee-graph 0.1.1

Knowledge-graph database abstraction (embedded Ladybug) for the cognee pipeline.
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

//! Graph database abstraction layer for Cognee.
//!
//! This crate provides a trait-based interface for graph database operations,
//! enabling pluggable graph database backends for knowledge graph storage.
//!
//! # Architecture
//!
//! - `GraphDBTrait` - Async trait defining graph database operations
//! - `LadybugAdapter` - Implementation using Ladybug (lbug) embedded graph database
//! - `GraphNode` / `GraphEdge` - Type aliases for node and edge data
//!
//! # Example
//!
//! ```ignore
//! use cognee_graph::{GraphDBTrait, LadybugAdapter};
//! use cognee_models::Entity;
//!
//! let db = LadybugAdapter::new("./graph_data").await?;
//! db.initialize().await?;
//!
//! let entity = Entity::new("Alice", EntityType::new("Person", None), Some("user-1"));
//! db.add_node(&entity).await?;
//! ```

mod error;
mod formatted;
mod traits;
mod types;

#[cfg(feature = "ladybug")]
mod ladybug;

#[cfg(feature = "postgres")]
mod pg_graph_adapter;

#[cfg(any(test, feature = "testing"))]
pub mod mock;

pub use error::{GraphDBError, GraphDBResult};
pub use formatted::get_formatted_graph_data;
pub use traits::{EdgeKey, GraphDBTrait, GraphDBTraitExt};
pub use types::{EdgeData, GraphEdge, GraphNode, NodeData};

#[cfg(feature = "ladybug")]
pub use ladybug::LadybugAdapter;

#[cfg(feature = "postgres")]
pub use pg_graph_adapter::PgGraphAdapter;

#[cfg(any(test, feature = "testing"))]
pub use mock::MockGraphDB;