Struct RetrievalContext 

pub struct RetrievalContext {
    pub query: String,
    pub chunks: Vec<ContextChunk>,
    pub overall_relevance: f32,
    pub sources_used: Vec<ChunkSource>,
    pub retrieval_time_us: u64,
    pub explanation: Option<String>,
}

Complete context retrieved for a query

Fields

query: String

Original query

chunks: Vec<ContextChunk>

Retrieved context chunks

overall_relevance: f32

Overall relevance score

sources_used: Vec<ChunkSource>

Sources used in retrieval

retrieval_time_us: u64

Total retrieval time in microseconds

explanation: Option<String>

Explanation of retrieval strategy

Implementations

impl RetrievalContext

pub fn new(query: impl Into<String>) -> Self

Create a new retrieval context

pub fn add_chunk(&mut self, chunk: ContextChunk)

Add a chunk to the context

pub fn sort_by_relevance(&mut self)

Sort chunks by relevance (descending)

pub fn calculate_overall_relevance(&mut self)

Calculate overall relevance from chunks

pub fn limit(&mut self, n: usize)

Limit to top N chunks

pub fn chunks_for_type(&self, entity_type: EntityType) -> Vec<&ContextChunk>

Get chunks for a specific entity type

pub fn chunks_from_source(&self, source: &ChunkSource) -> Vec<&ContextChunk>

Get chunks from a specific source

pub fn is_empty(&self) -> bool

Check if context is empty

pub fn len(&self) -> usize

Number of chunks

pub fn top_chunk(&self) -> Option<&ContextChunk>

Get the top chunk

pub fn to_context_string(&self) -> String

Convert to a text representation for LLM context

pub fn entity_ids(&self) -> Vec<&str>

Get entity IDs mentioned in the context

pub fn merge(&mut self, other: RetrievalContext)

Merge another context into this one

pub fn with_explanation(self, explanation: impl Into<String>) -> Self

Set explanation

Trait Implementations

impl Clone for RetrievalContext

fn clone(&self) -> RetrievalContext

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RetrievalContext

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

Formats the value using the given formatter.