better get human msg

2026-01-26 17:40:09 +08:00
parent 4986671f30
commit 3d76179a01
1 changed files with 6 additions and 1 deletions
--- a/lang_agent/graphs/routing.py
+++ b/lang_agent/graphs/routing.py
@@ -130,7 +130,12 @@ class RoutingGraph(GraphBase):
        get user message of current invocation
        """
        msgs = state["inp"][0]["messages"]
-        candidate_hum_msg = msgs[1]
+
+        candidate_hum_msg = None
+        for msg in msgs:
+            if isinstance(msg, HumanMessage):
+                candidate_hum_msg = msg
+                
        assert isinstance(candidate_hum_msg, HumanMessage), "not a human message"

        return candidate_hum_msg