Struct AgentNode 

pub struct AgentNode { /* private fields */ }

Available on crate feature graph only.

Wrapper to use an existing ADK Agent as a graph node

Implementations

impl AgentNode

pub fn new(agent: Arc<dyn Agent>) -> AgentNode

Create a new agent node

pub fn with_input_mapper<F>(self, mapper: F) -> AgentNode

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

Set custom input mapper

pub fn with_output_mapper<F>(self, mapper: F) -> AgentNode

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

Set custom output mapper

pub fn with_goto_mapper<F>(self, mapper: F) -> AgentNode

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

Chooses this node’s successors from the updates the output mapper produced.

An agent’s answer often decides where control goes next. The output mapper turns the agent’s events into state; this turns that state into a route, so the classification is parsed once.

Returning None leaves the declared edges in charge. Returning targets replaces them, exactly as NodeOutput::with_goto does for a plain node.

Example

node.with_goto_mapper(|updates: &HashMap<String, serde_json::Value>| {
    match updates.get("category").and_then(|v| v.as_str()) {
        Some("refund") => Some(vec!["refund_desk".to_string()]),
        Some(_) => Some(vec!["general_desk".to_string()]),
        None => None,
    }
})

Trait Implementations

impl Node for AgentNode

fn name(&self) -> &str

Node identifier

fn execute<'life0, 'life1, 'async_trait>( &'life0 self, ctx: &'life1 NodeContext, ) -> Pin<Box<dyn Future<Output = Result<NodeOutput, GraphError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, AgentNode: 'async_trait,

Execute the node and return state updates

fn execute_stream<'a>( &'a self, ctx: &'a NodeContext, ) -> Pin<Box<dyn Stream<Item = Result<StreamEvent, GraphError>> + Send + 'a>>

Streams execution events for this node.

fn validate_against(&self, _parent: &StateSchema) -> Result<(), GraphError>

Rejects a node that cannot execute, before the graph runs.

fn validate(&self) -> Result<(), GraphError>