pub struct AgentContextManager { /* private fields */ }Expand description
Agent Context Manager
Manages the lifecycle of agent contexts including:
- Creating new contexts with unique IDs
- Inheriting context from parent agents
- Compressing contexts to reduce token usage
- Filtering sensitive data
- Merging multiple contexts
- Persisting and loading contexts
Implementations§
Source§impl AgentContextManager
impl AgentContextManager
Sourcepub fn with_storage_dir(storage_dir: impl Into<PathBuf>) -> Self
pub fn with_storage_dir(storage_dir: impl Into<PathBuf>) -> Self
Create a new context manager with a custom storage directory
Sourcepub fn create_context(
&mut self,
parent: Option<&AgentContext>,
config: Option<ContextInheritanceConfig>,
) -> AgentContext
pub fn create_context( &mut self, parent: Option<&AgentContext>, config: Option<ContextInheritanceConfig>, ) -> AgentContext
Create a new agent context, optionally inheriting from a parent
Sourcepub fn inherit(
&self,
parent: &AgentContext,
config: &ContextInheritanceConfig,
) -> AgentContext
pub fn inherit( &self, parent: &AgentContext, config: &ContextInheritanceConfig, ) -> AgentContext
Inherit context from a parent based on configuration
Sourcepub fn compress(
&self,
context: &mut AgentContext,
target_tokens: usize,
) -> AgentContextResult<CompressionResult>
pub fn compress( &self, context: &mut AgentContext, target_tokens: usize, ) -> AgentContextResult<CompressionResult>
Compress a context to reduce token count
This method reduces the context size by:
- Summarizing older conversation messages
- Removing older file contexts
- Removing older tool results
Sourcepub fn filter(
&self,
context: &AgentContext,
filter: &ContextFilter,
) -> AgentContext
pub fn filter( &self, context: &AgentContext, filter: &ContextFilter, ) -> AgentContext
Filter sensitive data from a context
Sourcepub fn merge(&self, contexts: Vec<&AgentContext>) -> AgentContext
pub fn merge(&self, contexts: Vec<&AgentContext>) -> AgentContext
Merge multiple contexts into one
Sourcepub fn get_context(&self, context_id: &str) -> Option<&AgentContext>
pub fn get_context(&self, context_id: &str) -> Option<&AgentContext>
Get a context by ID
Sourcepub fn get_context_mut(&mut self, context_id: &str) -> Option<&mut AgentContext>
pub fn get_context_mut(&mut self, context_id: &str) -> Option<&mut AgentContext>
Get a mutable context by ID
Sourcepub fn update_context(
&mut self,
context_id: &str,
updates: ContextUpdate,
) -> AgentContextResult<()>
pub fn update_context( &mut self, context_id: &str, updates: ContextUpdate, ) -> AgentContextResult<()>
Update a context with the given updates
Sourcepub fn delete_context(&mut self, context_id: &str) -> bool
pub fn delete_context(&mut self, context_id: &str) -> bool
Delete a context by ID
Sourcepub async fn persist_context(
&self,
context: &AgentContext,
) -> AgentContextResult<()>
pub async fn persist_context( &self, context: &AgentContext, ) -> AgentContextResult<()>
Persist a context to disk
Sourcepub async fn load_context(
&mut self,
context_id: &str,
) -> AgentContextResult<Option<AgentContext>>
pub async fn load_context( &mut self, context_id: &str, ) -> AgentContextResult<Option<AgentContext>>
Load a context from disk
Sourcepub fn estimate_token_count(&self, context: &AgentContext) -> usize
pub fn estimate_token_count(&self, context: &AgentContext) -> usize
Estimate token count for a context
This is a rough estimate based on character count. For more accurate counts, use a proper tokenizer.
Sourcepub fn update_token_count(&self, context: &mut AgentContext)
pub fn update_token_count(&self, context: &mut AgentContext)
Update the token count in context metadata
Sourcepub fn list_context_ids(&self) -> Vec<String>
pub fn list_context_ids(&self) -> Vec<String>
List all context IDs in memory
Sourcepub fn storage_dir(&self) -> &PathBuf
pub fn storage_dir(&self) -> &PathBuf
Get the storage directory path
Trait Implementations§
Source§impl Debug for AgentContextManager
impl Debug for AgentContextManager
Auto Trait Implementations§
impl Freeze for AgentContextManager
impl RefUnwindSafe for AgentContextManager
impl Send for AgentContextManager
impl Sync for AgentContextManager
impl Unpin for AgentContextManager
impl UnsafeUnpin for AgentContextManager
impl UnwindSafe for AgentContextManager
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
Source§impl<T> FutureExt for T
impl<T> FutureExt for T
Source§fn with_context(self, otel_cx: Context) -> WithContext<Self>
fn with_context(self, otel_cx: Context) -> WithContext<Self>
Source§fn with_current_context(self) -> WithContext<Self>
fn with_current_context(self) -> WithContext<Self>
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>
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>
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>
T in a tonic::RequestSource§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,
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,
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,
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
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
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
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
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
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
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
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
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
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
.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
.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
.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
.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
.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
.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
.tap_deref() only in debug builds, and is erased in release
builds.