Struct RagContextManager 

pub struct RagContextManager<R: Retriever, G: GroundingStrategy = PrependStrategy> { /* private fields */ }

A ContextManager that retrieves documents and injects them as grounded context.

Implementations

impl<R: Retriever> RagContextManager<R, PrependStrategy>

pub fn new(retriever: R) -> Self

Create a new RagContextManager with PrependStrategy as the default grounding.

impl<R: Retriever, G: GroundingStrategy> RagContextManager<R, G>

pub fn with_grounding<G2: GroundingStrategy>( self, grounding: G2, ) -> RagContextManager<R, G2>

Set the grounding strategy used to format retrieved documents.

pub fn with_max_docs(self, max_docs: usize) -> Self

Maximum number of retrieved documents to inject (default: 5).

Trait Implementations

impl<R: Retriever, G: GroundingStrategy> ContextManager for RagContextManager<R, G>

fn prepare<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, messages: &'life1 mut Vec<Message>, _context_window: usize, _state: &'life2 mut AgentState, ) -> Pin<Box<dyn Future<Output = ()> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Prepare messages: retrieve docs for last user query, prepend as system message.

fn estimate_tokens(&self, messages: &[Message]) -> usize

Estimate the total token count for a message list.