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

//! Telemetry surface for embedders.
//!
//! Re-exports the public API of [`cognee_observability`] (OTEL setup,
//! gap 01) and [`cognee_telemetry`] (`send_telemetry` product-analytics
//! client, gap 02) so consumers reach both through the same
//! `cognee_lib::telemetry` namespace.
//!
//! When the `telemetry` cargo feature is on, observability re-exports
//! pull in `cognee-observability` and the `send_telemetry` re-exports
//! pull in the real `cognee-telemetry` impl. When the feature is off,
//! observability re-exports vanish and the `send_telemetry` re-exports
//! resolve to the noop bodies inside `cognee-telemetry` (so
//! `cognee_telemetry::send_telemetry` and
//! `cognee_telemetry::TelemetryError` are always available — they
//! exist regardless of feature state in the leaf crate).

// --- gap 01: OTEL/observability surface (feature-gated) ---------------------

#[cfg(feature = "telemetry")]
pub use cognee_observability::{
    BoxedTelemetryLayer, SettingsView, TelemetryGuard, TelemetryInitError, already_instrumented,
    init_telemetry, is_tracing_enabled, parse_otlp_headers,
};

#[cfg(feature = "telemetry")]
use crate::config::Settings;

#[cfg(feature = "telemetry")]
impl SettingsView for Settings {
    fn tracing_enabled(&self) -> bool {
        self.cognee_tracing_enabled
    }

    fn service_name(&self) -> &str {
        &self.otel_service_name
    }

    fn otlp_endpoint(&self) -> &str {
        &self.otel_exporter_otlp_endpoint
    }

    fn otlp_headers(&self) -> &str {
        &self.otel_exporter_otlp_headers
    }

    fn otlp_protocol(&self) -> &str {
        &self.otel_exporter_otlp_protocol
    }

    fn span_processor(&self) -> &str {
        &self.otel_span_processor
    }

    fn traces_sampler(&self) -> &str {
        &self.otel_traces_sampler
    }

    fn traces_sampler_arg(&self) -> &str {
        &self.otel_traces_sampler_arg
    }
}

// --- gap 02: send_telemetry product-analytics surface (always available) ----
//
// `cognee_telemetry::{send_telemetry, try_send_telemetry, TelemetryError,
// UserIdRef, PropertyValue}` are exported by the leaf crate in BOTH
// feature states (the leaf crate switches between real and noop bodies
// internally, but the symbols are stable). Re-export them unconditionally
// so callers compile under `--no-default-features` and can name
// `cognee_lib::telemetry::TelemetryError`.

pub use cognee_telemetry::{
    PropertyValue, TelemetryError, UserIdRef, send_telemetry, try_send_telemetry,
};

#[cfg(feature = "telemetry")]
pub use cognee_telemetry::{env, ids, payload, sanitize};