Skip to main content

Multi-layer agent defense framework for AI security

Project description

agentshield

Multi-layer agent defense framework for AI security

CI PyPI version Python versions License

Part of the AumOS open-source agent infrastructure portfolio.


Features

  • SecurityPipeline scans agent input, output, and tool calls through an ordered chain of scanners with configurable severity thresholds and BLOCK/WARN/LOG actions
  • Eight built-in scanners: regex injection detection, PII detection, credential detection, output safety, tool call validation, behavioral checking, output validation, and tool call integrity checking
  • Scanner ABC makes it straightforward to write and register custom scanners; load them from a shield.yaml config file or inject them at runtime
  • Phase-aware dispatch — each scanner declares which phases (INPUT, OUTPUT, TOOL_CALL) it runs in, so no scanner is invoked unnecessarily
  • Cumulative findings tracked across the session with JSON, Markdown, and HTML report generation
  • Adapters for LangChain, CrewAI, AutoGen, OpenAI Agents, and MCP that wrap existing agent code with a single decorator or context manager
  • OWASP ASI Top 10 category mapper tags each finding with its corresponding ASI category for structured vulnerability reporting

Quick Start

Install from PyPI:

pip install agentshield

Verify the installation:

agentshield version

Basic usage:

import agentshield

# See examples/01_quickstart.py for a working example

Documentation

Enterprise Upgrade

For production deployments requiring SLA-backed support and advanced integrations, contact the maintainers or see the commercial extensions documentation.

Contributing

Contributions are welcome. Please read CONTRIBUTING.md before opening a pull request.

License

Apache 2.0 — see LICENSE for full terms.


Part of AumOS — open-source agent infrastructure.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

aumos_agentshield-0.1.0.tar.gz (132.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

aumos_agentshield-0.1.0-py3-none-any.whl (124.6 kB view details)

Uploaded Python 3

File details

Details for the file aumos_agentshield-0.1.0.tar.gz.

File metadata

  • Download URL: aumos_agentshield-0.1.0.tar.gz
  • Upload date:
  • Size: 132.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.9

File hashes

Hashes for aumos_agentshield-0.1.0.tar.gz
Algorithm Hash digest
SHA256 236d221dcf9ac916332a196039ffcf353a857b4fed643513ff3466c2c77d5a3e
MD5 e35f74b041e2e14f624aa34f9b4e7170
BLAKE2b-256 828c8288cf1bc7538324681d8674b83c95ff50f200fb39ac3b8de398fb3b7791

See more details on using hashes here.

File details

Details for the file aumos_agentshield-0.1.0-py3-none-any.whl.

File metadata

File hashes

Hashes for aumos_agentshield-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 19ecb6aaa1476b5ec9f62bd8c5a38774e4d8a217c233991267f94dd38b8df125
MD5 a761ac8662ea40f69a2d13df401e1001
BLAKE2b-256 40c13818d59a48536bfda6612cb2449ef7d682d24d6770222b247e553b16ebe7

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page