20 lines
825 B
Python
20 lines
825 B
Python
import tyro
|
|
|
|
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(),
|
|
# "child_demo": ChildDemoGraphConfig(),
|
|
"deepagent": DeepAgentConfig()
|
|
}
|
|
|
|
graph_union = tyro.extras.subcommand_type_from_defaults(graph_dict, prefix_names=False)
|
|
AnnotatedGraph = tyro.conf.OmitSubcommandPrefixes[tyro.conf.SuppressFixed[graph_union]] |