mirror of
https://github.com/Shiva108/ai-llm-red-team-handbook.git
synced 2026-02-12 14:42:46 +00:00
Configuration Files
This directory contains configuration files for the AI LLM Red Team scripts.
📄 Files
requirements.txt
Python package dependencies for all scripts.
Install dependencies:
pip install -r config/requirements.txt
Key packages:
requests- HTTP client librarytransformers- HuggingFace transformers (optional)tiktoken- OpenAI tokenization (optional)
pytest.ini
pytest configuration for running tests.
Run tests:
pytest tests/ -v
Configuration includes:
- Test discovery patterns
- Output formatting
- Coverage settings
- Timeout configurations
🔧 Usage
Installation
# Automated (recommended)
./tools/install.sh
# Manual
pip install -r config/requirements.txt
Testing
# Run all tests
pytest tests/ -v
# Run with coverage
pytest tests/ --cov=. --cov-report=html
# Run specific tests
pytest tests/test_prompt_injection.py -v
📝 Customization
Adding Dependencies
Edit requirements.txt and add new packages:
# Example: adding new packages
numpy>=1.21.0
pandas>=1.3.0
Then reinstall:
pip install -r config/requirements.txt
Test Configuration
Edit pytest.ini to customize test behavior:
[pytest]
testpaths = tests
python_files = test_*.py
python_classes = Test*
python_functions = test_*
🔗 Related
Last Updated: 2026-01-07