Skip to main content

RedisContextProvider

Struct RedisContextProvider 

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

Redis-backed ContextProvider: recency-based memory storage/retrieval scoped by application/agent/user/thread id, upgraded to a real FT.SEARCH BM25 full-text index when the connected server supports RediSearch (Redis Stack). See the module docs for the RediSearch-vs-SCAN behavior and the remaining vector-search divergence from Python’s RedisProvider.

use agent_framework_redis::RedisContextProvider;
use agent_framework_core::memory::{ContextProvider, SessionContext};
use agent_framework_core::types::Message;

let provider = RedisContextProvider::new("redis://127.0.0.1:6379")?
    .with_user_id("user-42")
    .with_limit(5);

let request = vec![Message::user("I love hiking in the Cascades")];
provider.after_run(&request, &[], None).await?;

let mut ctx = SessionContext::new(vec![Message::user("Any outdoor hobbies?")]);
provider.before_run(&mut ctx).await?;

Implementations§

Source§

impl RedisContextProvider

Source

pub fn new(redis_url: impl Into<String>) -> Result<Self>

Create a provider for redis_url with no scope configured yet (at least one of application/agent/user/thread id must be set via the builder methods before before_run/after_run are called).

Source

pub fn with_key_prefix(self, key_prefix: impl Into<String>) -> Self

Namespace Redis keys under key_prefix (builder style). Defaults to "context".

Source

pub fn with_application_id(self, application_id: impl Into<String>) -> Self

Scope memories to an application id (builder style).

Source

pub fn with_agent_id(self, agent_id: impl Into<String>) -> Self

Scope memories to an agent id (builder style).

Source

pub fn with_user_id(self, user_id: impl Into<String>) -> Self

Scope memories to a user id (builder style).

Source

pub fn with_thread_id(self, thread_id: impl Into<String>) -> Self

Scope memories to a thread id (builder style).

Source

pub fn with_scope_to_per_operation_thread_id(self, value: bool) -> Self

When true, the thread id used for scoping is captured from the session_id of the first SessionContext seen by ContextProvider::before_run instead of the static thread_id above, and a conflicting session_id on a later call is an error (builder style).

Source

pub fn with_context_prompt(self, context_prompt: impl Into<String>) -> Self

Override the header prepended to injected memories (builder style). Defaults to DEFAULT_CONTEXT_PROMPT.

Source

pub fn with_limit(self, limit: usize) -> Self

Maximum memories returned per before_run() call (builder style). Defaults to DEFAULT_LIMIT.

Source

pub fn with_force_scan_fallback(self, force: bool) -> Self

Force the plain-SCAN retrieval path even when the connected server supports RediSearch (builder style). false by default: RediSearch is detected and used automatically. Useful for tests, for pinning behavior during a Redis Stack rollout, or for working around an operational RediSearch problem without a code change. See the module docs for why entries written under one path are not visible via the other.

Trait Implementations§

Source§

impl ContextProvider for RedisContextProvider

Source§

fn before_run<'life0, 'life1, 'async_trait>( &'life0 self, ctx: &'life1 mut SessionContext, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Called before the model is invoked; mutate ctx to inject instructions, messages, and/or tools. Read ctx.input_messages / ctx.session_id.
Source§

fn after_run<'life0, 'life1, 'life2, 'life3, 'async_trait>( &'life0 self, request_messages: &'life1 [Message], response_messages: &'life2 [Message], _error: Option<&'life3 Error>, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait,

Called after an invocation completes, on BOTH success and failure. On success, error is None and response_messages holds the output. On failure, error is Some and response_messages is empty.
Source§

fn is_history_provider(&self) -> bool

Whether this provider manages conversation history (a HistoryProvider). Agent and WorkflowAgent use this to detect an already-attached history provider among a session’s context_providers and avoid auto-attaching a redundant InMemoryHistoryProvider. Defaults to false; history providers override it to true.

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, 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<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