Wallarm MCP Server for Pydantic AI 10 tools — connect in under 2 minutes
Pydantic AI brings type-safe agent development to Python with first-class MCP support. Connect Wallarm through the Vinkius and every tool is automatically validated against Pydantic schemas — catch errors at build time, not in production.
ASK AI ABOUT THIS MCP SERVER
Vinkius supports streamable HTTP and SSE.
import asyncio
from pydantic_ai import Agent
from pydantic_ai.mcp import MCPServerHTTP
async def main():
# Your Vinkius token — get it at cloud.vinkius.com
server = MCPServerHTTP(url="https://edge.vinkius.com/[YOUR_TOKEN_HERE]/mcp")
agent = Agent(
model="openai:gpt-4o",
mcp_servers=[server],
system_prompt=(
"You are an assistant with access to Wallarm "
"(10 tools)."
),
)
result = await agent.run(
"What tools are available in Wallarm?"
)
print(result.data)
asyncio.run(main())
* Every MCP server runs on Vinkius-managed infrastructure inside AWS - a purpose-built runtime with per-request V8 isolates, Ed25519 signed audit chains, and sub-40ms cold starts optimized for native MCP execution. See our infrastructure
About Wallarm MCP Server
Connect your Wallarm account to any AI agent and secure your API infrastructure through natural conversation.
Pydantic AI validates every Wallarm tool response against typed schemas, catching data inconsistencies at build time. Connect 10 tools through the Vinkius and switch between OpenAI, Anthropic, or Gemini without changing your integration code — full type safety, structured output guarantees, and dependency injection for testable agents.
What you can do
- Security Attacks — Monitor and search for active security attacks detected by Wallarm, grouped by vector (SQLi, XSS, etc.)
- Granular Hits — Perform deep forensics by searching for individual malicious HTTP request hits with full payloads
- Vulnerability Management — List and triage security vulnerabilities discovered in live API traffic directly from your agent
- API Inventory — Retrieve the automatically discovered API inventory to see all exposed endpoints and methods
- Filtering Nodes — Verify the health and heartbeat status of your deployed WAF and API gateway filtering nodes
- IP Control — Audit and manage IP allowlist/denylist rules to immediately block malicious sources or allow trusted partners
- Remediation Guidance — Access comprehensive diagnostic data and CWE mappings for specific vulnerabilities
The Wallarm MCP Server exposes 10 tools through the Vinkius. Connect it to Pydantic AI in under two minutes — no API keys to rotate, no infrastructure to provision, no vendor lock-in. Your configuration, your data, your control.
How to Connect Wallarm to Pydantic AI via MCP
Follow these steps to integrate the Wallarm MCP Server with Pydantic AI.
Install Pydantic AI
Run pip install pydantic-ai
Replace the token
Replace [YOUR_TOKEN_HERE] with your Vinkius token
Run the agent
Save to agent.py and run: python agent.py
Explore tools
The agent discovers 10 tools from Wallarm with type-safe schemas
Why Use Pydantic AI with the Wallarm MCP Server
Pydantic AI provides unique advantages when paired with Wallarm through the Model Context Protocol.
Full type safety: every MCP tool response is validated against Pydantic models, catching data inconsistencies before they reach your application
Model-agnostic architecture — switch between OpenAI, Anthropic, or Gemini without changing your Wallarm integration code
Structured output guarantee: Pydantic AI ensures tool results conform to defined schemas, eliminating runtime type errors
Dependency injection system cleanly separates your Wallarm connection logic from agent behavior for testable, maintainable code
Wallarm + Pydantic AI Use Cases
Practical scenarios where Pydantic AI combined with the Wallarm MCP Server delivers measurable value.
Type-safe data pipelines: query Wallarm with guaranteed response schemas, feeding validated data into downstream processing
API orchestration: chain multiple Wallarm tool calls with Pydantic validation at each step to ensure data integrity end-to-end
Production monitoring: build validated alert agents that query Wallarm and output structured, schema-compliant notifications
Testing and QA: use Pydantic AI's dependency injection to mock Wallarm responses and write comprehensive agent tests
Wallarm MCP Tools for Pydantic AI (10)
These 10 tools become available when you connect Wallarm to Pydantic AI via MCP:
create_ip_acl_rule
list_type must be "white" or "black". Adds an IP or CIDR range to the global allowlist or denylist
get_client_info
Retrieves details about the Wallarm account, subscription, and feature status
get_discovered_api_inventory
Retrieves the API inventory automatically discovered through passive traffic analysis
get_vulnerability_details
Retrieves comprehensive diagnostic data and exploit evidence for a specific vulnerability ID
list_filtering_nodes
Lists all deployed Wallarm WAF/API gateway filtering nodes
list_ip_acl_rules
Lists configured IP allowlist and denylist rules
search_security_attacks
Searches for security attacks detected by Wallarm, grouped by vector (SQLi, XSS, etc.)
search_security_hits
Shows full request headers and payloads for blocked traffic. Searches for granular individual malicious HTTP request hits intercepted by WAF nodes
search_vulnerabilities
Lists all open security vulnerabilities discovered in live API traffic
update_vulnerability_status
Valid statuses: open, closed, falsepositive. Changes the lifecycle status of a vulnerability (e.g., mark as closed or false positive)
Example Prompts for Wallarm in Pydantic AI
Ready-to-use prompts you can give your Pydantic AI agent to start working with Wallarm immediately.
"List all security attacks detected in the last hour."
"Block the malicious IP address 1.2.3.4 immediately."
"What vulnerabilities are currently open in our production API?"
Troubleshooting Wallarm MCP Server with Pydantic AI
Common issues when connecting Wallarm to Pydantic AI through the Vinkius, and how to resolve them.
MCPServerHTTP not found
pip install --upgrade pydantic-aiWallarm + Pydantic AI FAQ
Common questions about integrating Wallarm MCP Server with Pydantic AI.
How does Pydantic AI discover MCP tools?
MCPServerHTTP instance with the server URL. Pydantic AI connects, discovers all tools, and generates typed Python interfaces automatically.Does Pydantic AI validate MCP tool responses?
Can I switch LLM providers without changing MCP code?
Connect Wallarm with your favorite client
Step-by-step setup guides for every MCP-compatible client and framework:
Anthropic's native desktop app for Claude with built-in MCP support.
AI-first code editor with integrated LLM-powered coding assistance.
GitHub Copilot in VS Code with Agent mode and MCP support.
Purpose-built IDE for agentic AI coding workflows.
Autonomous AI coding agent that runs inside VS Code.
Anthropic's agentic CLI for terminal-first development.
Python SDK for building production-grade OpenAI agent workflows.
Google's framework for building production AI agents.
Type-safe agent development for Python with first-class MCP support.
TypeScript toolkit for building AI-powered web applications.
TypeScript-native agent framework for modern web stacks.
Python framework for orchestrating collaborative AI agent crews.
Leading Python framework for composable LLM applications.
Data-aware AI agent framework for structured and unstructured sources.
Microsoft's framework for multi-agent collaborative conversations.
Connect Wallarm to Pydantic AI
Get your token, paste the configuration, and start using 10 tools in under 2 minutes. No API key management needed.
