Struct StateGraph 

pub struct StateGraph {
    pub schema: StateSchema,
    pub nodes: HashMap<String, Arc<dyn Node>>,
    pub edges: Vec<Edge>,
}

Available on crate feature graph only.

Builder for constructing graphs

Fields

schema: StateSchema

State schema

nodes: HashMap<String, Arc<dyn Node>>

Registered nodes

edges: Vec<Edge>

Registered edges

Implementations

impl StateGraph

pub fn new(schema: StateSchema) -> StateGraph

Create a new graph with the given state schema

pub fn with_channels(channels: &[&str]) -> StateGraph

Create with a simple schema (just channel names, all overwrite)

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

where N: Node + 'static,

Add a node to the graph

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

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

Add a function as a node

pub fn add_edge(self, source: &str, target: &str) -> StateGraph

Add a direct edge from source to target

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

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

Add a conditional edge with a router function

pub fn add_conditional_edges_arc<I>( self, source: &str, router: Arc<dyn Fn(&HashMap<String, Value>) -> String + Sync + Send>, targets: I, ) -> StateGraph

where I: IntoIterator<Item = (&'static str, &'static str)>,

Add a conditional edge with an Arc router (for pre-built routers)

pub fn compile(self) -> Result<CompiledGraph, GraphError>

Compile the graph for execution