Struct RagAiGenerator 

pub struct RagAiGenerator { /* private fields */ }

RAG-based AI generator that uses the mockforge-data RAG engine

Implementations

impl RagAiGenerator

pub fn new(rag_config: RagConfig) -> Result<Self>

Create a new RAG-based AI generator

Arguments

• rag_config - Configuration for the RAG engine (provider, model, API key, etc.)

Returns

A new RagAiGenerator instance

pub fn from_env() -> Result<Self>

Create a RAG AI generator from environment variables

Reads configuration from:

• MOCKFORGE_AI_PROVIDER: LLM provider (openai, anthropic, ollama, etc.)
• MOCKFORGE_AI_API_KEY: API key for the LLM provider
• MOCKFORGE_AI_MODEL: Model name (e.g., gpt-4, claude-3-opus)
• MOCKFORGE_AI_ENDPOINT: API endpoint (optional, uses provider default)
• MOCKFORGE_AI_TEMPERATURE: Temperature for generation (optional, default: 0.7)
• MOCKFORGE_AI_MAX_TOKENS: Max tokens for generation (optional, default: 1024)

Trait Implementations

impl AiGenerator for RagAiGenerator

fn generate<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, prompt: &'life1 str, config: &'life2 AiResponseConfig, ) -> Pin<Box<dyn Future<Output = Result<Value>> + Send + 'async_trait>>

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

Generate an AI response from a prompt