Struct KnowledgeBase 

pub struct KnowledgeBase<E: Embedder, S: VectorStore> { /* private fields */ }

Implementations

impl<E: Embedder, S: VectorStore> KnowledgeBase<E, S>

pub fn new(embedder: Arc<E>, store: Arc<S>) -> Self

pub fn with_reranker( self, reranker: Arc<dyn Fn(&ScoredDocument) -> f32 + Send + Sync>, ) -> Self

pub fn with_chunker(self, chunker: Arc<dyn DocumentChunker>) -> Self

pub fn with_config(self, config: RetrievalConfig) -> Self

pub fn config(&self) -> &RetrievalConfig

pub async fn add_document(&self, document: Document) -> Result<()>

pub async fn retrieve( &self, query: &str, top_k: usize, ) -> Result<Vec<ScoredDocument>>

pub async fn retrieve_with_overrides( &self, query: &str, overrides: RetrievalOverrides, ) -> Result<Vec<ScoredDocument>>

pub async fn evaluate( &self, query: &str, relevant_document_ids: &[String], overrides: RetrievalOverrides, ) -> Result<RetrievalEvaluation>

Trait Implementations

impl<E, S> Retriever for KnowledgeBase<E, S>

where E: Embedder, S: VectorStore,

fn retrieve<'life0, 'life1, 'async_trait>( &'life0 self, query: &'life1 str, top_k: usize, ) -> Pin<Box<dyn Future<Output = Result<Vec<String>>> + Send + 'async_trait>>

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