Environment Steering: Using Data Flow Control to Improve Agent Utility and Safety

Charlie Summers, Prajwal Raghunath, Aaditya Pai, Mayur Kulkarni, Zhuo Zhang, Oliver Kennedy, Eugene Wu


Abstract
LLM agents can make unsafe tool calls even when instructed to behave safely. Existing defenses constrain agents before execution, modify tool inputs/outputs, or rely on LLM judges; these approaches may depend on model behavior or block unsafe actions without helping the agent recover. We argue that the execution environment should instead enforce safety as the agent runs and steer it toward safe alternatives when violations occur—we call this Environment Steering. We implement this by modeling the agent and harness execution state as database tables, track the record-level data flows, and check these data flows against declarative policies during runtime. When violations are detected, policy- and context-specific feedback steers the agent toward safe trajectories. On AgentDyn, this enables the agent to improve task success rate over no-defense while achieving 0% attack success rate.
Anthology ID:
2026.realm-1.59
Volume:
Proceedings of the Second Workshop for REsearch on Agent Language Models (REALM 2026)
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Ehsan Kamalloo, Nicolas Gontier, Nikolai Rozanov, Maryam Hashemzadeh, Akhil Arora
Venues:
REALM | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
856–880
Language:
URL:
https://aclanthology.org/2026.realm-1.59/
DOI:
Bibkey:
Cite (ACL):
Charlie Summers, Prajwal Raghunath, Aaditya Pai, Mayur Kulkarni, Zhuo Zhang, Oliver Kennedy, and Eugene Wu. 2026. Environment Steering: Using Data Flow Control to Improve Agent Utility and Safety. In Proceedings of the Second Workshop for REsearch on Agent Language Models (REALM 2026), pages 856–880, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Environment Steering: Using Data Flow Control to Improve Agent Utility and Safety (Summers et al., REALM 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.realm-1.59.pdf