diff --git a/lang_agent/graphs/__init__.py b/lang_agent/graphs/__init__.py new file mode 100644 index 0000000..a635539 --- /dev/null +++ b/lang_agent/graphs/__init__.py @@ -0,0 +1,10 @@ +import tyro + +from lang_agent.graphs.react import ReactGraphConfig + +graph_dict = { + "react": ReactGraphConfig() +} + +graph_union = tyro.extras.subcommand_type_from_defaults(graph_dict, prefix_names=False) +AnnotatedGraph = tyro.conf.OmitSubcommandPrefixes[tyro.conf.SuppressFixed[graph_union]] \ No newline at end of file