Add test import check for all build target under /rl, /utils, /tests folder.

tests/rl/experimental/agentic/QueueManager/group_queue_manager_test.py → tests/rl/experimental/agentic/queue_manager/group_queue_manager_test.py

@@ -17,7 +17,7 @@
 import asyncio
 
 from absl.testing import absltest
-from tunix.rl.experimental.agentic.QueueManager import group_queue_manager
+from tunix.rl.experimental.agentic.queue_manager import group_queue_manager
 
 
 def _create_item(

tunix/rl/experimental/agentic/agents/tool_agent.py

@@ -238,10 +238,11 @@ def update_from_model(self, response: str, **kwargs) -> Action:
     # Record complete step with conversation context and parsed action
     step = Step(
         chat_completions=copy.deepcopy(self._messages),
-        model_response=response,
-        action=tool_calls_dict,
+        action=Action(action=tool_calls_dict),
         observation=self._obs_cache,
+        model_response=response,
     )
+
     self._trajectory.steps.append(step)
 
     return Action(action=tool_calls_dict)

tunix/rl/experimental/agentic/Pipeline/rollout_orchestrator.py → tunix/rl/experimental/agentic/pipeline/rollout_orchestrator.py

@@ -26,7 +26,7 @@
 import logging
 from typing import Any, Callable, Dict, List, Optional, Tuple, Type
 
-from tunix.rl.experimental.agentic.QueueManager import group_queue_manager
+from tunix.rl.experimental.agentic.queue_manager import group_queue_manager
 from tunix.rl.experimental.agentic.trajectory import trajectory_collect_engine