Struct GraphAgentBuilder 

pub struct GraphAgentBuilder { /* private fields */ }

Available on crate feature graph only.

Builder for GraphAgent

Implementations

impl GraphAgentBuilder

pub fn new(name: &str) -> GraphAgentBuilder

Create a new builder

pub fn description(self, desc: &str) -> GraphAgentBuilder

Set description

pub fn state_schema(self, schema: StateSchema) -> GraphAgentBuilder

Set state schema

pub fn channels(self, channels: &[&str]) -> GraphAgentBuilder

Add channels to state schema

pub fn node<N>(self, node: N) -> GraphAgentBuilder

where N: Node + 'static,

Add a node

pub fn node_fn<F, Fut>(self, name: &str, func: F) -> GraphAgentBuilder

where F: Fn(NodeContext) -> Fut + Send + Sync + 'static, Fut: Future<Output = Result<NodeOutput, GraphError>> + Send + 'static,

Add a function as a node

pub fn edge(self, source: &str, target: &str) -> GraphAgentBuilder

Add a direct edge

pub fn conditional_edge<F, I>( self, source: &str, router: F, targets: I, ) -> GraphAgentBuilder

where F: Fn(&HashMap<String, Value>) -> String + Send + Sync + 'static, I: IntoIterator<Item = (&'static str, &'static str)>,

Add a conditional edge

pub fn checkpointer<C>(self, checkpointer: C) -> GraphAgentBuilder

where C: Checkpointer + 'static,

Set checkpointer

pub fn checkpointer_arc( self, checkpointer: Arc<dyn Checkpointer>, ) -> GraphAgentBuilder

Set checkpointer with Arc

pub fn interrupt_before(self, nodes: &[&str]) -> GraphAgentBuilder

Set nodes to interrupt before

pub fn interrupt_after(self, nodes: &[&str]) -> GraphAgentBuilder

Set nodes to interrupt after

pub fn recursion_limit(self, limit: usize) -> GraphAgentBuilder

Set recursion limit

pub fn input_mapper<F>(self, mapper: F) -> GraphAgentBuilder

where F: Fn(&dyn InvocationContext) -> HashMap<String, Value> + Send + Sync + 'static,

Set custom input mapper

pub fn output_mapper<F>(self, mapper: F) -> GraphAgentBuilder

where F: Fn(&HashMap<String, Value>) -> Vec<Event> + Send + Sync + 'static,

Set custom output mapper

pub fn before_agent_callback<F, Fut>(self, callback: F) -> GraphAgentBuilder

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

Set before agent callback

pub fn after_agent_callback<F, Fut>(self, callback: F) -> GraphAgentBuilder

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

Set after agent callback

Note: The callback receives a cloned Event to avoid lifetime issues.

pub fn build(self) -> Result<GraphAgent, GraphError>

Build the GraphAgent