Struct MockEmbedder 

pub struct MockEmbedder { /* private fields */ }

测试用 Mock Embedder

Implementations

impl MockEmbedder

pub fn new(dimensions: usize, max_batch_size: usize) -> Self

创建新的 MockEmbedder

pub fn expect_embed( &mut self, inputs: Vec<&str>, outputs: Vec<Vec<f32>>, ) -> &mut Self

设置期望输入和返回输出

pub fn set_error(&mut self, error: EmbedError)

设置应返回的错误

pub fn set_output(&mut self, vectors: Vec<Vec<f32>>)

直接设置返回向量

Trait Implementations

impl Debug for MockEmbedder

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

Formats the value using the given formatter.

impl EmbeddingModel for MockEmbedder

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

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

核心嵌入方法。对一批文本生成向量。

fn model_info(&self) -> &EmbedModelInfo

模型信息

fn max_batch_size(&self) -> usize

最大批次大小

fn dimensions(&self) -> usize

向量维度

fn embed_single<'life0, 'life1, 'async_trait>( &'life0 self, text: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Vec<f32>, EmbedError>> + Send + 'async_trait>>

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

便捷方法：对单个文本生成向量