alirezaaminzadeh commited on
Commit
0b28085
·
verified ·
1 Parent(s): df999f2

Add secure agent pipeline card

Browse files
Files changed (1) hide show
  1. README.md +61 -0
README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: apache-2.0
4
+ tags:
5
+ - agent
6
+ - tool-calling
7
+ - security
8
+ - mcp
9
+ - guardrails
10
+ pipeline_tag: text-generation
11
+ ---
12
+
13
+ # AgentShield Secure Agent
14
+
15
+ Guardrailed ToolCallingAgent pipeline combining four security classifiers as pre-execution guardrails.
16
+
17
+ ## Architecture
18
+
19
+ ```
20
+ User Input + Untrusted Content
21
+ |
22
+ v
23
+ +---------------------------+
24
+ | Content Trust Classifier | --> untrusted? sanitize / block
25
+ +---------------------------+
26
+ |
27
+ v
28
+ +---------------------------+
29
+ | Prompt Injection Classifier| --> injection? reject / escalate
30
+ +---------------------------+
31
+ |
32
+ v
33
+ +---------------------------+
34
+ | Policy Decision Model | --> allow / deny / escalate
35
+ +---------------------------+
36
+ |
37
+ v
38
+ +---------------------------+
39
+ | Tool Risk Classifier | --> per tool-call risk scoring
40
+ +---------------------------+
41
+ |
42
+ v
43
+ Safe Task Execution
44
+ ```
45
+
46
+ ## Component Models
47
+
48
+ | Component | Model |
49
+ |-----------|-------|
50
+ | Prompt Injection | `alirezaaminzadeh/agentshield-prompt-injection-classifier` |
51
+ | Tool Risk | `alirezaaminzadeh/agentshield-tool-risk-classifier` |
52
+ | Content Trust | `alirezaaminzadeh/agentshield-content-trust-classifier` |
53
+ | Policy Decision | `alirezaaminzadeh/agentshield-policy-decision-model` |
54
+
55
+ ## Demo
56
+
57
+ Interactive evaluation: [AgentShield-Bench Space](https://huggingface.co/spaces/alirezaaminzadeh/agentshield-bench)
58
+
59
+ ## License
60
+
61
+ Apache 2.0