Struct GraphAgentBuilder 

pub struct GraphAgentBuilder { /* private fields */ }

Builder for GraphAgent

Implementations

impl GraphAgentBuilder

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

Create a new builder

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

Set description

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

Set state schema

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

Add channels to state schema

pub fn node<N: Node + 'static>(self, node: N) -> Self

Add a node

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

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

Add a function as a node

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

Add a direct edge

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

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

Add a conditional edge

pub fn checkpointer<C: Checkpointer + 'static>(self, checkpointer: C) -> Self

Set checkpointer

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

Set checkpointer with Arc

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

Set nodes to interrupt before

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

Set nodes to interrupt after

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

Set recursion limit

pub fn node_timeout(self, node_name: &str, policy: TimeoutPolicy) -> Self

Set a timeout policy for a specific node.

The policy is applied when the named node executes, enforcing wall-clock and/or idle timeouts with the configured recovery action.

Example

use std::time::Duration;
use adk_graph::timeout::{TimeoutPolicy, OnTimeout};

let agent = GraphAgent::builder("my_graph")
    .node_timeout("slow_node", TimeoutPolicy {
        run_timeout: Some(Duration::from_secs(10)),
        idle_timeout: None,
        on_timeout: OnTimeout::Fail,
    })
    .build()?;

pub fn default_timeout(self, policy: TimeoutPolicy) -> Self

Set a default timeout policy applied to all nodes without an explicit override.

Nodes that have a per-node policy set via node_timeout will use their specific policy instead of this default.

Example

use std::time::Duration;
use adk_graph::timeout::{TimeoutPolicy, OnTimeout};

let agent = GraphAgent::builder("my_graph")
    .default_timeout(TimeoutPolicy {
        run_timeout: Some(Duration::from_secs(30)),
        idle_timeout: Some(Duration::from_secs(5)),
        on_timeout: OnTimeout::Skip,
    })
    .build()?;

pub fn deferred_node<F, Fut>( self, name: &str, func: F, config: DeferredNodeConfig, ) -> Self

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

Add a deferred (fan-in barrier) node to the graph.

A deferred node waits for all upstream parallel paths to complete before executing. The provided function is wrapped as a FunctionNode and the DeferredNodeConfig controls how upstream outputs are merged and how long the node waits for all paths.

Arguments

• name - The name of the deferred node.
• func - The async function to execute once all upstream paths complete.
• config - Configuration controlling merge strategy and fan-in timeout.

Example

use std::time::Duration;
use adk_graph::deferred::{DeferredNodeConfig, MergeStrategy};
use adk_graph::node::NodeOutput;

let agent = GraphAgent::builder("scatter_gather")
    .deferred_node("aggregator", |_ctx| async {
        Ok(NodeOutput::new().with_update("status", serde_json::json!("merged")))
    }, DeferredNodeConfig {
        merge_strategy: MergeStrategy::Collect,
        fan_in_timeout: Some(Duration::from_secs(30)),
    })
    .build()?;

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

where F: Fn(&dyn InvocationContext) -> State + Send + Sync + 'static,

Set custom input mapper

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

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

Set custom output mapper

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

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

Set before agent callback

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

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

Set after agent callback

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

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

Build the GraphAgent