Struct PluginBuilder 

pub struct PluginBuilder { /* private fields */ }

Available on crate feature plugin only.

Builder for creating plugins with a fluent API.

Example

let plugin = PluginBuilder::new("my-plugin")
    .on_user_message(|ctx, content| Box::pin(async move { Ok(None) }))
    .on_event(|ctx, event| Box::pin(async move { Ok(None) }))
    .build();

Implementations

impl PluginBuilder

pub fn new(name: impl Into<String>) -> PluginBuilder

Create a new plugin builder with the given name.

pub fn on_user_message( self, callback: Box<dyn Fn(Arc<dyn InvocationContext>, Content) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>, ) -> PluginBuilder

Set the on_user_message callback.

pub fn on_event( self, callback: Box<dyn Fn(Arc<dyn InvocationContext>, Event) -> Pin<Box<dyn Future<Output = Result<Option<Event>, AdkError>> + Send>> + Send + Sync>, ) -> PluginBuilder

Set the on_event callback.

pub fn before_run( self, callback: Box<dyn Fn(Arc<dyn InvocationContext>) -> Pin<Box<dyn Future<Output = Result<Option<Content>, AdkError>> + Send>> + Send + Sync>, ) -> PluginBuilder

Set the before_run callback.

pub fn after_run( self, callback: Box<dyn Fn(Arc<dyn InvocationContext>) -> Pin<Box<dyn Future<Output = ()> + Send>> + Send + Sync>, ) -> PluginBuilder

Set the after_run callback.

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

Set the before_agent callback.

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

Set the after_agent callback.

pub fn before_model( self, callback: Box<dyn Fn(Arc<dyn CallbackContext>, LlmRequest) -> Pin<Box<dyn Future<Output = Result<BeforeModelResult, AdkError>> + Send>> + Send + Sync>, ) -> PluginBuilder

Set the before_model callback.

pub fn after_model( self, callback: Box<dyn Fn(Arc<dyn CallbackContext>, LlmResponse) -> Pin<Box<dyn Future<Output = Result<Option<LlmResponse>, AdkError>> + Send>> + Send + Sync>, ) -> PluginBuilder

Set the after_model callback.

pub fn on_model_error( self, callback: Box<dyn Fn(Arc<dyn CallbackContext>, LlmRequest, String) -> Pin<Box<dyn Future<Output = Result<Option<LlmResponse>, AdkError>> + Send>> + Send + Sync>, ) -> PluginBuilder

Set the on_model_error callback.

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

Set the before_tool callback.

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

Set the after_tool callback.

pub fn on_tool_error( self, callback: Box<dyn Fn(Arc<dyn CallbackContext>, Arc<dyn Tool>, Value, String) -> Pin<Box<dyn Future<Output = Result<Option<Value>, AdkError>> + Send>> + Send + Sync>, ) -> PluginBuilder

Set the on_tool_error callback.

pub fn close_fn( self, f: impl Fn() -> Pin<Box<dyn Future<Output = ()> + Send>> + Send + Sync + 'static, ) -> PluginBuilder

Set the close function.

pub fn build(self) -> Plugin

Build the plugin.