diff --git a/lang_agent/graphs/__init__.py b/lang_agent/graphs/__init__.py
index bd2b832..de9cc71 100644
--- a/lang_agent/graphs/__init__.py
+++ b/lang_agent/graphs/__init__.py
@@ -3,11 +3,15 @@ import tyro
 from lang_agent.graphs.react import ReactGraphConfig
 from lang_agent.graphs.routing import RoutingConfig
 from lang_agent.graphs.dual_path import DualConfig
+from lang_agent.graphs.vision_routing import VisionRoutingConfig
+# from lang_agent.graphs.xiaoai_demo import XiaoAiConfig
 
 graph_dict = {
     "react": ReactGraphConfig(),
     "route": RoutingConfig(),
     "dual": DualConfig(),
+    "vision": VisionRoutingConfig(),
+    # "ai_demo": XiaoAiConfig()
 }
 
 graph_union = tyro.extras.subcommand_type_from_defaults(graph_dict, prefix_names=False)