From 441707b67f065bca48f84514a9f9b6bed6cff1f8 Mon Sep 17 00:00:00 2001 From: AAnirudh07 Date: Tue, 7 Jul 2026 10:43:21 +0530 Subject: [PATCH] Allow returning custom fields --- verl/experimental/agent_loop/tool_agent_loop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/verl/experimental/agent_loop/tool_agent_loop.py b/verl/experimental/agent_loop/tool_agent_loop.py index e107bb37b51..c16fa12a63d 100644 --- a/verl/experimental/agent_loop/tool_agent_loop.py +++ b/verl/experimental/agent_loop/tool_agent_loop.py @@ -179,7 +179,7 @@ async def run(self, sampling_params: dict[str, Any], **kwargs) -> AgentLoopOutpu else None, num_turns=agent_data.user_turns + agent_data.assistant_turns + 1, metrics=agent_data.metrics, - extra_fields={}, + extra_fields=dict(agent_data.extra_fields), ) output.extra_fields.update({"turn_scores": agent_data.turn_scores, "tool_rewards": agent_data.tool_rewards}) return output