From 262cb916aee8479e554f3b15c198e5f8dc9f5e0f Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Thu, 30 Oct 2025 17:31:52 +0800
Subject: [PATCH] bug fix

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

diff --git a/lang_agent/graphs/react.py b/lang_agent/graphs/react.py
index 51de1d9..d1a978d 100644
--- 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