Struct RagEngine 

pub struct RagEngine { /* private fields */ }

RAG engine for enhanced data generation

Implementations

impl RagEngine

pub fn new(config: RagConfig) -> Self

Create a new RAG engine

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

Add a document to the knowledge base

pub fn add_schema(&mut self, schema: &SchemaDefinition) -> Result<()>

Add schema information to knowledge base

pub async fn generate_with_rag( &self, schema: &SchemaDefinition, config: &DataConfig, ) -> Result<Vec<Value>>

Generate data using RAG-augmented prompts

pub fn keyword_search(&self, query: &str, limit: usize) -> Vec<&DocumentChunk>

Perform keyword-based search (fallback)

pub async fn compute_embeddings(&mut self) -> Result<()>

Compute embeddings for all document chunks

pub fn update_config(&mut self, config: RagConfig)

Update RAG configuration

pub fn config(&self) -> &RagConfig

Get current configuration

pub fn chunk_count(&self) -> usize

Get number of indexed chunks

pub fn schema_count(&self) -> usize

Get number of indexed schemas

pub fn get_chunk(&self, index: usize) -> Option<&DocumentChunk>

Get chunk by index

pub fn has_schema(&self, name: &str) -> bool

Check if schema exists in knowledge base

pub async fn generate_text(&self, prompt: &str) -> Result<String>

Generate text using LLM (for intelligent mock generation)

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.