Struct Embeddings

pub struct Embeddings {
    pub model_name: String,
    pub max_length: usize,
    /* private fields */
}

Fields

model_name: String

max_length: usize

Implementations

impl Embeddings

pub async fn init(model_name: String, max_length: usize) -> Result<Self>

Initializes the embeddings model.

Errors

Will return an error if the model can’t be initialized.

pub fn embed<'a>(&'a self, text: &'a str) -> Result<HashMap<&'a str, Vec<f32>>>

Embeds a piece of text.

Errors

Will return an error if the text can’t be split into sentences or if the text can’t be embedded.

pub fn embed_sentences<'a>( &self, sentences: Vec<&'a str>, ) -> Result<HashMap<&'a str, Vec<f32>>>

Embeds a list of sentences.

Errors

Will return an error if the sentences can’t be tokenized or if the embeddings can’t be generated.