graphai.graph

​

NodeProtocol Objects

class NodeProtocol(Protocol)

Protocol defining the interface of a decorated node.

​

Graph Objects

class Graph()

​

get_state

def get_state() -> Dict[str, Any]

Get the current graph state.

​

set_state

def set_state(state: Dict[str, Any])

Set the graph state.

​

update_state

def update_state(values: Dict[str, Any])

Update the graph state with new values.

​

reset_state

def reset_state()

Reset the graph state to an empty dict.

​

add_edge

def add_edge(source: NodeProtocol | str, destination: NodeProtocol | str)

Adds an edge between two nodes that already exist in the graph.

Arguments:

• source - The source node or its name.
• destination - The destination node or its name.

​

get_callback

def get_callback()

Get a new instance of the callback class.

Returns:

Callback: A new instance of the callback class.

​

set_callback

def set_callback(callback_class: Type[Callback])

Set the callback class that is returned by the get_callback method and used

as the default callback when no callback is passed to the execute method.

Arguments:

• callback_class (Type[Callback]): The callback class to use as the default callback.