Struct PyTorchEmbedder 

pub struct PyTorchEmbedder { /* private fields */ }

PyTorch-shaped embedding generator — currently a deterministic mock, not real PyTorch inference. See the pytorch module-level docs.

Implementations

impl PyTorchEmbedder

pub fn new(config: PyTorchConfig) -> Result<Self>

Create a new PyTorch embedder

pub fn load_model(&mut self) -> Result<()>

“Load” a PyTorch model from file.

This does not actually load or run a PyTorch model. It only verifies model_path exists on disk, then populates PyTorchModelMetadata with hardcoded values; no weights are read and no tch/candle-core runtime is invoked. See the module-level docs. Real inference calls (Self::embed_text, Self::embed_batch) log a warning the first time they run against a model “loaded” this way.

pub fn embed_text(&self, text: &str) -> Result<Vector>

Generate embeddings for text

pub fn embed_batch(&self, texts: &[String]) -> Result<Vec<Vector>>

Generate embeddings for multiple texts

pub fn get_metadata(&self) -> Option<&PyTorchModelMetadata>

Get model metadata

pub fn get_dimensions(&self) -> Option<usize>

Get embedding dimensions

pub fn set_tokenizer(&mut self, tokenizer: PyTorchTokenizer)

Update tokenizer

pub fn supports_mixed_precision(&self) -> bool

Check if model supports mixed precision

pub fn get_device(&self) -> &PyTorchDevice

Get current device

Trait Implementations

impl Debug for PyTorchEmbedder

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

Formats the value using the given formatter.

impl EmbeddingGenerator for PyTorchEmbedder

fn generate_embedding(&self, content: &ContentItem) -> Result<Vector>

Generate embedding vector from content

fn generate_batch_embeddings( &self, content: &[ContentItem], ) -> Result<Vec<ProcessingResult>>

Generate embeddings for a batch of content items

fn embedding_dimensions(&self) -> usize

Get the embedding dimensions

fn get_config(&self) -> Value

Get generator configuration

fn is_ready(&self) -> bool

Check if the generator is ready

fn get_statistics(&self) -> GeneratorStatistics

Get generator statistics