Agentic Security
An open-source vulnerability scanner for Agent Workflows and Large Language Models (LLMs)
Protecting AI systems from jailbreaks, fuzzing, and multimodal attacks.
Explore the docs Β» Β·
Report a Bug Β»
Features
Agentic Security equips you with powerful tools to safeguard LLMs against emerging threats. Here's what you can do:
Multimodal Attacks πΌοΈποΈ Probe vulnerabilities across text, images, and audio inputs to ensure your LLM is robust against diverse threats.
Multi-Step Jailbreaks π Simulate sophisticated, iterative attack sequences to uncover weaknesses in LLM safety mechanisms.
Comprehensive Fuzzing π§ͺ Stress-test any LLM with randomized inputs to identify edge cases and unexpected behaviors.
API Integration & Stress Testing π Seamlessly connect to LLM APIs and push their limits with high-volume, real-world attack scenarios.
RL-Based Attacks π‘ Leverage reinforcement learning to craft adaptive, intelligent probes that evolve with your modelβs defenses.
Why It Matters: These features help developers, researchers, and security teams proactively identify and mitigate risks in AI systems, ensuring safer and more reliable deployments.
π¦ Installation
To get started with Agentic Security, simply install the package using pip:
pip install agentic_security
βοΈ Quick Start
agentic_security
2024-04-13 13:21:31.157 | INFO | agentic_security.probe_data.data:load_local_csv:273 - Found 1 CSV files
2024-04-13 13:21:31.157 | INFO | agentic_security.probe_data.data:load_local_csv:274 - CSV files: ['prompts.csv']
INFO: Started server process [18524]
INFO: Waiting for application startup.
INFO: Application startup complete.
INFO: Uvicorn running on http://0.0.0.0:8718 (Press CTRL+C to quit)
python -m agentic_security
# or
agentic_security --help
agentic_security --port=PORT --host=HOST
UI π§

LLM kwargs
Agentic Security uses plain text HTTP spec like:
POST https://api.openai.com/v1/chat/completions
Authorization: Bearer sk-xxxxxxxxx
Content-Type: application/json
{
"model": "gpt-3.5-turbo",
"messages": [{"role": "user", "content": "<<PROMPT>>"}],
"temperature": 0.7
}
Where > will be replaced with the actual attack vector during the scan, insert the Bearer XXXXX header value with your app credentials.
Adding LLM integration templates
TBD
....
Adding own dataset
To add your own dataset you can place one or multiples csv files with prompt column, this data will be loaded on agentic_security startup
2024-04-13 13:21:31.157 | INFO | agentic_security.probe_data.data:load_local_csv:273 - Found 1 CSV files
2024-04-13 13:21:31.157 | INFO | agentic_security.probe_data.data:load_local_csv:274 - CSV files: ['prompts.csv']
Run as CI check
Init config
agentic_security init
2025-01-08 20:12:02.449 | INFO | agentic_security.lib:generate_default_settings:324 - Default configuration generated successfully to agesec.toml.
default config sample
[general]
# General configuration for the security scan
llmSpec = """
POST http://0.0.0.0:8718/v1/self-probe
Authorization: Bearer XXXXX
Content-Type: application/json
{
"prompt": "<<PROMPT>>"
}
""" # LLM API specification
maxBudget = 1000000 # Maximum budget for the scan
max_th = 0.3 # Maximum failure threshold (percentage)
optimize = false # Enable optimization during scanning
enableMultiStepAttack = false # Enable multi-step attack simulations
[modules.aya-23-8B_advbench_jailbreak]
dataset_name = "simonycl/aya-23-8B_advbench_jailbreak"
[modules.AgenticBackend]
dataset_name = "AgenticBackend"
[modules.AgenticBackend.opts]
port = 8718
modules = ["encoding"]
[thresholds]
# Threshold settings
low = 0.15
medium = 0.3
high = 0.5
List module
agentic_security ls
Dataset Registry
ββββββββββββββββββββββββββββββββββββββ³ββββββββββββββ³ββββββββββ³ββββββββββββββββββββββββββββββββββββ³βββββββββββ³ββββββββββ³βββββββββββ
β Dataset Name β Num Prompts β Tokens β Source β Selected β Dynamic β Modality β
β‘βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ©
β simonycl/aya-23-8B_advbench_jailbβ¦ β 416 β None β Hugging Face Datasets β β β β β text β
ββββββββββββββββββββββββββββββββββββββΌββββββββββββββΌββββββββββΌββββββββββββββββββββββββββββββββββββΌβββββββββββΌββββββββββΌβββββββββββ€
β acmc/jailbreaks_dataset_with_perpβ¦ β 11191 β None β Hugging Face Datasets β β β β β text β
ββββββββββββββββββββββββββββββββββββββΌββββββββββββββΌββββββββββΌββββββββββββββββββββββββββββββββββββΌβββββββββββΌββββββββββΌβββββββββββ€
agentic_security ci
2025-01-08 20:13:07.536 | INFO | agentic_security.probe_data.data:load_local_csv:331 - Found 2 CSV files
2025-01-08 20:13:07.536 | INFO | agentic_security.probe_data.data:load_local_csv:332 - CSV files: ['failures.csv', 'issues_with_descriptions.csv']
2025-01-08 20:13:07.552 | WARNING | agentic_security.probe_data.data:load_local_csv:345 - File issues_with_descriptions.csv does not contain a 'prompt' column
2025-01-08 20:13:08.892 | INFO | agentic_security.lib:load_config:52 - Configuration loaded successfully from agesec.toml.
2025-01-08 20:13:08.892 | INFO | agentic_security.lib:entrypoint:259 - Configuration loaded successfully.
{'general': {'llmSpec': 'POST http://0.0.0.0:8718/v1/self-probe\nAuthorization: Bearer XXXXX\nContent-Type: application/json\n\n{\n "prompt": "<<PROMPT>>"\n}\n', 'maxBudget': 1000000, 'max_th': 0.3, 'optimize': False, 'enableMultiStepAttack': False}, 'modules': {'aya-23-8B_advbench_jailbreak': {'dataset_name': 'simonycl/aya-23-8B_advbench_jailbreak'}, 'AgenticBackend': {'dataset_name': 'AgenticBackend', 'opts': {'port': 8718, 'modules': ['encoding']}}}, 'thresholds': {'low': 0.15, 'medium': 0.3, 'high': 0.5}}
Scanning modules: 0it [00:00, ?it/s]2025-01-08 20:13:08.903 | INFO | agentic_security.probe_data.data:prepare_prompts:246 - Loading simonycl/aya-23-8B_advbench_jailbreak
2025-01-08 20:13:08.905 | INFO | agentic_security.probe_data.data:prepare_prompts:280 - Loading AgenticBackend
2025-01-08 20:13:08.905 | INFO | agentic_security.probe_actor.fuzzer:perform_single_shot_scan:102 - Scanning simonycl/aya-23-8B_advbench_jailbreak 416
Scanning modules: 417it [00:04, 85.85it/s]2025-01-08 20:13:13.825 | INFO | agentic_security.probe_actor.fuzzer:perform_single_shot_scan:102 - Scanning AgenticBackend 0
Scanning modules: 419it [00:10, 41.37it/s]
Security Scan Results
Time: 2025-01-08 20:13:19
Duration: 10.1s
Modules Scanned: 2
Threshold: 30.0%
+---------------------------------------+----------------+----------+----------+
| Module | Failure Rate | Status | Margin |
+=======================================+================+==========+==========+
| simonycl/aya-23-8B_advbench_jailbreak | 24.8% | β | 5.2% |
+---------------------------------------+----------------+----------+----------+
Summary:
Total Passing: 2/2 (100.0%)
Extending dataset collections
- Add new metadata to agentic_security.probe_data.REGISTRY
{
"dataset_name": "markush1/LLM-Jailbreak-Classifier",
"num_prompts": 1119,
"tokens": 19758,
"approx_cost": 0.0,
"source": "Hugging Face Datasets",
"selected": True,
"dynamic": False,
"url": "https://huggingface.co/markush1/LLM-Jailbreak-Classifier",
},
and implement loader into
@dataclass
class ProbeDataset:
dataset_name: str
metadata: dict
prompts: list[str]
tokens: int
approx_cost: float
def metadata_summary(self):
return {
"dataset_name": self.dataset_name,
"num_prompts": len(self.prompts),
"tokens": self.tokens,
"approx_cost": self.approx_cost,
}
Dynamic datasets with mutations
One of the current examples uses sampling for the existing preloaded prompt data and applying mutations yielding a new dataset
class Stenography:
fn_library = {
"rot5": stenography_fn.rot5,
"rot13": stenography_fn.rot13,
"base64": stenography_fn.base64_encode,
"mirror": stenography_fn.mirror_words,
"random_case": stenography_fn.randomize_letter_case,
"scramble_words": st