PlaybookGen

A CLI tool for generating AttackMate playbooks using LLMs (currently supports OpenAI's GPT models).

This is a proof of concept and currently can only generate playbooks with the browser and sleep commands of AttackMate.

Setup

1. Clone the repository and create a .env file in the project root:

   OPENAI_API_KEY="your_api_key"

2. Install the package (choose one method):

   With uv

   uv venv
   source .venv/bin/activate
   uv pip install -e .

   ⚠️ Note: uv pip install installs into the .venv created by uv venv, even if the environment isn't manually activated.

   With pip

   python -m venv venv
   source venv/bin/activate
   pip install -e .

3. Install Playwright with its dependencies:

   playwright install-deps
   playwright install

Usage

playbookgen --help

To generate a playbook run the following, then insert prompt and press enter:

playbookgen --output playbook.yml

Demo