mirror of
https://github.com/GH05TCREW/pentestagent.git
synced 2026-03-07 06:14:08 +00:00
117 lines
3.5 KiB
Bash
117 lines
3.5 KiB
Bash
#!/bin/bash
|
|
# PentestAgent Setup Script
|
|
|
|
set -e
|
|
|
|
echo "=================================================================="
|
|
echo " PENTESTAGENT"
|
|
echo " AI Penetration Testing"
|
|
echo "=================================================================="
|
|
echo ""
|
|
|
|
# Check Python version
|
|
echo "Checking Python version..."
|
|
python_version=$(python3 --version 2>&1 | cut -d' ' -f2 | cut -d'.' -f1,2)
|
|
required_version="3.10"
|
|
|
|
if [ "$(printf '%s\n' "$required_version" "$python_version" | sort -V | head -n1)" != "$required_version" ]; then
|
|
echo "Error: Python $required_version or higher is required (found $python_version)"
|
|
exit 1
|
|
fi
|
|
echo "[OK] Python $python_version"
|
|
|
|
# Create virtual environment
|
|
echo "Creating virtual environment..."
|
|
if [ ! -d "venv" ]; then
|
|
python3 -m venv venv
|
|
echo "[OK] Virtual environment created"
|
|
else
|
|
echo "[OK] Virtual environment exists"
|
|
fi
|
|
|
|
# Activate virtual environment
|
|
echo "Activating virtual environment..."
|
|
source venv/bin/activate
|
|
|
|
# Upgrade pip
|
|
echo "Upgrading pip..."
|
|
pip install --upgrade pip
|
|
|
|
# Install dependencies
|
|
echo "Installing dependencies..."
|
|
pip install -e ".[all]"
|
|
echo "[OK] Dependencies installed"
|
|
|
|
# Install playwright browsers
|
|
echo "Installing Playwright browsers..."
|
|
playwright install chromium
|
|
echo "[OK] Playwright browsers installed"
|
|
|
|
# Create .env file if not exists
|
|
if [ ! -f ".env" ]; then
|
|
echo "Creating .env file..."
|
|
cat > .env << EOF
|
|
# PentestAgent Configuration
|
|
|
|
# API Keys (set at least one for chat model)
|
|
OPENAI_API_KEY=
|
|
ANTHROPIC_API_KEY=
|
|
GEMINI_API_KEY=
|
|
|
|
# For web search functionality (optional)
|
|
TAVILY_API_KEY=
|
|
|
|
# Chat Model (any LiteLLM-supported model)
|
|
# OpenAI: gpt-5, gpt-4.1, gpt-4.1-mini
|
|
# Anthropic: claude-sonnet-4-20250514, claude-opus-4-20250514
|
|
# Google: gemini models require gemini/ prefix (e.g., gemini/gemini-2.5-flash)
|
|
# Other providers: azure/, bedrock/, groq/, ollama/, together_ai/ (see litellm docs)
|
|
PENTESTAGENT_MODEL=gpt-5
|
|
|
|
# Embeddings (for RAG knowledge base)
|
|
# Options: openai, local (default: openai if OPENAI_API_KEY set, else local)
|
|
# PENTESTAGENT_EMBEDDINGS=local
|
|
|
|
# Settings
|
|
PENTESTAGENT_DEBUG=false
|
|
|
|
# Auto-launch vendored HexStrike on connect (true/false)
|
|
# If true, the MCP manager will attempt to start vendored HexStrike servers
|
|
# that are configured or detected under `third_party/hexstrike`.
|
|
LAUNCH_HEXTRIKE=false
|
|
|
|
# Agent max iterations (regular agent + crew workers, default: 30)
|
|
# PENTESTAGENT_AGENT_MAX_ITERATIONS=30
|
|
|
|
# Orchestrator max iterations (crew mode coordinator, default: 50)
|
|
# PENTESTAGENT_ORCHESTRATOR_MAX_ITERATIONS=50
|
|
EOF
|
|
echo "[OK] .env file created"
|
|
echo "[!] Please edit .env and add your API keys"
|
|
fi
|
|
|
|
# Create loot directory for reports
|
|
mkdir -p loot
|
|
echo "[OK] Loot directory created"
|
|
|
|
# Install vendored HexStrike dependencies automatically if present
|
|
if [ -f "third_party/hexstrike/requirements.txt" ]; then
|
|
echo "Installing vendored HexStrike dependencies..."
|
|
bash scripts/install_hexstrike_deps.sh
|
|
fi
|
|
|
|
echo ""
|
|
echo "=================================================================="
|
|
echo "Setup complete!"
|
|
echo ""
|
|
echo "To get started:"
|
|
echo " 1. Edit .env and add your API keys"
|
|
echo " 2. Activate the virtual environment: source venv/bin/activate"
|
|
echo " 3. Run PentestAgent: pentestagent or python -m pentestagent"
|
|
echo ""
|
|
echo "For Docker usage:"
|
|
echo " docker-compose up pentestagent"
|
|
echo " docker-compose --profile kali up pentestagent-kali"
|
|
echo ""
|
|
echo "=================================================================="
|