From fde8c854880e8e7b43fe4bfcce297c39aad9f359 Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Fri, 30 Jan 2026 10:38:25 +0800
Subject: [PATCH] bug fix

---
 lang_agent/graphs/routing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py
index dd6470a..f70d865 100644
--- a/lang_agent/graphs/routing.py
+++ b/lang_agent/graphs/routing.py
@@ -166,7 +166,7 @@ class RoutingGraph(GraphBase):
 
 
         out = self.chat_model.invoke(*inp)
-        return {"messages": out}
+        return {"messages": out["messages"]}
 
 
     def _tool_model_call(self, state:State):