Struct Callbacks 

pub struct Callbacks {
    pub before_model: Vec<Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>>,
    pub after_model: Vec<Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>>,
    pub before_tool: Vec<Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>>,
    pub after_tool: Vec<Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>>,
}

Available on crate feature runner only.

Collection of all callback types for intercepting model and tool execution.

Fields

before_model: Vec<Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>>

Callbacks invoked before each model call.

after_model: Vec<Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>>

Callbacks invoked after each model call.

before_tool: Vec<Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>>

Callbacks invoked before each tool execution.

after_tool: Vec<Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>>

Callbacks invoked after each tool execution.

Implementations

impl Callbacks

pub fn new() -> Callbacks

Create an empty callback collection.

pub fn add_before_model( &mut self, callback: Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>, )

Register a callback to run before each model call.

pub fn add_after_model( &mut self, callback: Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>, )

Register a callback to run after each model call.

pub fn add_before_tool( &mut self, callback: Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>, )

Register a callback to run before each tool execution.

pub fn add_after_tool( &mut self, callback: Box<dyn Fn(Arc<dyn CallbackContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>, )

Register a callback to run after each tool execution.

pub async fn execute_before_model( &self, ctx: Arc<dyn CallbackContext>, ) -> Result<Vec<Content>, AdkError>

Execute all before_model callbacks

pub async fn execute_after_model( &self, ctx: Arc<dyn CallbackContext>, ) -> Result<Vec<Content>, AdkError>

Execute all after_model callbacks

pub async fn execute_before_tool( &self, ctx: Arc<dyn CallbackContext>, ) -> Result<Vec<Content>, AdkError>

Execute all before_tool callbacks

pub async fn execute_after_tool( &self, ctx: Arc<dyn CallbackContext>, ) -> Result<Vec<Content>, AdkError>

Execute all after_tool callbacks

Trait Implementations

impl Default for Callbacks

fn default() -> Callbacks

Returns the “default value” for a type.