Struct EmbeddingGenerator 

pub struct EmbeddingGenerator {
    pub skip_existing: bool,
    /* private fields */
}

Embedding generator using sentence-transformers.

Fields

skip_existing: bool

If true, skip entities that already have real (non-zero) embeddings.

Implementations

impl EmbeddingGenerator

pub fn new() -> Self

Create a new embedding generator with default configuration.

pub fn with_config(config: EmbeddingConfig) -> Self

Create a new embedding generator with custom configuration.

pub fn with_force(self, force: bool) -> Self

Set force mode: if true, regenerate embeddings even for entities that already have real (non-zero) vectors.

pub fn generate_embeddings(&self, texts: Vec<String>) -> Result<Vec<Vec<f32>>>

Generate embeddings for a list of texts.

pub fn generate_for_papers(&self, conn: &Connection) -> Result<EmbeddingStats>

Generate embeddings for all paper entities in the database.

pub fn generate_for_skills(&self, conn: &Connection) -> Result<EmbeddingStats>

Generate embeddings for all skill entities in the database.

pub fn generate_for_all(&self, conn: &Connection) -> Result<EmbeddingStats>

Generate embeddings for all entities in the database.

Trait Implementations

impl Default for EmbeddingGenerator

fn default() -> Self

Returns the “default value” for a type.