diff --git a/lang_agent/graphs/__init__.py b/lang_agent/graphs/__init__.py index a635539..d1d0a36 100644 --- a/lang_agent/graphs/__init__.py +++ b/lang_agent/graphs/__init__.py @@ -1,9 +1,11 @@ import tyro from lang_agent.graphs.react import ReactGraphConfig +from lang_agent.graphs.routing import RoutingConfig graph_dict = { - "react": ReactGraphConfig() + "react": ReactGraphConfig(), + "route": RoutingConfig() } graph_union = tyro.extras.subcommand_type_from_defaults(graph_dict, prefix_names=False)