Skip to main content

hirn_exec/
lib.rs

1//! `hirn-exec` — DataFusion physical operators, scoring UDFs, and optimizer rules.
2//!
3//! This crate provides the execution layer for hirn's cognitive memory engine,
4//! built on top of Apache DataFusion. Every cognitive operation (activation,
5//! scoring, budgeting, causal reasoning) is expressed as a composable DataFusion
6//! physical operator over Arrow columnar batches.
7//!
8//! # Modules
9//!
10//! - [`operators`] — Physical `ExecutionPlan` implementations (19 operators)
11//! - [`udfs`] — Scalar UDF implementations (8 UDFs)
12//! - [`rules`] — Physical optimizer rule implementations (5 rules)
13//! - [`extensions`] — `HirnSessionExt` for runtime state injection
14
15pub mod extensions;
16pub mod operators;
17pub mod planner;
18pub mod rules;
19pub mod udfs;
20
21#[cfg(test)]
22pub(crate) mod test_utils;
23
24pub use extensions::{
25    ContextAssemblyRuntime, GraphActivationOutput, GraphCausalChainRow, GraphReadRuntime,
26    GraphTraverseRow, HirnSessionExt, QueryReadRuntime, RegisteredContextAssemblyRuntime,
27    RegisteredQueryReadRuntime, register_context_assembly_runtime, register_query_read_runtime,
28};
29pub use operators::{
30    ActivationMode, CausalChainExec, CausalQueryReadExec, CausalReadKind, ContextAssemblyExec,
31    ContextBudgetExec, GlobalSearchExec, GlobalSearchParams, GraphActivationExec,
32    GraphTraverseExec, HebbianBufferExec, HybridSearchParams, LanceHybridSearchExec,
33    PolicyQueryReadExec, PolicyReadKind, RaptorSearchExec, RaptorSearchParams, RecallMergeExec,
34    SearchComparisonOp, SearchNumericField, SearchNumericFilter, SvoEventScanExec,
35    TargetedQueryReadExec, TargetedReadKind,
36};
37pub use planner::{HirnExtensionPlanner, HirnQueryPlanner};
38pub use rules::{ActivationFusionRule, TemporalIndexRule, all_rules};
39pub use udfs::{
40    CausalRelevanceUdf, CompositeScoreUdf, FadeMemDecayUdf, RpeScoreUdf, SourceReliabilityUdf,
41    SurpriseScoreUdf, TemporalDecayUdf, TokenCountUdf, register_all_udfs,
42};