Skip to main content

RAI is a next-gen CLI framework to automate the creation of intelligent agents and teams for cybersecurity and offensive security operations

Project description

RAI โ€“ Next-Level Automation Tool & Framework for Building LLM Agents and Teams in Cybersecurity

RAI

lightweight, faster LLM Agents,Team building with YAML Configuration

Features | Installation | Usage | Building Agents & Teams with YAML Configuration

GitHub last commit GitHub release (latest by date) GitHub license

RAI (Revolt AI Agent) is a modern, YAML-driven CLI tool and framework for building intelligent agents and agent teams tailored for cybersecurity automation, offensive security, and penetration testing operations. Built on top of the powerful Agno framework, RAI enables security professionals, red teamers, and AI hackers to design, orchestrate, and deploy advanced LLM-powered agents without writing traditional code. Its no-code architecture leverages structured YAML configurations to define agent behavior, tools, and team collaboration logic.

Features๐Ÿ”ง:


demo

  • ๐Ÿง  Interactive Shell Mode โ€“ Engage in real-time conversations with LLM agents and teams via a powerful interactive CLI. Seamlessly switch between agents or teams with intuitive commands.

  • ๐Ÿ“ YAML-Based Agent & Team Building โ€“ Define agents and teams using easy-to-edit YAML templates. Accelerate development with low-code configurations and smart defaults.

  • ๐Ÿค– Multi-Agent & Team Support โ€“ Build, run, and manage multiple agents or teams in parallel with full operational isolation and coordination.

  • ๐Ÿ”Œ Tool Integration (SSE & stdio) โ€“ Integrate custom tools via Server-Sent Events or standard I/O for dynamic agent-tool communication.

  • ๐Ÿ”„ Dynamic Team Allocation โ€“ Flexibly assign, reassign, or reconfigure agents across different teams at runtime to optimize task workflows.

  • ๐Ÿงฉ MCP-Compatible Infrastructure โ€“ Built with modularity in mind, RAI is ready for integration with Model Context Protocol (MCP) tooling and future agent standards.

  • โš™๏ธ Fast & Flexible Configuration โ€“ Lightweight setup with extensible configuration options. Designed for developers who want control without the clutter.

  • ๐Ÿง  Built-In Reasoning Engine โ€“ Agents can think, reason, and decide intelligently before taking action, enabling smarter task execution.

  • ๐Ÿ›ก๏ธ Cybersecurity-First Design โ€“ Purpose-built for red teaming, bug bounty automation, recon, exploit development, and offensive security workflows.

  • ๐Ÿงฌ Agent-to-Agent Communication โ€“ Enable inter-agent messaging within teams, allowing agents to delegate tasks, collaborate, and share results autonomously.

Supported LLM Providers:

  • Anthropic
  • AWS
  • Azure
  • Cohere
  • DeepInfra
  • DeepSeek
  • Fireworks
  • Google (Gemini)
  • Groq
  • Hugging Face
  • IBM
  • InternLM
  • LiteLLM
  • LMStudio
  • Meta (LLaMA)
  • Mistral
  • NVIDIA
  • Ollama
  • OpenAI
  • OpenRouter
  • Perplexity
  • SambaNova
  • Together.ai
  • xAI

Installation๐Ÿš€

RAI can be easily installed using uv โ€” a fast Python package manager designed for modern workflows.

๐Ÿ“ฆ Install with uv

uv tool install rai

โœ… Make sure you have Python 3.13 or newer installed.
โœ… uv automatically handles virtual environments, speed, and dependency resolution.

Usage:


rai -h
    ____     ___     ____
   / __ \   /   |   /  _/
  / /_/ /  / /| |   / /  
 / _, _/  / ___ | _/ /   
/_/ |_|  /_/  |_|/___/   
                         

                     - RevoltSecurities


[DESCRIPTION]: 

    RAI is a next-gen CLI tool and framework to automate the creation of intelligent agents and teams for cybersecurity and offensive security operations

[USAGE]: 

    rai [flags]

[FLAGS]:

    -h,    --help                 :  Show this help message and exit.
    -v,    --version              :  Show current version of RAI.
    -cp,   --config-path          :  Path to YAML config file (default: $HOME/.config/RAI/raiagent.yaml).
    -sup,  --show-updates         :  Show latest update details.
    -up,   --update               :  Update RAI to the latest version (manual YAML update).

๐Ÿ›  YAML Configuration

RAI allows you to define AI Agents and Agent Teams using a simple YAML configuration. This configuration determines how agents behave, what models they use, what tools are attached, and how they collaborate as teams.


โœณ๏ธ Agent Configuration (agents)

Each agent must define the following required fields:

Field Type Description
name string Unique agent name, must use - or _ only (e.g., web_pentest_agent)
model string The provider name (e.g., openai, gemini, groq, xai)
model-id string The specific model ID to use (e.g., gpt-4, gemini-2.0-pro)
apikey string API key for the chosen provider
role string A short sentence describing the agentโ€™s purpose
description string Multi-line detailed description of the agentโ€™s capabilities
instructions string A clear set of multi-line operational guidelines for the agent
tools toolconfig Tools assigned to the agent (sse or stdio types with required params)

โœ… The only optional field is think: true, which enables your agent to think and analyze before its response


๐Ÿง  Team Configuration (teams)

Teams are collaborative groups of agents that share analysis tasks.

Each team must define the following required fields:

Field Type Description
name string Team name, must use - or _ only (e.g., pentest_team)
mode string Team mode (e.g., coordinate, route, collaborate)
model string Provider name for internal logic (same as in agents)
model-id string Model ID used for internal processing
apikey string API key for the teamโ€™s model
instructions string Multi-line instructions on how the team should collaborate
members list List of agent names (must match agent name fields)
tools toolconfig Tools assigned to the agent (sse or stdio types with required params)
success_criteria string Configure your teams collaboration success criteria to achieve your goal for the RAI Team task

โœ… think: true is an optional field to allow team-wide reasoning before responding.


๐Ÿงฉ Agent Naming Convention

To ensure consistency and compatibility:

  • Agent and team name fields must:
    • Be unique across agents and teams
    • Use only lowercase characters, numbers, -, or _
    • Not contain spaces or special characters

โœ… Valid: api_pentest_agent, web-agent-1
โŒ Invalid: Agent 01, Web*Pentest


๐Ÿ”— Team Member Allocation Rules

  • members must list agent names already defined under the agents: section.
  • All member agents must be valid and fully configured before referencing them in a team.
  • Duplicate agent names or undeclared agents will raise errors.

๐Ÿ”ง Tools Configuration

Agents or Teams can integrate external or internal MCP tools via yaml configuration:

SSE Tool (Server-Sent Events)

- type: "sse"
  name: "tool_name"
  params:
    url: "http://host:port/endpoint"
    headers:
      Authorization: "Bearer your_token_here"

Stdio Tool (Local MCP server execution)

- type: "stdio"
  name: "tool_name"
  params:
    command: "command_to_run (ex:uv)"
    args: ["arg1", "arg2", "argN"]

๐Ÿ“ฆ Sample Full YAML Configuration

nano ~/.config/RAI/raiagent.yaml

and paste these below content and also use valid models and secret apikeys to work with RAIโšก

agents:
  - name: "web_pentest_agent"
    model: "gemini"
    model-id: "gemini-2.0-flash-exp"
    apikey: "AIzaSyDnd-REDACTED-1234567890"
    role: "An expert web application penetration tester."
    description: |
      This agent performs in-depth analysis of web applications, including:
      - XSS, SQLi, CSRF, SSRF detection
      - Payload recommendations
      - Mitigation advice using OWASP guidelines
    instructions: |
      - Analyze HTML, JS, HTTP requests
      - Follow ethical boundaries
      - Provide clear markdown-formatted results
    tools:
      - type: "sse"
        name: "web_tool"
        params:
          url: "http://localhost:8000/sse"
          headers:
            Authorization: "Bearer sample-token-123"
    markdown: true
    enable_history: true

  - name: "api_pentest_agent"
    model: "gemini"
    model-id: "gemini-2.0-flash-exp"
    apikey: "AIzaSyDnd-REDACTED-0987654321"
    role: "API security expert."
    description: |
      Focused on discovering API vulnerabilities:
      - BOLA, Mass Assignment, Broken Auth
      - CORS and Rate Limiting checks
    instructions: |
      Use OWASP API Top 10 as the baseline. Respond only with ethical suggestions.
    tools:
      - type: "stdio"
        name: "api_tool"
        params:
          command: "python3"
          args: ["api_runner.py"]
    markdown: true
    enable_history: true

  - name: "ctf_agent"
    model: "gemini"
    model-id: "gemini-2.0-flash-exp"
    apikey: "AIzaSyDnd-REDACTED-CTFKEY"
    role: "CTF solving agent"
    description: |
      Solves CTF challenges in web, pwn, reverse engineering, crypto, and forensics.
    instructions: |
      - Give step-by-step answers
      - Output payloads and flag extraction logic
    markdown: true
    enable_history: true

teams:
  - name: "pentest_team_alpha"
    mode: "coordinate"
    model: "gemini"
    model-id: "gemini-2.0-flash-exp"
    apikey: "AIzaSyDnd-REDACTED-TEAMKEY"
    instructions: |
      Collaborate across agents to detect and explain vulnerabilities in web or API apps.
    tools:
      - type: "stdio"
        name: "api_tool"
        params:
          command: "python3"
          args: ["api_runner.py"]
    members:
      - "web_pentest_agent"
      - "api_pentest_agent"
      - "ctf_agent"
    success_criteria: "All vulnerabilities are clearly identified, explained, and mitigated."
    think: true

๐Ÿš€ Future Enhancement Plan:

RAI (Revolt AI Agent) is under continuous development ๐Ÿ› ๏ธ โ€” evolving rapidly to empower cybersecurity automation with intelligent, collaborative agents. Upcoming features include:

  • ๐Ÿ–ฅ Agent UI with Agno UI Integration
    A powerful web-based UI to manage, monitor, and interact with agents, tools, and teams visually โ€” built on Agno's robust interface layer.

  • ๐Ÿง  Memory & Session Storage Management
    Agents will support persistent session memory:

    • Maintain conversation continuity
    • Recall user preferences, task history, and learned context
  • ๐Ÿ“š Agent & Team Knowledge Injection
    Allow agents and teams to use custom knowledge bases, enabling them to:

    • Ingest structured/unstructured data sources (e.g., markdown, PDFs, JSON, code)
    • Improve reasoning and task performance through embedded knowledge
    • Learn iteratively and adapt during operations
  • ๐Ÿงฉ Agent Knowledge Learning Loop
    Equip agents with mechanisms to analyze outcomes, refine their behavior, and build contextual awareness from completed tasks.

โš ๏ธ RAI is Under Continuous Development

โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€

๐Ÿ› ๏ธ RAI (Revolt AI Agent) is an actively evolving project built on top of the powerful Agno framework. This means you can expect:

๐Ÿ”„ Regular updates & new feature drops
๐Ÿงช Experimental support for cutting-edge agent workflows
๐Ÿ”ง Frequent performance and usability improvements
๐Ÿงฐ Expanding tool integrations and LLM backend compatibility
๐Ÿ“ฆ Community-driven contributions & enhancements welcome!

While RAI is already production-capable, itโ€™s designed to grow fastโ€” so expect changes, iteration, and rapid innovation.

โž• Stay updated. Join the journey. Contribute. Hack with AI. โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€


โค๏ธ Acknowledgements & Community Contribution:

A special thanks to the Agno Framework for providing a powerful foundation for RAI. Their contributions have made it possible to build a sophisticated, flexible, and scalable platform that empowers cybersecurity professionals worldwide. ๐Ÿ™

RAI (Revolt AI Agent) is developed with โค๏ธ by RevoltSecurities, driven by a passion for open-source and cybersecurity innovation. We are excited to share this tool with the community and empower the next generation of red teamers, security researchers, and AI hackers. ๐Ÿš€ We welcome contributions, ideas, and feedback from the open-source community. Together, we can make RAI even more powerful and continue to drive innovation in the cybersecurity field. Your contributions, whether in the form of code, documentation, bug reports, or ideas, are highly appreciated. Let's build, learn, and grow together! ๐Ÿค

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

revolt_rai-1.0.0.tar.gz (72.8 kB view details)

Uploaded Source

Built Distribution

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

revolt_rai-1.0.0-py3-none-any.whl (23.2 kB view details)

Uploaded Python 3

File details

Details for the file revolt_rai-1.0.0.tar.gz.

File metadata

  • Download URL: revolt_rai-1.0.0.tar.gz
  • Upload date:
  • Size: 72.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.6.14

File hashes

Hashes for revolt_rai-1.0.0.tar.gz
Algorithm Hash digest
SHA256 53a1d33aa4663b269a7c807ee820ad309bd275c17eed41e2e37529558c0fff78
MD5 561e4f8be3c230f94122ccd136ad3108
BLAKE2b-256 af09d560b5e543a8bcfe871569a456c4e282256520baf371005975ba9edaaae2

See more details on using hashes here.

File details

Details for the file revolt_rai-1.0.0-py3-none-any.whl.

File metadata

  • Download URL: revolt_rai-1.0.0-py3-none-any.whl
  • Upload date:
  • Size: 23.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.6.14

File hashes

Hashes for revolt_rai-1.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 8278d6004c0dec3b7854bb9cedc4fcc0ce3bfb91160afc3b9faf2ce3f2bd8a21
MD5 fe26a90236acb05598427615d4888c00
BLAKE2b-256 2143f9ae2983cb245d549f9921261f83b77ab9b5556719a3266b142fcb9f0613

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