agentlab.agents.agent_args

Classes

AgentArgs([agent_name])

Base class for agent arguments for instantiating an agent.
class agentlab.agents.agent_args.AgentArgs(agent_name: str = None)

Bases: AbstractAgentArgs

Base class for agent arguments for instantiating an agent.

Define agent arguments as dataclass variables of this class. For example:

class MyAgentArgs(AgentArgs):

my_arg: str = “default_value” my_other_arg: int = 42

Note: for working properly with AgentXRay, the arguments need to be serializable and hasable.

set_benchmark(benchmark: Benchmark, demo_mode: bool)

Optional method to set benchmark specific flags.

This allows the agent to have minor adjustments based on the benchmark. E.g. using a benchmark specific action space. Or letting the agent see HTML on MiniWoB since AXTree is not enough. Users should avoid making extensive benchmark specific prompt engineering.

Parameters:

  • benchmark – str Name of the benchmark.

  • demo_mode – bool If True, the agent should adapt to demo mode. E.g. it can set the demo_mode flag in the browsergym action space.

set_reproducibility_mode()

Optional method to set the agent in a reproducibility mode.

This should adjust the agent configuration to make it as deterministic as possible e.g. setting the temperature of the model to 0.

This is only called when reproducibility is requested.

Raises:

NotImplementedError – If the agent does not support reproducibility.