Agent guards for OpenAI Agents Python

This example gives an AI agent two tools. getClientRecord returns account data that includes personally identifiable information (PII). Arcjet guards sendEmail, so the model can read the record but cannot send its sensitive fields outside the application. Arcjet checks the model-selected recipient and generated message body before the email provider runs.

What is Arcjet?

Arcjet is the runtime security platform that ships with your code. Enforce budgets, stop prompt injection, detect bots, and protect personal information with Arcjet's AI security building blocks.

Quick start

This guide shows you how to guard an agent tool in your project.

1. Configure the policy

In the Arcjet Console, create an Agent Guard policy with the label email.sent. Add these inputs:

Input	Exposure	Type
recipient	SERVER	String
allowed_recipients	SERVER	String list
body	LOCAL	String

Then add:

1. A string-list membership rule requiring recipient to be a member of allowed_recipients.
2. A sensitive information rule on body that denies BANK_ACCOUNT and ROUTING_NUMBER.

The examples configure the on-device Rampart backend, which detects these entity types locally.

Publish the policy before running the example.

Framework wrappers take action for the same slug you configured as the policy label. Direct guard() calls still use the field name label.

2. Install Arcjet

In your project root, install the SDK:

Install the Guard extra, Rampart, and the OpenAI Agents SDK:

pip install "arcjet[openai-agents,sensitive-info-rampart]"

Or with uv:

uv add "arcjet[openai-agents,sensitive-info-rampart]"

3. Set your key

Create a free Arcjet account and follow the instructions to add a site and get a key.

Set your environment variables:

.env

# Export your Arcjet API key from https://console.arcjet.com
ARCJET_KEY=ajkey_yourkey
ARCJET_ENV=development
OPENAI_API_KEY=sk-yourkey

4. Wrap the tool

Wrap the send-email tool so Arcjet evaluates the remote policy before the email provider runs.

This adapter accepts inputs and actor. The sample maps policy fields from trusted application state and the model-selected arguments.

Create one client and wrap send_email with guard_tool:

agent.py

import os

from agents import Agent, Runner, function_tool
from arcjet.guard import launch_arcjet, local_input, server_input
from arcjet.guard.openai_agents import guard_tool
from arcjet_sensitive_info_rampart import rampart

# Create one Arcjet client and enable local bank-detail detection.
arcjet = launch_arcjet(
    key=os.environ["ARCJET_KEY"],
    sensitive_info_backend=rampart(),
)


@function_tool
def get_client_record() -> dict:
    """Get the account details on file for the current customer."""
    return current_user.record


@function_tool
def send_email(recipient: str, body: str) -> str:
    """Send an email."""
    email_provider.send(to=recipient, body=body)
    return "sent"


# guard_tool attaches tool_input_guardrails plus reject_content.
guarded_send_email = guard_tool(
    guard=arcjet,
    tool=send_email,
    action="email.sent",
    actor=lambda arguments: current_user.id,
    inputs=lambda arguments: {
        "recipient": server_input.string(arguments["recipient"]),
        "allowed_recipients": server_input.string_list(
            current_user.allowed_recipients
        ),
        "body": local_input.string(arguments["body"]),
    },
)

agent = Agent(
    name="support-agent",
    instructions=(
        "Use get_client_record when the user asks for account "
        "details. Use send_email exactly once to complete the "
        "request."
    ),
    tools=[get_client_record, guarded_send_email],
)


async def run_email_agent(user, prompt: str):
    global current_user
    current_user = user
    return await Runner.run(agent, prompt)

5. Try the policy

Keep identity, allowed recipients, and sensitive records on the server. The browser sends only the scenario name.

Expose a small server endpoint. The browser sends only the scenario name:

app.py

from typing import Literal

from fastapi import FastAPI
from pydantic import BaseModel

from agent import run_email_agent

app = FastAPI()


class User:
    def __init__(self):
        self.id = "customer-123"
        self.allowed_recipients = ["approved@example.com"]
        self.record = {
            "name": "Alex Morgan",
            "bank_account": "0123456789",
            "routing_number": "022000020",
        }


user = User()
scenarios = {
    "allowed": (
        "Send the message 'Your report is ready' to "
        "approved@example.com."
    ),
    "blocked": (
        "Send the message 'Your report is ready' to "
        "outside@example.net."
    ),
    "pii": (
        "Email the account details you have on file to "
        "approved@example.com."
    ),
}


class AgentRequest(BaseModel):
    scenario: Literal["allowed", "blocked", "pii"]


@app.post("/api/agent")
async def run_agent(request: AgentRequest):
    result = await run_email_agent(user, scenarios[request.scenario])
    return {"output": str(result.final_output)}

Add a minimal page that calls the endpoint and displays the agent’s response:

public/index.html

<h1>Agent Guard policy demo</h1>
<p>Test recipient and sensitive-information policies on the same email tool.</p>

<button data-scenario="allowed">Allowed recipient</button>
<button data-scenario="blocked">Blocked recipient</button>
<button data-scenario="pii">Sensitive information</button>
<pre id="output">Choose a scenario.</pre>

<script>
  // Send only the selected scenario name to the server.
  const output = document.querySelector("#output");

  for (const button of document.querySelectorAll("button")) {
    button.addEventListener("click", async () => {
      output.textContent = "Running agent…";
      const response = await fetch("/api/agent", {
        method: "POST",
        headers: { "content-type": "application/json" },
        body: JSON.stringify({ scenario: button.dataset.scenario }),
      });
      const result = await response.json();
      output.textContent = result.output ?? result.error;
    });
  }
</script>

Each scenario demonstrates a different result from the same guarded tool:

• Allowed recipient: The recipient is on the allow list and the body has no sensitive data, so sendEmail can reach the email provider.
• Blocked recipient: The model calls the same tool with an external address. When the adapter maps inputs, the remote membership policy denies the call before the provider runs, and the model receives the denial result.
• Sensitive information: The recipient is allowed, but the agent first calls getClientRecord and receives test bank account and routing numbers. When the model includes that tool result in the email body, local sensitive information policy blocks sendEmail before the data leaves the application.

What next?

Framework integrations Helpers, denial envelopes, and fail-closed defaults for every adapter.

Remote policies Labels, actors, typed inputs, and supported policy rules.

Capture events Record that an allowed action happened, without changing the decision.

Testing and reference Test denial and availability behavior and review the API.

OpenAI Agents Python agent guard OpenAI Agents Python integration guide.

Get help

Need help with anything? Email support@arcjet.com to get support from our engineering team.

Discussion