Back to Product

Agent SDK

Wrap any AI agent with a single decorator for instant protection. Native SDKs for Python, TypeScript, and Go. Works with every major AI framework.

Installation

Get started in seconds

pip install notarylabs
Examplev2.4.1
from notarylabs import Notary Labs, protect

client = Notary Labs(api_key="aw_...")

@protect("pii-mask")
def query_customer(customer_id: str):
    # Your agent logic here
    return db.query(f"SELECT * FROM customers WHERE id = {customer_id}")

# Or use the context manager
with client.protect("read-only"):
    result = db.query("SELECT * FROM transactions")
Features

Production-ready from day one

Our SDKs are built with the same care as the rest of the platform. Fully typed, well-documented, and battle-tested in production.

Zero-config policy enforcement
Automatic PII detection
Built-in retry and circuit breaker
OpenTelemetry tracing
Async/await support
Type-safe APIs
Comprehensive test helpers
Hot policy reload

CLI Tools

Manage policies, view logs, and debug issues right from your terminal.

$ notarylabs policies list $ notarylabs logs tail --agent support-bot $ notarylabs test policy.yaml
Integrations

Works with your AI stack

Native integrations with popular AI frameworks and model providers. Drop-in protection for your existing code.

OpenAI

native

Wrap ChatGPT and GPT-4 calls

Anthropic

native

Claude integration out of the box

LangChain

native

Works with chains and agents

LlamaIndex

native

Protect your RAG pipelines

AWS Bedrock

native

All Bedrock models supported

Azure OpenAI

native

Enterprise OpenAI deployments

Hugging Face

community

Local and hosted models

Cohere

native

Command and embed models

Add security in one line of code

Start protecting your AI agents today with our production-ready SDKs.

Start free trial