diff --git a/lang_agent/graphs/__init__.py b/lang_agent/graphs/__init__.py index e028546..ced9e04 100644 --- a/lang_agent/graphs/__init__.py +++ b/lang_agent/graphs/__init__.py @@ -4,12 +4,16 @@ from lang_agent.graphs.react import ReactGraphConfig, ReactGraph from lang_agent.graphs.routing import RoutingConfig, RoutingGraph from lang_agent.graphs.dual_path import DualConfig, Dual from lang_agent.graphs.vision_routing import VisionRoutingConfig, VisionRoutingGraph +# from lang_agent.graphs.child_demo import ChildDemoGraphConfig, ChildDemoGraph +from lang_agent.graphs.qt_deepagents import DeepAgentConfig graph_dict = { "react": ReactGraphConfig(), "route": RoutingConfig(), "dual": DualConfig(), - "vision": VisionRoutingConfig() + "vision": VisionRoutingConfig(), + # "child_demo": ChildDemoGraphConfig(), + "deepagent": DeepAgentConfig() } graph_union = tyro.extras.subcommand_type_from_defaults(graph_dict, prefix_names=False)