pub struct FastEmbedder { /* private fields */ }Expand description
Fast embedding model using fastembed library
Implementations§
Source§impl FastEmbedder
impl FastEmbedder
Sourcepub fn with_model(model_type: ModelType) -> Result<Self>
pub fn with_model(model_type: ModelType) -> Result<Self>
Create a new embedder with specified model
Sourcepub fn with_cache_dir(
model_type: ModelType,
cache_dir: Option<&Path>,
) -> Result<Self>
pub fn with_cache_dir( model_type: ModelType, cache_dir: Option<&Path>, ) -> Result<Self>
Create a new embedder with specified model and cache directory
Sourcepub fn embed_batch(&mut self, texts: Vec<String>) -> Result<Vec<Vec<f32>>>
pub fn embed_batch(&mut self, texts: Vec<String>) -> Result<Vec<Vec<f32>>>
Embed a batch of texts (processes in mini-batches to avoid OOM) Uses adaptive batch size based on model dimensions Can be overridden with CODESEARCH_BATCH_SIZE environment variable
Sourcepub fn embed_batch_chunked(
&mut self,
texts: Vec<String>,
batch_size: usize,
) -> Result<Vec<Vec<f32>>>
pub fn embed_batch_chunked( &mut self, texts: Vec<String>, batch_size: usize, ) -> Result<Vec<Vec<f32>>>
Embed a batch of texts with configurable mini-batch size
Sourcepub fn dimensions(&self) -> usize
pub fn dimensions(&self) -> usize
Get the dimensionality of embeddings
Sourcepub fn model_name(&self) -> &str
pub fn model_name(&self) -> &str
Get the model name
Sourcepub fn model_type(&self) -> ModelType
pub fn model_type(&self) -> ModelType
Get the model type
Trait Implementations§
Auto Trait Implementations§
impl !Freeze for FastEmbedder
impl !RefUnwindSafe for FastEmbedder
impl Send for FastEmbedder
impl Sync for FastEmbedder
impl Unpin for FastEmbedder
impl UnsafeUnpin for FastEmbedder
impl !UnwindSafe for FastEmbedder
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
Source§fn into_any(self: Box<T>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
Convert
Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>. Box<dyn Any> can
then be further downcast into Box<ConcreteType> where ConcreteType implements Trait.Source§fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
Convert
Rc<Trait> (where Trait: Downcast) to Rc<Any>. Rc<Any> can then be
further downcast into Rc<ConcreteType> where ConcreteType implements Trait.Source§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Convert
&Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &Any’s vtable from &Trait’s.Source§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
Convert
&mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &mut Any’s vtable from &mut Trait’s.Source§impl<T> DowncastSync for T
impl<T> DowncastSync for T
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> PolicyExt for Twhere
T: ?Sized,
Source§impl<R, P> ReadPrimitive<R> for P
impl<R, P> ReadPrimitive<R> for P
Source§fn read_from_little_endian(read: &mut R) -> Result<Self, Error>
fn read_from_little_endian(read: &mut R) -> Result<Self, Error>
Read this value from the supplied reader. Same as
ReadEndian::read_from_little_endian().