consistent out

2025-10-23 20:40:31 +08:00
parent da2f1575bd
commit 4305d26907
1 changed files with 3 additions and 1 deletions
--- 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