Struct MockLanguageModel 

pub struct MockLanguageModel { /* private fields */ }

Mock language model for testing

Implementations

impl MockLanguageModel

pub fn new() -> Self

Create a new mock language model

pub fn with_response( self, prompt: impl Into<String>, response: impl Into<String>, ) -> Self

Set a specific response for a prompt

pub fn with_default_response(self, response: impl Into<String>) -> Self

Set the default response for unmatched prompts

Trait Implementations

impl AsyncLanguageModel for MockLanguageModel

type Error = GraphRAGError

The error type returned by generation operations

fn complete<'life0, 'life1, 'async_trait>( &'life0 self, prompt: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<String>> + Send + 'async_trait>>

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

Generate text completion

fn complete_with_params<'life0, 'life1, 'async_trait>( &'life0 self, prompt: &'life1 str, _params: GenerationParams, ) -> Pin<Box<dyn Future<Output = Result<String>> + Send + 'async_trait>>

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

Generate text with custom parameters

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

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

Check if the model is available

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

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

Get model information

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

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

Get model usage statistics

fn complete_batch<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, prompts: &'life1 [&'life2 str], ) -> Pin<Box<dyn Future<Output = Result<Vec<String>>> + Send + 'async_trait>>

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

Generate multiple text completions concurrently

fn complete_batch_concurrent<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, prompts: &'life1 [&'life2 str], max_concurrent: usize, ) -> Pin<Box<dyn Future<Output = Result<Vec<String>>> + Send + 'async_trait>>

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

Generate multiple text completions with concurrency control

fn complete_streaming<'life0, 'life1, 'async_trait>( &'life0 self, prompt: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Pin<Box<dyn Stream<Item = Result<String>> + Send>>>> + Send + 'async_trait>>

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

Generate streaming completion (if supported)

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 language model service

fn estimate_tokens<'life0, 'life1, 'async_trait>( &'life0 self, prompt: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<usize>> + Send + 'async_trait>>

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

Estimate tokens for prompt

impl Clone for MockLanguageModel

fn clone(&self) -> MockLanguageModel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for MockLanguageModel

fn default() -> Self

Returns the “default value” for a type.