Struct PartitionedTemporalHnsw

pub struct PartitionedTemporalHnsw<D: DistanceMetric> {
    partitions: Vec<Partition<D>>,
    config: PartitionConfig,
    global_entity_index: BTreeMap<u64, Vec<(usize, u32)>>,
    total_points: usize,
    next_global_id: u32,
    metric: D,
}

Time-partitioned temporal HNSW index.

Routes inserts to the correct partition by timestamp, prunes queries to only touch relevant partitions, and merges results globally.

Fields

partitions: Vec<Partition<D>>

Partitions sorted by start time.

config: PartitionConfig

Configuration.

global_entity_index: BTreeMap<u64, Vec<(usize, u32)>>

Global entity index: entity_id → vec of (partition_idx, local_node_id).

total_points: usize

Total number of points across all partitions.

next_global_id: u32

Next global ID to assign.

metric: D

Metric (cloned to create new partitions).

Implementations

impl<D: DistanceMetric + Clone> PartitionedTemporalHnsw<D>

pub fn new(config: PartitionConfig, metric: D) -> Self

Create a new empty partitioned index.

pub fn from_single( index: TemporalHnsw<D>, config: PartitionConfig, metric: D, ) -> Self

Create from an existing TemporalHnsw (migration path).

The entire existing index becomes partition 0.

pub fn insert(&mut self, entity_id: u64, timestamp: i64, vector: &[f32]) -> u32

Insert a temporal point, routing to the correct partition.

pub fn search( &self, query: &[f32], k: usize, filter: TemporalFilter, alpha: f32, query_timestamp: i64, ) -> Vec<(u32, f32)>

Search across relevant partitions, merging results.

pub fn trajectory( &self, entity_id: u64, filter: TemporalFilter, ) -> Vec<(i64, u32)>

Retrieve trajectory for an entity across all partitions.

pub fn vector(&self, global_id: u32) -> Vec<f32>

Get vector by global node ID.

pub fn entity_id(&self, global_id: u32) -> u64

Get entity ID by global node ID.

pub fn timestamp(&self, global_id: u32) -> i64

Get timestamp by global node ID.

pub fn len(&self) -> usize

Total number of indexed points.

pub fn is_empty(&self) -> bool

Whether the index is empty.

pub fn num_partitions(&self) -> usize

Number of partitions.

pub fn partition_info(&self) -> Vec<(i64, i64, usize)>

Get partition info: (start_us, end_us, point_count).

pub fn save(&self, dir: &Path) -> Result<()>

Save all partitions to a directory.

pub fn load(dir: &Path, metric: D) -> Result<Self>

Load partitioned index from a directory.

fn ensure_partition_for(&mut self, timestamp: i64) -> usize

Find or create the partition for a given timestamp.

fn resolve_global_id(&self, global_id: u32) -> (&Partition<D>, u32)

Resolve a global node ID to (partition, local_id).

Trait Implementations

impl<D: DistanceMetric + Clone> TemporalIndexAccess for PartitionedTemporalHnsw<D>

fn search_raw( &self, query: &[f32], k: usize, filter: TemporalFilter, alpha: f32, query_timestamp: i64, ) -> Vec<(u32, f32)>

Search with temporal filtering, returning (node_id, score) pairs.

fn trajectory(&self, entity_id: u64, filter: TemporalFilter) -> Vec<(i64, u32)>

Retrieve trajectory for an entity: (timestamp, node_id) pairs.

fn vector(&self, node_id: u32) -> Vec<f32>

Get the vector for a node. Returns owned vec for thread safety.

fn entity_id(&self, node_id: u32) -> u64

Get the entity_id for a node.

fn timestamp(&self, node_id: u32) -> i64

Get the timestamp for a node.

fn len(&self) -> usize

Number of points in the index.

fn is_empty(&self) -> bool

Whether the index is empty.

fn regions(&self, _level: usize) -> Vec<(u32, Vec<f32>, usize)>

Get semantic regions at a given HNSW level (RFC-004). Returns (hub_node_id, hub_vector, n_assigned) per region.

fn region_members( &self, _region_hub: u32, _level: usize, _filter: TemporalFilter, ) -> Vec<(u32, u64, i64)>

Get points belonging to a specific region, optionally time-filtered (RFC-005). Returns (node_id, entity_id, timestamp) per member.

fn region_assignments( &self, _level: usize, _filter: TemporalFilter, ) -> HashMap<u32, Vec<(u64, i64)>>

Assign all nodes to regions in a single O(N) pass, optionally time-filtered. Returns HashMap<hub_id, Vec<(entity_id, timestamp)>>.

fn region_trajectory( &self, _entity_id: u64, _level: usize, _window_days: i64, _alpha: f32, ) -> Vec<(i64, Vec<f32>)>

Smoothed region-distribution trajectory for an entity (RFC-004).

fn metadata(&self, _node_id: u32) -> HashMap<String, String>

Get metadata for a node. Returns empty map if not available.

fn search_with_metadata( &self, query: &[f32], k: usize, filter: TemporalFilter, alpha: f32, query_timestamp: i64, metadata_filter: &MetadataFilter, ) -> Vec<(u32, f32)>

Search with metadata filtering (post-filter on search results). Default: ignores metadata filter and delegates to search_raw.