Skip to main content

FastEmbedder

Struct FastEmbedder 

Source
pub struct FastEmbedder { /* private fields */ }
Expand description

Local CPU embedder backed by fastembed-rs (ONNX runtime, all-MiniLM-L6-v2).

Why: Default to local-only embeddings so consumers have zero external network dependency and predictable latency. The LRU cache keeps the hot path free of redundant ONNX work for repeat strings (queries, common chunks). What: wraps a single TextEmbedding behind a parking_lot::Mutex (the underlying embed requires &mut self) and an LruCache<String, Vec<f32>>. Initialisation warms the ORT graph with a small batch so the first user query doesn’t pay the one-shot compile cost. Test: embed_batch_returns_correct_dim and cache_hit_is_idempotent (marked #[ignore] — they download a real model).

Implementations§

Source§

impl FastEmbedder

Source

pub async fn new() -> Result<Self>

Construct a new FastEmbedder with the default cache size.

Source

pub fn provider(&self) -> ExecutionProvider

Identifier for the execution provider this embedder is actually using.

Why: callers (e.g. trusty-search startup logs) want to surface whether the daemon is running on CPU or GPU/ANE without poking at internals. What: returns ExecutionProvider::CoreML on Apple Silicon (when EP registration succeeded), otherwise Cpu (or Cuda if/when wired). Test: covered by the public-surface compile check.

Source

pub async fn with_cache_size(capacity: usize) -> Result<Self>

Construct with an explicit LRU capacity.

Trait Implementations§

Source§

impl Embedder for FastEmbedder

Source§

fn embed_batch<'life0, 'life1, 'async_trait>( &'life0 self, texts: &'life1 [String], ) -> Pin<Box<dyn Future<Output = Result<Vec<Vec<f32>>>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Embed a batch of texts. Returns one Vec<f32> per input, each of length self.dimension(). An empty input batch returns an empty Vec.
Source§

fn dimension(&self) -> usize

Output dimension of the produced embeddings.

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

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 more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

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
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more