pub struct EmbeddingsManager { /* private fields */ }Expand description
Main embeddings manager
Implementations§
Source§impl EmbeddingsManager
impl EmbeddingsManager
Sourcepub fn new(config: Option<EmbeddingsConfig>) -> Self
pub fn new(config: Option<EmbeddingsConfig>) -> Self
Create a new embeddings manager
Sourcepub fn is_enabled(&self) -> bool
pub fn is_enabled(&self) -> bool
Check if embeddings are enabled
Sourcepub fn set_repository(&mut self, repo: PathBuf)
pub fn set_repository(&mut self, repo: PathBuf)
Set the current repository
Sourcepub const fn set_intelligence_mode(&mut self, mode: IntelligenceMode)
pub const fn set_intelligence_mode(&mut self, mode: IntelligenceMode)
Set intelligence mode
Sourcepub fn current_model_id(&self) -> Option<String>
pub fn current_model_id(&self) -> Option<String>
Get current model ID
Sourcepub fn current_dimensions(&self) -> Option<usize>
pub fn current_dimensions(&self) -> Option<usize>
Get current dimensions
Sourcepub async fn embed(
&self,
text: &str,
) -> Result<Option<EmbeddingVector>, EmbeddingError>
pub async fn embed( &self, text: &str, ) -> Result<Option<EmbeddingVector>, EmbeddingError>
Embed a single text (returns None if disabled)
Sourcepub async fn embed_batch(
&self,
texts: &[String],
) -> Result<Option<Vec<EmbeddingVector>>, EmbeddingError>
pub async fn embed_batch( &self, texts: &[String], ) -> Result<Option<Vec<EmbeddingVector>>, EmbeddingError>
Embed multiple texts in batch
Sourcepub async fn search_in_index(
&self,
repo: &Path,
model_id: &str,
dimensions: usize,
query: &str,
) -> Result<Vec<SearchResult>, EmbeddingError>
pub async fn search_in_index( &self, repo: &Path, model_id: &str, dimensions: usize, query: &str, ) -> Result<Vec<SearchResult>, EmbeddingError>
Search in the appropriate index for current repo/model
Sourcepub fn stats(&self) -> EmbeddingsStats
pub fn stats(&self) -> EmbeddingsStats
Get statistics about the embeddings system
Auto Trait Implementations§
impl Freeze for EmbeddingsManager
impl !RefUnwindSafe for EmbeddingsManager
impl Send for EmbeddingsManager
impl Sync for EmbeddingsManager
impl Unpin for EmbeddingsManager
impl !UnwindSafe for EmbeddingsManager
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>
Converts
Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>, which can then be
downcast into Box<dyn 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>
Converts
Rc<Trait> (where Trait: Downcast) to Rc<Any>, which 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)
Converts
&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)
Converts
&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> DowncastSend for T
impl<T> DowncastSend for T
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