Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

consistent out

Browse Source
This commit is contained in:
goulustis
2025-10-23 20:40:31 +08:00
parent da2f1575bd
commit 4305d26907
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lang_agent/graphs/react.py
View File

@@ -62,7 +62,9 @@ class ReactGraph(GraphBase):
else: else:
out = self.agent.invoke(*nargs, **kwargs) out = self.agent.invoke(*nargs, **kwargs)
msgs_list = jax.tree.leaves(out)
if as_raw: if as_raw:
return out return out
else: else:
return jax.tree.leaves(out)[-1].content return msgs_list[-1].content