parametrized env output to agent_args

Author: TLSDC

src/agentlab/agents/tool_use_agent/agent.py

@@ -65,6 +65,7 @@ class ToolUseAgentArgs(AgentArgs):
     model_args: OpenAIResponseModelArgs = None
     use_first_obs: bool = True
     tag_screenshot: bool = True
+    use_raw_page_output: bool = True
 
     def __post_init__(self):
         try:

src/agentlab/experiments/loop.py

@@ -401,11 +401,15 @@ def run(self):
             agent = self.agent_args.make_agent()
             logger.debug("Agent created.")
 
+            if hasattr(self.agent_args, "use_raw_page_output"):
+                use_raw_page_output = self.agent_args.use_raw_page_output
+            else:
+                use_raw_page_output = False
+
             env = self.env_args.make_env(
                 action_mapping=agent.action_set.to_python_code,
                 exp_dir=self.exp_dir,
-                use_raw_page_output=True,  # TODO:thibault Parametrize this one way or the other.
-                # Could be smth like self.agent_args.use_raw_page_output
+                use_raw_page_output=use_raw_page_output,
             )
 
             logger.debug("Environment created.")