---
language: en
license: apache-2.0
tags:
- agent
- tool-calling
- security
- mcp
- guardrails
pipeline_tag: text-generation
---

# AgentShield Secure Agent

Guardrailed ToolCallingAgent pipeline combining four security classifiers as pre-execution guardrails.

## Architecture

```
User Input + Untrusted Content
        |
        v
+---------------------------+
| Content Trust Classifier  | --> untrusted? sanitize / block
+---------------------------+
        |
        v
+---------------------------+
| Prompt Injection Classifier| --> injection? reject / escalate
+---------------------------+
        |
        v
+---------------------------+
| Policy Decision Model     | --> allow / deny / escalate
+---------------------------+
        |
        v
+---------------------------+
| Tool Risk Classifier      | --> per tool-call risk scoring
+---------------------------+
        |
        v
   Safe Task Execution
```

## Component Models

| Component | Model |
|-----------|-------|
| Prompt Injection | `alirezaaminzadeh/agentshield-prompt-injection-classifier` |
| Tool Risk | `alirezaaminzadeh/agentshield-tool-risk-classifier` |
| Content Trust | `alirezaaminzadeh/agentshield-content-trust-classifier` |
| Policy Decision | `alirezaaminzadeh/agentshield-policy-decision-model` |

## Demo

Interactive evaluation: [AgentShield-Bench Space](https://huggingface.co/spaces/alirezaaminzadeh/agentshield-bench)

## License

Apache 2.0