graph_union for later

2025-10-22 12:30:27 +08:00
parent d2f1f9b809
commit 07ed84c273
1 changed files with 10 additions and 0 deletions
--- a/lang_agent/graphs/init.py
+++ 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]]