From 4305d26907bb4119e5027718a5a50d27269a313f Mon Sep 17 00:00:00 2001 From: goulustis Date: Thu, 23 Oct 2025 20:40:31 +0800 Subject: [PATCH] consistent out --- lang_agent/graphs/react.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lang_agent/graphs/react.py b/lang_agent/graphs/react.py index 6282325..ff46214 100644 --- a/lang_agent/graphs/react.py +++ b/lang_agent/graphs/react.py @@ -62,7 +62,9 @@ class ReactGraph(GraphBase): else: out = self.agent.invoke(*nargs, **kwargs) + msgs_list = jax.tree.leaves(out) + if as_raw: return out else: - return jax.tree.leaves(out)[-1].content + return msgs_list[-1].content