Struct ScoreFn 

pub struct ScoreFn(pub Arc<dyn Fn(f32, &RecordBatch) -> f32 + Send + Sync>);

Injectable per-result scoring function for hybrid ranking.

Called after HNSW retrieval with the HNSW distance and a single-row RecordBatch containing all Parquet columns for that result. Returns a replacement score (lower = better rank, same convention as distance).

Typical use: combine HNSW distance with recency and importance signals from the episodic_columns for agent memory tables:

use ailake_core::{hybrid_score, episodic_columns};
use ailake_query::scanner::ScoreFn;
use arrow_array::{RecordBatch, cast::AsArray};
use arrow_array::types::Float32Type;

let score_fn = ScoreFn::new(|distance, row| {
    let recency = row
        .column_by_name(episodic_columns::RECENCY_WEIGHT)
        .and_then(|c| c.as_primitive_opt::<Float32Type>())
        .and_then(|a| a.iter().next().flatten())
        .unwrap_or(1.0);
    let importance = row
        .column_by_name(episodic_columns::IMPORTANCE_SCORE)
        .and_then(|c| c.as_primitive_opt::<Float32Type>())
        .and_then(|a| a.iter().next().flatten())
        .unwrap_or(1.0);
    hybrid_score(distance, recency, importance)
});

Tuple Fields

0: Arc<dyn Fn(f32, &RecordBatch) -> f32 + Send + Sync>

Implementations

impl ScoreFn

pub fn new(f: impl Fn(f32, &RecordBatch) -> f32 + Send + Sync + 'static) -> Self

pub fn call(&self, distance: f32, row: &RecordBatch) -> f32

Trait Implementations

impl Clone for ScoreFn

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ScoreFn

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

Formats the value using the given formatter.