Struct MockRetriever 

pub struct MockRetriever { /* private fields */ }

Mock retriever for testing

Implementations

impl MockRetriever

pub fn new() -> Self

Create a new mock retriever

pub fn with_results(self, results: Vec<String>) -> Self

Pre-populate with results for testing

Trait Implementations

impl AsyncRetriever for MockRetriever

type Query = String

The query type this retriever accepts

type Result = String

The result type this retriever returns

type Error = GraphRAGError

The error type returned by retrieval operations

fn search<'life0, 'async_trait>( &'life0 self, _query: Self::Query, k: usize, ) -> Pin<Box<dyn Future<Output = Result<Vec<Self::Result>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Perform a search query

fn search_with_context<'life0, 'life1, 'async_trait>( &'life0 self, query: Self::Query, _context: &'life1 str, k: usize, ) -> Pin<Box<dyn Future<Output = Result<Vec<Self::Result>>> + Send + 'async_trait>>

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

Perform a search with additional context

fn update<'life0, 'async_trait>( &'life0 mut self, content: Vec<String>, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Update the retriever with new content

fn health_check<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<bool>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Health check for retrieval system

fn search_batch<'life0, 'async_trait>( &'life0 self, queries: Vec<Self::Query>, k: usize, ) -> Pin<Box<dyn Future<Output = Result<Vec<Vec<Self::Result>>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Perform multiple search queries concurrently

fn update_batch<'life0, 'async_trait>( &'life0 mut self, content_batches: Vec<Vec<String>>, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Update the retriever with new content in batches

fn refresh_index<'life0, 'async_trait>( &'life0 mut self, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Refresh/rebuild the retrieval index

fn get_stats<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<RetrievalStats>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Get retrieval statistics

impl Default for MockRetriever

fn default() -> Self

Returns the “default value” for a type.