Struct ContextBuilder 

pub struct ContextBuilder { /* private fields */ }

Builds LLM context from a knowledge graph by extracting relevant subgraphs.

Implementations

impl ContextBuilder

pub fn new() -> Self

Create a builder with default configuration.

pub fn with_config(config: ContextBuilderConfig) -> Self

Create with a custom configuration.

pub fn extract_neighborhood( &self, kg: &KnowledgeGraph, entity: &str, max_hops: Option<usize>, ) -> Vec<ContextTriple>

Extract the N-hop neighborhood around entity.

pub fn entity_neighborhood( &self, kg: &KnowledgeGraph, entity: &str, ) -> Vec<ContextTriple>

Retrieve direct connections of an entity (1-hop neighborhood).

pub fn relation_context( &self, kg: &KnowledgeGraph, predicate: &str, ) -> Vec<ContextTriple>

Retrieve all triples involving a specific predicate.

pub fn rank_triples( &self, triples: &[ContextTriple], seed_entities: &[&str], ) -> Vec<ScoredTriple>

Rank triples by relevance to a set of seed entities.

Triples mentioning more seed entities score higher.

pub fn truncate_to_budget( &self, triples: &[ContextTriple], ) -> Vec<ContextTriple>

Truncate a list of triples to fit within the token budget.

pub fn format_triples(&self, triples: &[ContextTriple]) -> String

Format a list of triples using the configured template.

pub fn merge_contexts( &self, contexts: &[Vec<ContextTriple>], ) -> Vec<ContextTriple>

Merge contexts extracted for multiple entities, deduplicating triples.

pub fn deduplicate(&self, triples: &[ContextTriple]) -> Vec<ContextTriple>

Deduplicate a list of triples (preserving order).

pub fn build(&self, kg: &KnowledgeGraph, entity: &str) -> BuiltContext

Build context for a single entity.

pub fn build_multi( &self, kg: &KnowledgeGraph, entities: &[&str], ) -> BuiltContext

Build context for multiple entities, merging and deduplicating.

pub fn config(&self) -> &ContextBuilderConfig

Access the configuration.

Trait Implementations

impl Default for ContextBuilder

fn default() -> Self

Returns the “default value” for a type.