ToolShield banner
CHATS-lab CHATS-lab

ToolShield

Development community

Description

[ICML 2026] Official implementation for paper "Unsafer in Many Turns: Benchmarking and Defending Multi-Turn Safety Risks in Tool-Using Agents"

Installation

This entry records only its repository, not the path inside it, so there is no exact command to give. Open the source below and copy the folder into ~/.claude/skills/, or the file into ~/.claude/agents/.

README

[ICML 2026] ToolShield: A Package to Guard Your Agent

Accepted to the ICML 2026 main conference  ยท  ๐Ÿ† additionally selected as a Spotlight at the AI4GOOD Workshop @ ICML 2026

[![PyPI](https://img.shields.io/pypi/v/toolshield?style=for-the-badge&logo=pypi&logoColor=white)](https://pypi.org/project/toolshield/) [![Paper](https://img.shields.io/badge/arXiv-2602.13379-b31b1b?style=for-the-badge&logo=arxiv&logoColor=white)](https://arxiv.org/abs/2602.13379) [![Homepage](https://img.shields.io/badge/Homepage-4d8cd8?style=for-the-badge&logo=google-chrome&logoColor=white)](https://unsafer-in-many-turns.github.io) [![License](https://img.shields.io/badge/License-MIT-green?style=for-the-badge)](LICENSE) [![HuggingFace](https://img.shields.io/badge/%F0%9F%A4%97%20Dataset-FFD21E?style=for-the-badge)](https://huggingface.co/datasets/CHATS-Lab/MT-AgentRisk) [![Downloads](https://img.shields.io/pepy/dt/toolshield?style=for-the-badge&logo=pypi&logoColor=white&color=green)](https://pepy.tech/projects/toolshield)

Supports:  Claude Code