Struct MockEmbeddingProvider 

pub struct MockEmbeddingProvider { /* private fields */ }

Mock embedding provider for testing and development

Provides deterministic, fast embedding generation for testing purposes. The mock provider generates consistent embeddings based on input text, making it suitable for unit tests and development workflows.

Features

• Deterministic: Same input always produces the same embedding
• Fast: No network calls or heavy computation
• Configurable: Adjustable dimensions and behavior
• Testing-Friendly: Predictable behavior for assertions

Example

use rrag::prelude::*;
use std::sync::Arc;

let provider = Arc::new(MockEmbeddingProvider::new());
let service = EmbeddingService::new(provider);

let document = Document::new("Test content");
let embedding = service.embed_document(&document).await?;

// Mock provider always returns 384 dimensions
assert_eq!(embedding.dimensions, 384);
assert_eq!(embedding.model, "mock-model");

Implementations

impl MockEmbeddingProvider

pub fn new() -> Self

Create a new mock embedding provider

Trait Implementations

impl EmbeddingProvider for MockEmbeddingProvider

fn name(&self) -> &str

Provider name (e.g., “openai”, “huggingface”)

fn supported_models(&self) -> Vec<&str>

Supported models for this provider

fn max_batch_size(&self) -> usize

Maximum batch size supported

fn embedding_dimensions(&self) -> usize

Embedding dimensions for the current model

fn embed_text<'life0, 'life1, 'async_trait>( &'life0 self, text: &'life1 str, ) -> Pin<Box<dyn Future<Output = RragResult<Embedding>> + Send + 'async_trait>>

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

Generate embedding for a single text

fn embed_batch<'life0, 'async_trait>( &'life0 self, requests: Vec<EmbeddingRequest>, ) -> Pin<Box<dyn Future<Output = RragResult<EmbeddingBatch>> + Send + 'async_trait>>

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

Generate embeddings for multiple texts (more efficient)

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

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

Health check for the provider