From 672b6c199e481c11143ddded1394b8ff73e4bfcb Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Thu, 15 Jan 2026 16:02:38 +0800
Subject: [PATCH] register vision routing

---
 lang_agent/graphs/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

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)