diff --git a/lang_agent/graphs/__init__.py b/lang_agent/graphs/__init__.py index d1d0a36..bd2b832 100644 --- a/lang_agent/graphs/__init__.py +++ b/lang_agent/graphs/__init__.py @@ -2,10 +2,12 @@ import tyro from lang_agent.graphs.react import ReactGraphConfig from lang_agent.graphs.routing import RoutingConfig +from lang_agent.graphs.dual_path import DualConfig graph_dict = { "react": ReactGraphConfig(), - "route": RoutingConfig() + "route": RoutingConfig(), + "dual": DualConfig(), } graph_union = tyro.extras.subcommand_type_from_defaults(graph_dict, prefix_names=False)