bug fix

2025-10-30 17:31:52 +08:00
parent 5b0a486d2c
commit 262cb916ae
1 changed files with 1 additions and 1 deletions
--- a/lang_agent/graphs/react.py
+++ b/lang_agent/graphs/react.py
@@ -65,6 +65,6 @@ class ReactGraph(GraphBase):
        msgs_list = jax.tree.leaves(out)

        if as_raw:
-            return out
+            return msgs_list
        else:
            return msgs_list[-1].content