Skip to content

ServiceNow/AgentLab

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

AgentLab Banner

pypi PyPI - License PyPI - Downloads GitHub star chart Code Format Tests

πŸ› οΈ Setup Β |Β  πŸ€– Assistant Β |Β  πŸš€ Launch Experiments Β |Β  πŸ” Analyse Results Β |Β  πŸ€– Build Your Agent Β |Β  ↻ Reproducibility

4x4.grid.mp4

AgentLab is a framework for developing and evaluating agents on a variety of benchmarks supported by BrowserGym.

AgentLab Features:

  • Easy large scale parallel agent experiments using ray
  • Building blocks for making agents over BrowserGym
  • Unified LLM API for OpenRouter, OpenAI, Azure, or self hosted using TGI.
  • Prefered way for running benchmarks like WebArena
  • Various reproducibility features
  • Unified LeaderBoard (soon)

🎯 Supported Benchmarks

Benchmark Setup
Link
# Task
Template
Seed
Diversity
Max
Step
Multi-tab Hosted Method BrowserGym
Leaderboard
WebArena setup 812 None 30 yes self hosted (docker) soon
WorkArena L1 setup 33 High 30 no demo instance soon
WorkArena L2 setup 341 High 50 no demo instance soon
WorkArena L3 setup 341 High 50 no demo instance soon
WebLinx - 31586 None 1 no self hosted (dataset) soon
VisualWebArena setup 910 None 30 yes self hosted (docker) soon
AssistantBench setup 214 None 30 yes live web soon
GAIA (soon) - - None - - live web soon
Mind2Web-live (soon) - - None - - live web soon
MiniWoB setup 125 Medium 10 no self hosted (static files) soon

πŸ› οΈ Setup AgentLab

pip install agentlab

If not done already, install playwright:

playwright install

Make sure to prepare the required benchmark according to instructions provided in the setup column.

export AGENTLAB_EXP_ROOT=<root directory of experiment results>  # defaults to $HOME/agentlab_results
export OPENAI_API_KEY=<your openai api key> # if openai models are used
Setup OpenRouter API
export OPENROUTER_API_KEY=<your openrouter api key> # if openrouter models are used
Setup Azure API
export AZURE_OPENAI_API_KEY=<your azure api key> # if using azure models
export AZURE_OPENAI_ENDPOINT=<your endpoint> # if using azure models

πŸ€– UI-Assistant

Use an assistant to work for you (at your own cost and risk).

agentlab-assistant --start_url https://www.google.com

Try your own agent:

agentlab-assistant --agent_config="module.path.to.your.AgentArgs"

πŸš€ Launch experiments

# Import your agent configuration extending bgym.AgentArgs class
# Make sure this object is imported from a module accessible in PYTHONPATH to properly unpickle
from agentlab.agents.generic_agent import AGENT_4o_MINI 

from agentlab.experiments.study import make_study

study = make_study(
    benchmark="miniwob",  # or "webarena", "workarnea_l1" ...
    agent_args=[AGENT_4o_MINI],
    comment="My first study",
)

study.run(n_jobs=5)

Relaunching incomplete or errored tasks

from agentlab.experiments.study import Study
study = Study.load("/path/to/your/study/dir")
study.find_incomplete(include_errors=True)
study.run()

See main.py to launch experiments with a variety of options. This is like a lazy CLI that is actually more convenient. Just comment and uncomment the lines you need or modify at will (but don't push to the repo).

Job Timeouts

The complexity of the wild web, Playwright, and asyncio can sometimes cause jobs to hang. This disables workers until the study is terminated and relaunched. If you are running jobs sequentially or with a small number of workers, this could halt your entire study until you manually kill and relaunch it. In the Ray parallel backend, we've implemented a system to automatically terminate jobs exceeding a specified timeout. This feature is particularly useful when task hanging limits your experiments.

Debugging

For debugging, run experiments with n_jobs=1 and use VSCode's debug mode. This allows you to pause execution at breakpoints.

About Parallel Jobs

Running one agent on one task corresponds to a single job. Conducting ablation studies or random searches across hundreds of tasks with multiple seeds can generate more than 10,000 jobs. Efficient parallel execution is therefore critical. Agents typically wait for responses from the LLM server or updates from the web server. As a result, you can run 10–50 jobs in parallel on a single computer, depending on available RAM.

⚠️ Note for (Visual)WebArena: These benchmarks have task dependencies designed to minimize "corrupting" the instance between tasks. For example, an agent on task 323 could alter the instance state, making task 201 impossible. To address this, the Ray backend accounts for task dependencies, enabling some degree of parallelism. On WebArena, you can disable dependencies to increase parallelism, but this might reduce performance by 1–2%.

⚠️ Instance Reset for (Visual)WebArena: Before evaluating an agent, the instance is automatically reset, a process that takes about 5 minutes. When evaluating multiple agents, the make_study function returns a SequentialStudies object to ensure proper sequential evaluation of each agent. AgentLab currently does not support evaluations across multiple instances, but you could either create a quick script to handle this or submit a PR to AgentLab. For a smoother parallel experience, consider using benchmarks like WorkArena instead.

πŸ” Analyse Results

Loading Results

The class ExpResult provides a lazy loader for all the information of a specific experiment. You can use yield_all_exp_results to recursivley find all results in a directory. Finally load_result_df gathers all the summary information in a single dataframe. See inspect_results.ipynb for example usage.

from agentlab.analyze import inspect_results
result_df = inspect_results.load_result_df("path/to/your/study")

AgentXray

AgentXray.demo.mov

In a terminal, execute:

agentlab-xray

You can load previous or ongoing experiments in the directory AGENTLAB_EXP_ROOT and visualize the results in a gradio interface.

In the following order, select:

  • The experiment you want to visualize
  • The agent if there is more than one
  • The task
  • And the seed

Once this is selected, you can see the trace of your agent on the given task. Click on the profiling image to select a step and observe the action taken by the agent.

⚠️ Note: Gradio is still in developement and unexpected behavior have been frequently noticed. Version 5.5 seems to work properly so far. If you're not sure that the proper information is displaying, refresh the page and select your experiment again.

πŸ€– Implement a new Agent

Get inspiration from the MostBasicAgent in agentlab/agents/most_basic_agent/most_basic_agent.py. For a better integration with the tools, make sure to implement most functions in the AgentArgs API and the extended bgym.AbstractAgentArgs.

If you think your agent should be included directly in AgenLab, let us know and it can be added in agentlab/agents/ with the name of your agent.

↻ Reproducibility

Several factors can influence reproducibility of results in the context of evaluating agents on dynamic benchmarks.

Factors affecting reproducibility

  • Software version: Different version of Playwright or any package in the software stack could influence the behavior of the benchmark or the agent.
  • API based LLMs silently changing: Even for a fixed version, an LLM may be updated e.g. to incorporate latest web knowledge.
  • Live websites:
    • WorkArena: The demo instance is mostly fixed in time to a specific version but ServiceNow sometime push minor modifications.
    • AssistantBench and GAIA: These rely on the agent navigating the open web. The experience may change depending on which country or region, some websites might be in different languages by default.
  • Stochastic Agents: Setting temperature of the LLM to 0 can reduce most stochasticity.
  • Non deterministic tasks: For a fixed seed, the changes should be minimal

Reproducibility Features

  • Study contains a dict of information about reproducibility, including benchmark version, package version and commit hash
  • The Study class allows automatic upload of your results to reproducibility_journal.csv. This makes it easier to populate a large amount of reference points.
  • Reproduced results in the leaderboard. For agents that are repdocudibile, we encourage users to try to reproduce the results and upload them to the leaderboard. There is a special column containing information about all reproduced results of an agent on a benchmark.
  • ReproducibilityAgent: You can run this agent on an existing study and it will try to re-run the same actions on the same task seeds. A vsiual diff of the two prompts will be displayed in the AgentInfo HTML tab of AgentXray. You will be able to inspect on some tasks what kind of changes between to two executions. Note: this is a beta feature and will need some adaptation for your own agent.

Misc

if you want to download HF models more quickly

pip install hf-transfer
pip install torch
export HF_HUB_ENABLE_HF_TRANSFER=1