Struct CustomAgentBuilder 

pub struct CustomAgentBuilder { /* private fields */ }

Available on crate feature agents only.

Builder for constructing a CustomAgent.

Implementations

impl CustomAgentBuilder

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

Create a new builder with the given agent name.

pub fn description(self, description: impl Into<String>) -> CustomAgentBuilder

Set the agent description.

pub fn sub_agent(self, agent: Arc<dyn Agent>) -> CustomAgentBuilder

Add a sub-agent.

pub fn sub_agents(self, agents: Vec<Arc<dyn Agent>>) -> CustomAgentBuilder

Set all sub-agents, replacing any previously added.

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

Add a before-agent callback.

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

Add an after-agent callback.

pub fn handler<F, Fut>(self, handler: F) -> CustomAgentBuilder

where F: Fn(Arc<dyn InvocationContext>) -> Fut + Send + Sync + 'static, Fut: Future<Output = Result<Pin<Box<dyn Stream<Item = Result<Event, AdkError>> + Send>>, AdkError>> + Send + 'static,

Set the async handler function that implements the agent’s logic.

pub fn build(self) -> Result<CustomAgent, AdkError>

Build the CustomAgent, returning an error if no handler was set.