Struct RagEngine 

pub struct RagEngine { /* private fields */ }

RAG engine for document retrieval and generation

Implementations

impl RagEngine

pub fn new(config: RagConfig, storage: Arc<dyn DocumentStorage>) -> Result<Self>

Create a new RAG engine

pub fn config(&self) -> &RagConfig

Get configuration

pub fn storage(&self) -> &Arc<dyn DocumentStorage>

Get storage backend

pub fn update_config(&mut self, config: RagConfig) -> Result<()>

Update configuration

pub async fn add_document( &self, document_id: String, content: String, metadata: HashMap<String, String>, ) -> Result<()>

Add document to the knowledge base

pub async fn search( &mut self, query: &str, top_k: Option<usize>, ) -> Result<Vec<SearchResult>>

Search for relevant documents

pub async fn generate( &mut self, query: &str, context: Option<&str>, ) -> Result<String>

Generate response using RAG

pub async fn generate_dataset( &mut self, schema: &SchemaDefinition, count: usize, context: Option<&str>, ) -> Result<Vec<HashMap<String, Value>>>

Generate enhanced dataset using RAG

pub async fn get_stats(&self) -> Result<RagStats>

Get engine statistics

Trait Implementations

impl Debug for RagEngine

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

Formats the value using the given formatter.

impl Default for RagEngine

fn default() -> Self

Returns the “default value” for a type.