pub struct TieredCache<B: BqSearchBackend, S: StorageLoader> { /* private fields */ }Expand description
Two-tier cache combining L1 (exact) and L2 (semantic).
Implementations§
Source§impl<B: BqSearchBackend, S: StorageLoader> TieredCache<B, S>
impl<B: BqSearchBackend, S: StorageLoader> TieredCache<B, S>
Sourcepub fn new(l1: L1CacheHandle, l2: L2SemanticCache<B, S>) -> Self
pub fn new(l1: L1CacheHandle, l2: L2SemanticCache<B, S>) -> Self
Creates a tiered cache from an L1 handle and an L2 cache.
Sourcepub fn l1(&self) -> &L1CacheHandle
pub fn l1(&self) -> &L1CacheHandle
Returns the L1 handle.
Sourcepub fn l2(&self) -> &L2SemanticCache<B, S>
pub fn l2(&self) -> &L2SemanticCache<B, S>
Returns the L2 cache.
Sourcepub async fn lookup(
&self,
prompt: &str,
tenant_id: u64,
) -> L2CacheResult<TieredLookupResult>
pub async fn lookup( &self, prompt: &str, tenant_id: u64, ) -> L2CacheResult<TieredLookupResult>
Looks up prompt in L1 (exact) then L2 (semantic).
Sourcepub async fn lookup_with_semantic_query(
&self,
exact_key: &str,
semantic_query: &str,
tenant_id: u64,
) -> L2CacheResult<TieredLookupResult>
pub async fn lookup_with_semantic_query( &self, exact_key: &str, semantic_query: &str, tenant_id: u64, ) -> L2CacheResult<TieredLookupResult>
Looks up with separate exact key (L1) and semantic query (L2).
Sourcepub fn insert_l1(
&self,
prompt: &str,
tenant_id: u64,
handle: MmapFileHandle,
) -> [u8; 32]
pub fn insert_l1( &self, prompt: &str, tenant_id: u64, handle: MmapFileHandle, ) -> [u8; 32]
Inserts an entry into L1 only (tenant-scoped).
Sourcepub async fn index_l2(
&self,
prompt: &str,
tenant_id: u64,
context_hash: u64,
storage_key: &str,
timestamp: i64,
) -> L2CacheResult<u64>
pub async fn index_l2( &self, prompt: &str, tenant_id: u64, context_hash: u64, storage_key: &str, timestamp: i64, ) -> L2CacheResult<u64>
Indexes an entry into L2 only.
Sourcepub async fn insert_both(
&self,
prompt: &str,
tenant_id: u64,
context_hash: u64,
storage_key: &str,
timestamp: i64,
handle: MmapFileHandle,
) -> L2CacheResult<([u8; 32], u64)>
pub async fn insert_both( &self, prompt: &str, tenant_id: u64, context_hash: u64, storage_key: &str, timestamp: i64, handle: MmapFileHandle, ) -> L2CacheResult<([u8; 32], u64)>
Inserts into L1 and indexes into L2.
Sourcepub fn remove_l1(&self, prompt: &str, tenant_id: u64) -> Option<MmapFileHandle>
pub fn remove_l1(&self, prompt: &str, tenant_id: u64) -> Option<MmapFileHandle>
Removes an entry from L1.
Sourcepub fn contains_l1(&self, prompt: &str, tenant_id: u64) -> bool
pub fn contains_l1(&self, prompt: &str, tenant_id: u64) -> bool
Returns true if L1 contains a key for this tenant+prompt.
Sourcepub fn l1_is_empty(&self) -> bool
pub fn l1_is_empty(&self) -> bool
Returns true if L1 is empty.
Sourcepub fn run_pending_tasks_l1(&self)
pub fn run_pending_tasks_l1(&self)
Runs any pending L1 maintenance tasks.
Source§impl TieredCache<MockBqClient, MockStorageLoader>
impl TieredCache<MockBqClient, MockStorageLoader>
Sourcepub async fn new_mock() -> L2CacheResult<Self>
Available on crate features mock only.
pub async fn new_mock() -> L2CacheResult<Self>
mock only.Creates a ready-to-use mock cache with default L2 config.
Sourcepub async fn new_mock_with_config(l2_config: L2Config) -> L2CacheResult<Self>
Available on crate features mock only.
pub async fn new_mock_with_config(l2_config: L2Config) -> L2CacheResult<Self>
mock only.Creates a ready-to-use mock cache with a custom L2 config.
Sourcepub fn mock_storage(&self) -> &MockStorageLoader
Available on crate features mock only.
pub fn mock_storage(&self) -> &MockStorageLoader
mock only.Returns the mock storage loader.
Sourcepub fn mock_bq_backend(&self) -> &MockBqClient
Available on crate features mock only.
pub fn mock_bq_backend(&self) -> &MockBqClient
mock only.Returns the mock BQ backend.
Trait Implementations§
Source§impl<B: BqSearchBackend, S: StorageLoader> Debug for TieredCache<B, S>
impl<B: BqSearchBackend, S: StorageLoader> Debug for TieredCache<B, S>
Auto Trait Implementations§
impl<B, S> Freeze for TieredCache<B, S>
impl<B, S> !RefUnwindSafe for TieredCache<B, S>
impl<B, S> Send for TieredCache<B, S>
impl<B, S> Sync for TieredCache<B, S>
impl<B, S> Unpin for TieredCache<B, S>
impl<B, S> !UnwindSafe for TieredCache<B, S>
Blanket Implementations§
Source§impl<T> ArchivePointee for T
impl<T> ArchivePointee for T
Source§type ArchivedMetadata = ()
type ArchivedMetadata = ()
The archived version of the pointer metadata for this type.
Source§fn pointer_metadata(
_: &<T as ArchivePointee>::ArchivedMetadata,
) -> <T as Pointee>::Metadata
fn pointer_metadata( _: &<T as ArchivePointee>::ArchivedMetadata, ) -> <T as Pointee>::Metadata
Converts some archived metadata to the pointer metadata for itself.
§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
§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> FmtForward for T
impl<T> FmtForward for T
Source§fn fmt_binary(self) -> FmtBinary<Self>where
Self: Binary,
fn fmt_binary(self) -> FmtBinary<Self>where
Self: Binary,
Causes
self to use its Binary implementation when Debug-formatted.Source§fn fmt_display(self) -> FmtDisplay<Self>where
Self: Display,
fn fmt_display(self) -> FmtDisplay<Self>where
Self: Display,
Causes
self to use its Display implementation when
Debug-formatted.Source§fn fmt_lower_exp(self) -> FmtLowerExp<Self>where
Self: LowerExp,
fn fmt_lower_exp(self) -> FmtLowerExp<Self>where
Self: LowerExp,
Causes
self to use its LowerExp implementation when
Debug-formatted.Source§fn fmt_lower_hex(self) -> FmtLowerHex<Self>where
Self: LowerHex,
fn fmt_lower_hex(self) -> FmtLowerHex<Self>where
Self: LowerHex,
Causes
self to use its LowerHex implementation when
Debug-formatted.Source§fn fmt_octal(self) -> FmtOctal<Self>where
Self: Octal,
fn fmt_octal(self) -> FmtOctal<Self>where
Self: Octal,
Causes
self to use its Octal implementation when Debug-formatted.Source§fn fmt_pointer(self) -> FmtPointer<Self>where
Self: Pointer,
fn fmt_pointer(self) -> FmtPointer<Self>where
Self: Pointer,
Causes
self to use its Pointer implementation when
Debug-formatted.Source§fn fmt_upper_exp(self) -> FmtUpperExp<Self>where
Self: UpperExp,
fn fmt_upper_exp(self) -> FmtUpperExp<Self>where
Self: UpperExp,
Causes
self to use its UpperExp implementation when
Debug-formatted.Source§fn fmt_upper_hex(self) -> FmtUpperHex<Self>where
Self: UpperHex,
fn fmt_upper_hex(self) -> FmtUpperHex<Self>where
Self: UpperHex,
Causes
self to use its UpperHex implementation when
Debug-formatted.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> IntoRequest<T> for T
impl<T> IntoRequest<T> for T
Source§fn into_request(self) -> Request<T>
fn into_request(self) -> Request<T>
Wrap the input message
T in a tonic::RequestSource§impl<T> LayoutRaw for T
impl<T> LayoutRaw for T
Source§fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
Returns the layout of the type.
Source§impl<T, N1, N2> Niching<NichedOption<T, N1>> for N2
impl<T, N1, N2> Niching<NichedOption<T, N1>> for N2
Source§unsafe fn is_niched(niched: *const NichedOption<T, N1>) -> bool
unsafe fn is_niched(niched: *const NichedOption<T, N1>) -> bool
Returns whether the given value has been niched. Read more
Source§fn resolve_niched(out: Place<NichedOption<T, N1>>)
fn resolve_niched(out: Place<NichedOption<T, N1>>)
Writes data to
out indicating that a T is niched.Source§impl<T> Pipe for Twhere
T: ?Sized,
impl<T> Pipe for Twhere
T: ?Sized,
Source§fn pipe<R>(self, func: impl FnOnce(Self) -> R) -> Rwhere
Self: Sized,
fn pipe<R>(self, func: impl FnOnce(Self) -> R) -> Rwhere
Self: Sized,
Pipes by value. This is generally the method you want to use. Read more
Source§fn pipe_ref<'a, R>(&'a self, func: impl FnOnce(&'a Self) -> R) -> Rwhere
R: 'a,
fn pipe_ref<'a, R>(&'a self, func: impl FnOnce(&'a Self) -> R) -> Rwhere
R: 'a,
Borrows
self and passes that borrow into the pipe function. Read moreSource§fn pipe_ref_mut<'a, R>(&'a mut self, func: impl FnOnce(&'a mut Self) -> R) -> Rwhere
R: 'a,
fn pipe_ref_mut<'a, R>(&'a mut self, func: impl FnOnce(&'a mut Self) -> R) -> Rwhere
R: 'a,
Mutably borrows
self and passes that borrow into the pipe function. Read moreSource§fn pipe_borrow<'a, B, R>(&'a self, func: impl FnOnce(&'a B) -> R) -> R
fn pipe_borrow<'a, B, R>(&'a self, func: impl FnOnce(&'a B) -> R) -> R
Source§fn pipe_borrow_mut<'a, B, R>(
&'a mut self,
func: impl FnOnce(&'a mut B) -> R,
) -> R
fn pipe_borrow_mut<'a, B, R>( &'a mut self, func: impl FnOnce(&'a mut B) -> R, ) -> R
Source§fn pipe_as_ref<'a, U, R>(&'a self, func: impl FnOnce(&'a U) -> R) -> R
fn pipe_as_ref<'a, U, R>(&'a self, func: impl FnOnce(&'a U) -> R) -> R
Borrows
self, then passes self.as_ref() into the pipe function.Source§fn pipe_as_mut<'a, U, R>(&'a mut self, func: impl FnOnce(&'a mut U) -> R) -> R
fn pipe_as_mut<'a, U, R>(&'a mut self, func: impl FnOnce(&'a mut U) -> R) -> R
Mutably borrows
self, then passes self.as_mut() into the pipe
function.Source§fn pipe_deref<'a, T, R>(&'a self, func: impl FnOnce(&'a T) -> R) -> R
fn pipe_deref<'a, T, R>(&'a self, func: impl FnOnce(&'a T) -> R) -> R
Borrows
self, then passes self.deref() into the pipe function.Source§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<T> Tap for T
impl<T> Tap for T
Source§fn tap_borrow<B>(self, func: impl FnOnce(&B)) -> Self
fn tap_borrow<B>(self, func: impl FnOnce(&B)) -> Self
Immutable access to the
Borrow<B> of a value. Read moreSource§fn tap_borrow_mut<B>(self, func: impl FnOnce(&mut B)) -> Self
fn tap_borrow_mut<B>(self, func: impl FnOnce(&mut B)) -> Self
Mutable access to the
BorrowMut<B> of a value. Read moreSource§fn tap_ref<R>(self, func: impl FnOnce(&R)) -> Self
fn tap_ref<R>(self, func: impl FnOnce(&R)) -> Self
Immutable access to the
AsRef<R> view of a value. Read moreSource§fn tap_ref_mut<R>(self, func: impl FnOnce(&mut R)) -> Self
fn tap_ref_mut<R>(self, func: impl FnOnce(&mut R)) -> Self
Mutable access to the
AsMut<R> view of a value. Read moreSource§fn tap_deref<T>(self, func: impl FnOnce(&T)) -> Self
fn tap_deref<T>(self, func: impl FnOnce(&T)) -> Self
Immutable access to the
Deref::Target of a value. Read moreSource§fn tap_deref_mut<T>(self, func: impl FnOnce(&mut T)) -> Self
fn tap_deref_mut<T>(self, func: impl FnOnce(&mut T)) -> Self
Mutable access to the
Deref::Target of a value. Read moreSource§fn tap_dbg(self, func: impl FnOnce(&Self)) -> Self
fn tap_dbg(self, func: impl FnOnce(&Self)) -> Self
Calls
.tap() only in debug builds, and is erased in release builds.Source§fn tap_mut_dbg(self, func: impl FnOnce(&mut Self)) -> Self
fn tap_mut_dbg(self, func: impl FnOnce(&mut Self)) -> Self
Calls
.tap_mut() only in debug builds, and is erased in release
builds.Source§fn tap_borrow_dbg<B>(self, func: impl FnOnce(&B)) -> Self
fn tap_borrow_dbg<B>(self, func: impl FnOnce(&B)) -> Self
Calls
.tap_borrow() only in debug builds, and is erased in release
builds.Source§fn tap_borrow_mut_dbg<B>(self, func: impl FnOnce(&mut B)) -> Self
fn tap_borrow_mut_dbg<B>(self, func: impl FnOnce(&mut B)) -> Self
Calls
.tap_borrow_mut() only in debug builds, and is erased in release
builds.Source§fn tap_ref_dbg<R>(self, func: impl FnOnce(&R)) -> Self
fn tap_ref_dbg<R>(self, func: impl FnOnce(&R)) -> Self
Calls
.tap_ref() only in debug builds, and is erased in release
builds.Source§fn tap_ref_mut_dbg<R>(self, func: impl FnOnce(&mut R)) -> Self
fn tap_ref_mut_dbg<R>(self, func: impl FnOnce(&mut R)) -> Self
Calls
.tap_ref_mut() only in debug builds, and is erased in release
builds.Source§fn tap_deref_dbg<T>(self, func: impl FnOnce(&T)) -> Self
fn tap_deref_dbg<T>(self, func: impl FnOnce(&T)) -> Self
Calls
.tap_deref() only in debug builds, and is erased in release
builds.