Struct HirnSessionExt 

pub struct HirnSessionExt {
    pub config: Arc<HirnConfig>,
    pub rpe_population_stats: Arc<RwLock<WelfordStats>>,
    /* private fields */
}

Shared runtime state accessible by all hirn DataFusion operators.

Registered in SessionContext at database open time. Operators retrieve it via HirnSessionExt::get — never through constructor injection.

Fields

config: Arc<HirnConfig>

Scoring weights and database configuration.

rpe_population_stats: Arc<RwLock<WelfordStats>>

Shared historical RPE population statistics (Welford’s online algorithm).

Seeded from WriteRuntime at session setup; updated by RpeScoreExec after each batch so that z-scores compare against the full historical distribution, not just the current write batch (N-H08).

Implementations

impl HirnSessionExt

pub fn new( graph: Arc<dyn Any + Send + Sync>, config: Arc<HirnConfig>, embedder: Option<Arc<dyn Embedder>>, ) -> Self

Create a new extension bundle. Create a new extension bundle.

pub fn with_rpe_population_stats(self, stats: Arc<RwLock<WelfordStats>>) -> Self

Seed historical RPE population statistics (from WriteRuntime).

Allows RpeScoreExec to z-score against the full historical distribution instead of only the current write batch (N-H08).

pub fn with_agent_id(self, agent_id: impl Into<String>) -> Self

Set the authenticated agent identity.

pub fn with_nli_classifier(self, clf: Arc<dyn NliClassifier>) -> Self

Inject an NLI classifier for InterferenceDetectorExec Check 3.

Use this to upgrade from the default HeuristicNliClassifier to a DeBERTa-MNLI ONNX model at database open time, without changing any InterferenceConfig fields.

pub fn nli_classifier(&self) -> Option<Arc<dyn NliClassifier>>

Returns the injected NLI classifier, if any.

None — InterferenceDetectorExec will use its own default.

pub fn with_storage(self, storage: Arc<dyn PhysicalStore>) -> Self

Set the storage backend for operators needing vector search.

pub fn with_graph_read_runtime( self, graph_read_runtime: Arc<dyn GraphReadRuntime>, ) -> Self

Set the authoritative graph read runtime.

pub fn with_tokenizer(self, tokenizer: Arc<dyn Tokenizer>) -> Self

Set the tokenizer for operators needing authoritative token counts.

pub fn with_allowed_namespaces(self, namespaces: Option<Vec<String>>) -> Self

Set pre-resolved allowed namespaces.

None means open mode (no filtering). Some(vec) restricts to those namespaces.

pub fn with_query_read_runtime_key(self, key: Option<String>) -> Self

Set a query-scoped runtime handle for compiled terminal read operators.

pub fn with_context_assembly_runtime_key(self, key: Option<String>) -> Self

Set a query-scoped runtime handle for THINK context assembly.

pub fn with_recall_search_binding( self, binding: Option<RecallSearchBinding>, ) -> Self

Set query-scoped compiled recall/search bindings.

pub fn agent_id(&self) -> Option<&str>

Returns the agent ID, if set.

pub fn allowed_namespaces(&self) -> Option<&[String]>

Returns the pre-resolved allowed namespaces.

None = open mode (no filtering), Some(&[]) = deny all.

pub fn get(ctx: &SessionContext) -> Result<Self>

Retrieve HirnSessionExt from a SessionContext.

Returns a clone since SessionContext::state() returns by value.

Errors

Returns an error if the extension was never registered.

pub fn register(self, ctx: &SessionContext) -> Result<()>

Register this extension in a SessionContext.

Errors

Returns an error if the SessionState has already been dropped.

pub fn graph_as<T: Send + Sync + 'static>(&self) -> Option<&T>

Downcast the type-erased graph handle to the concrete type T.

Returns None if the stored graph is not of type T.

pub fn graph_arc<T: Send + Sync + 'static>(&self) -> Option<Arc<T>>

Clone the graph Arc and downcast to Arc<T>.

Returns None if the stored graph is not of type T.

pub fn graph_any(&self) -> &Arc<dyn Any + Send + Sync>

Raw Arc<dyn Any> graph handle.

pub fn graph_read_runtime(&self) -> Option<Arc<dyn GraphReadRuntime>>

Optional graph read runtime.

pub fn query_read_runtime(&self) -> Option<Arc<dyn QueryReadRuntime>>

Optional terminal-read runtime resolved from the query-scoped registry.

pub fn context_assembly_runtime( &self, ) -> Option<Arc<dyn ContextAssemblyRuntime>>

Optional context-assembly runtime resolved from the query-scoped registry.

pub fn recall_search_binding(&self) -> Option<&RecallSearchBinding>

Optional query-scoped compiled recall/search binding.

pub fn embedder(&self) -> Option<&dyn Embedder>

Optional embedder reference.

pub fn embedder_arc(&self) -> Option<Arc<dyn Embedder>>

Optional embedder Arc clone.

pub fn storage(&self) -> Option<&dyn PhysicalStore>

Optional storage reference.

pub fn storage_arc(&self) -> Option<Arc<dyn PhysicalStore>>

Optional storage Arc clone.

pub fn tokenizer(&self) -> Option<&dyn Tokenizer>

Optional tokenizer reference.

pub fn tokenizer_arc(&self) -> Option<Arc<dyn Tokenizer>>

Optional tokenizer Arc clone.

Trait Implementations

impl Clone for HirnSessionExt

fn clone(&self) -> HirnSessionExt

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl ConfigExtension for HirnSessionExt

const PREFIX: &'static str = "hirn"

Configuration namespace prefix to use

impl Debug for HirnSessionExt

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl ExtensionOptions for HirnSessionExt

fn as_any(&self) -> &dyn Any

Return self as Any

fn as_any_mut(&mut self) -> &mut dyn Any

Return self as Any

fn cloned(&self) -> Box<dyn ExtensionOptions>

Return a deep clone of this ExtensionOptions

fn set(&mut self, _key: &str, _value: &str) -> Result<()>

Set the given key, value pair

fn entries(&self) -> Vec<ConfigEntry>

Returns the ConfigEntry stored in this ExtensionOptions