pub struct CachedBatchEmbedder {
pub batch_embedder: BatchEmbedder,
/* private fields */
}Expand description
Cached batch embedder that uses an embedding cache with memory limits
Fields§
§batch_embedder: BatchEmbedderImplementations§
Source§impl CachedBatchEmbedder
impl CachedBatchEmbedder
Sourcepub fn new(batch_embedder: BatchEmbedder) -> Self
pub fn new(batch_embedder: BatchEmbedder) -> Self
Create a new cached batch embedder with default memory limit
Sourcepub fn with_memory_limit(
batch_embedder: BatchEmbedder,
max_memory_mb: usize,
) -> Self
pub fn with_memory_limit( batch_embedder: BatchEmbedder, max_memory_mb: usize, ) -> Self
Create with custom memory limit (in MB)
Sourcepub fn embed_chunks(&mut self, chunks: Vec<Chunk>) -> Result<Vec<EmbeddedChunk>>
pub fn embed_chunks(&mut self, chunks: Vec<Chunk>) -> Result<Vec<EmbeddedChunk>>
Embed chunks using cache when possible
Sourcepub fn embed_chunk(&mut self, chunk: Chunk) -> Result<EmbeddedChunk>
pub fn embed_chunk(&mut self, chunk: Chunk) -> Result<EmbeddedChunk>
Embed a single chunk with caching
Sourcepub fn cache_stats(&self) -> CacheStats
pub fn cache_stats(&self) -> CacheStats
Get cache statistics
Sourcepub fn clear_cache(&self)
pub fn clear_cache(&self)
Clear cache
Sourcepub fn dimensions(&self) -> usize
pub fn dimensions(&self) -> usize
Get embedding dimensions
Auto Trait Implementations§
impl !Freeze for CachedBatchEmbedder
impl !RefUnwindSafe for CachedBatchEmbedder
impl Send for CachedBatchEmbedder
impl Sync for CachedBatchEmbedder
impl Unpin for CachedBatchEmbedder
impl UnsafeUnpin for CachedBatchEmbedder
impl !UnwindSafe for CachedBatchEmbedder
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 more