cognee-database 0.2.0

Relational metadata store (SQLite/Postgres via SeaORM) for the cognee AI-memory 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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! Relational metadata persistence (SeaORM/SQLite) for ingestion, search history, and deletion.

mod connection;
mod conversions;
pub mod entities;
pub mod migrator;
pub mod ops;
pub mod pipelines;
pub mod sync;
mod traits;
mod types;
pub mod uuid_hex;

// Re-export tutorial seeder for use by cognee-http-server (which can't depend on cognee).
pub use ops::tutorial_seeder::{
    TUTORIAL_BASICS_ID, TUTORIAL_PYTHON_DEV_ID, seed_tutorials_if_first_call,
};

pub use connection::{connect, initialize, sqlite_url_is_in_memory};

/// Generic backend-label helper shared by the concrete public fn below and the
/// transaction-scoped internals in `ops::graph_storage` (a `DatabaseTransaction`
/// also implements `ConnectionTrait`). Kept crate-private so the published API
/// keeps its pre-0.1.4 `&DatabaseConnection` signature.
pub(crate) fn connection_system_label<C: sea_orm::ConnectionTrait>(db: &C) -> &'static str {
    use sea_orm::DatabaseBackend;
    match db.get_database_backend() {
        DatabaseBackend::Sqlite => "sqlite",
        DatabaseBackend::Postgres => "postgres",
        DatabaseBackend::MySql => "mysql",
    }
}

/// Map the active SeaORM backend to a `cognee.db.system` string
/// matching the values used by the vector / graph adapters.
///
/// The tag values mirror Python's observability layer
/// (`cognee/modules/observability/tracing.py`) and the
/// `cognee.db.system` attribute exposed by every relational op span.
pub fn database_system_label(db: &DatabaseConnection) -> &'static str {
    connection_system_label(db)
}
pub use ops::checkpoint::{CheckpointStore, SeaOrmCheckpointStore};
pub use pipelines::sea_orm_impl::SeaOrmPipelineRunRepository;
pub use pipelines::{
    NoopPipelineRunRepository, PipelineRunRepository, PipelineRunWithAttributionRow,
};
pub use sea_orm::DatabaseConnection;
pub use sync::{
    SeaOrmSyncOperationRepository, SyncOperationRepository, SyncOperationRow, SyncOperationStatus,
};
pub use traits::{
    AclDb, CostByModelRow, DatasetConfigDb, DatasetConfiguration, DatasetConfigurationPatch,
    DeleteDb, IngestDb, Notebook, NotebookDb, NotebookUpdatePatch, SearchHistoryDb,
    SessionLifecycleDb, SessionListFilters, SessionListPage, SessionRowWithStatus, SessionStats,
};
pub use types::{
    DatabaseError, GraphEdge, GraphMetrics, GraphNode, PipelineRun, PipelineRunStatus,
    SearchHistoryEntry, SearchHistoryEntryType, TaskRun,
};

// The `auth`, `permissions`, `UserDb`/`RoleDb`/`TenantDb`,
// `SeaOrmUserAuthRepository`, `SeaOrmApiKeyRepository`, `ApiKey`, `AuthUser`,
// `CreateUserPayload`, `UpdateUserPayload`, `ActiveUserWithApiKeyCount` items
// moved to the closed `cognee-access-control` crate
//. The `types` module deliberately remains private —
// closed callers reach `DatabaseError` via the top-level re-export above.