AgentDojo: A Dynamic Environment to Evaluate Prompt Injection Attacks and Defenses for LLM Agents
Abstract
AI agents aim to solve complex tasks by combining text-based reasoning with external tool calls. Unfortunately, AI agents are vulnerable to prompt injection attacks where data returned by external tools hijacks the agent to execute malicious tasks. To measure the adversarial robustness of AI agents, we introduce AgentDojo, an evaluation framework for agents that execute tools over untrusted data. To capture the evolving nature of attacks and defenses, AgentDojo is not a static test suite, but rather an extensible environment for designing and evaluating new agent tasks, defenses, and adaptive attacks. We populate the environment with 97 realistic tasks (e.g., managing an email client, navigating an e-banking website, or making travel bookings), 629 security test cases, and various attack and defense paradigms from the literature. We find that AgentDojo poses a challenge for both attacks and defenses: state-of-the-art LLMs fail at many tasks (even in the absence of attacks), and existing prompt injection attacks break some security properties but not all. We hope that AgentDojo can foster research on new design principles for AI agents that solve common tasks in a reliable and robust manner. We release the code for AgentDojo at https://github.com/ethz-spylab/agentdojo Show more
Publication status
publishedExternal links
Editor
Book title
Advances in Neural Information Processing Systems 37Pages / Article No.
Publisher
CurranEvent
Subject
Machine Learning; Prompt InjectionOrganisational unit
09764 - Tramèr, Florian / Tramèr, Florian
Related publications and datasets
Is supplemented by: https://agentdojo.spylab.ai/
Is new version of: https://openreview.net/forum?id=m1YYAQjO3w
Is new version of: https://doi.org/10.48550/arXiv.2406.13352
Notes
Poster presentation on December 12, 2024. Datasets and Benchmarks Track.More
Show all metadata
ETH Bibliography
yes
Altmetrics