mirror of
https://github.com/Shiva108/ai-llm-red-team-handbook.git
synced 2026-02-12 14:42:46 +00:00
3.1 KiB
3.1 KiB
Quick Start Guide - AI LLM Red Team Scripts
Installation
Option 1: Automated Installation (Recommended)
cd /home/e/Desktop/ai-llm-red-team-handbook/scripts
./tools/install.sh
The installation script will:
- ✓ Check Python 3.8+ installation
- ✓ Create a virtual environment (
venv/) - ✓ Install all dependencies from
requirements.txt - ✓ Make all scripts executable
- ✓ Create helper scripts (
activate.sh,test_install.py) - ✓ Run verification tests
Option 2: Manual Installation
# Create virtual environment
python3 -m venv venv
# Activate it
source venv/bin/activate
# Install dependencies
pip install -r config/requirements.txt
# Make scripts executable
chmod +x workflows/*.py
Activation
After installation, activate the environment:
# Use the helper script
source activate.sh
# Or activate manually
source venv/bin/activate
Verification
Test that everything is installed correctly:
python3 test_install.py
Basic Usage
Running Individual Scripts
# Get help for any script
python3 prompt_injection/chapter_14_prompt_injection_01_prompt_injection.py --help
# Run a tokenization analysis
python3 utils/chapter_09_llm_architectures_and_system_components_01_utils.py
# Test RAG poisoning
python3 rag_attacks/chapter_12_retrieval_augmented_generation_rag_pipelines_01_rag_attacks.py
Running Workflows
# Full security assessment
python3 workflows/full_assessment.py \
--target https://api.example.com \
--output report.json \
--verbose
# RAG-focused testing
python3 workflows/rag_exploitation.py \
--target https://api.example.com \
--vector-db chromadb
# Plugin-focused testing
python3 workflows/plugin_pentest.py \
--target https://api.example.com \
--plugins weather,calculator
Troubleshooting
Python Version Issues
Ensure you have Python 3.8 or higher:
python3 --version
If you have an older version, install Python 3.8+ before running the installer.
Virtual Environment Issues
If the virtual environment fails to activate:
# Remove and recreate it
rm -rf venv
python3 -m venv venv
source venv/bin/activate
pip install -r config/requirements.txt
Missing Dependencies
If specific packages fail to install, try installing them individually:
pip install transformers
pip install tiktoken
pip install requests
Permission Denied
If you get "Permission denied" errors:
chmod +x install.sh
chmod +x workflows/*.py
Deactivation
When you're done, deactivate the virtual environment:
deactivate
Next Steps
- ✅ Read the main
README.mdfor detailed documentation - ✅ Explore scripts in each category folder
- ✅ Review the handbook chapters for theory
- ✅ Customize scripts for your specific needs
Support
For more information, refer to:
README.md- Main documentation- Individual script docstrings - Run with
--help - Handbook chapters - Source material in
/docs
Security Warning: Only use these scripts for authorized security testing!