Skip to main content

AI Red Teaming Tool and Framework

Project description

How to Install dtx

Before you begin, choose how you want to run dtx depending on your environment and requirements.


Option 1: Install dtx locally with full dependencies (torch etc.)

This is recommended if you plan to run local models (e.g., Hugging Face, Ollama) on your machine.

pip install dtx[torch]

Includes:

  • Core CLI
  • torch, transformers for local LLM and classifier execution
  • Supports all datasets and local execution

Option 2: Install dtx if torch is already installed

If your environment already has torch installed (for example, in a GPU-accelerated ML environment), you can skip extras:

pip install dtx

dtx will use your existing torch installation.

Tip: Verify torch is installed:

python -c "import torch; print(torch.__version__)"

Option 3: Use uv for fast installation in a clean environment

If you're creating a new environment and want fast dependency resolution with uv:

Install uv

curl -LsSf https://astral.sh/uv/install.sh | sh

Install dtx with full dependencies

uv pip install dtx[torch]

Option 4: Use Docker wrapper (ddtx)

If you prefer Dockerized execution (no local torch install required), you can use ddtx.

  1. Install dtx (for the ddtx wrapper CLI):
pip install dtx
  1. Use ddtx to run inside Docker:
ddtx redteam scope "Describe your agent" output.yml

Features:

  • No need to install torch locally
  • Fully containerized execution
  • Automatically mounts .env and working directories
  • Use Docker-managed templates and tools

Summary of Options

Method Use case Install command
Local, full dependencies Full feature set, local models pip install dtx[torch]
Local, existing torch You already have torch installed pip install dtx
New env, fast install Clean, fast setup uv pip install dtx[torch]
Docker (ddtx) No local Python dependencies, isolated pip install dtx + use ddtx CLI

How to Get Started (Development)

Follow these steps to set up dtx for local development.


1. Clone the Repository

git clone https://github.com/YOUR-ORG/dtx.git
cd dtx

2. Install Dependencies

Run the following command:

make install

This will:

  • Ensure the poetry-plugin-export is installed
  • Install all dependencies including dev and torch groups
  • Set up the dtx package in editable mode

3. Activate the Virtual Environment

Run this in your shell to activate the Poetry-managed virtual environment:

eval $(poetry env activate)

Alternatively, you can run:

make venv

This will print the exact eval command you need to run manually. For example:

Note: make cannot modify your current shell environment, so you must run the eval command directly.


4. Run the CLI

Once the environment is active, you can use the CLI:

dtx --help

5. Export Requirements (Optional)

To export the current environment to a requirements.txt file (useful for pip-based tools or deployment):

make export

This will generate a requirements.txt file including development and optional dependencies like torch.

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

dtx-0.29.0.tar.gz (772.1 kB view details)

Uploaded Source

Built Distribution

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

dtx-0.29.0-py3-none-any.whl (878.3 kB view details)

Uploaded Python 3

File details

Details for the file dtx-0.29.0.tar.gz.

File metadata

  • Download URL: dtx-0.29.0.tar.gz
  • Upload date:
  • Size: 772.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.1.4 CPython/3.11.8 Linux/6.11.0-1018-azure

File hashes

Hashes for dtx-0.29.0.tar.gz
Algorithm Hash digest
SHA256 81c2bd76c5c69b0ebc2bf38980aecd6e8046e840ead52d71b1193c8eaa792060
MD5 625ae584a7c474acc9f9a838dbde11d0
BLAKE2b-256 63fd5e93e4927e562e0da0515e052df286cd388b7cd15aaa7e2e808e144ac3ac

See more details on using hashes here.

File details

Details for the file dtx-0.29.0-py3-none-any.whl.

File metadata

  • Download URL: dtx-0.29.0-py3-none-any.whl
  • Upload date:
  • Size: 878.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/2.1.4 CPython/3.11.8 Linux/6.11.0-1018-azure

File hashes

Hashes for dtx-0.29.0-py3-none-any.whl
Algorithm Hash digest
SHA256 b393b7c0f849bc2e36376c937aa33aedc95dc430cf90a13aaff2e4b259fde773
MD5 5ed6e7cf4d23c5fa8501fa77f32d978a
BLAKE2b-256 46e459fc6f32364d8ecd649552cd9b84e2380e06badee830aa76e3f6f1cd03f6

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