From 8f3cc314ad98cfcba5a0370851084198e1089b9b Mon Sep 17 00:00:00 2001
From: GH05TCREW <harmasic@gmail.com>
Date: Sun, 28 Dec 2025 04:52:42 -0700
Subject: [PATCH] chore(setup): update scripts to match latest .env.example

---
 scripts/setup.ps1 | 28 ++++++++++++++++++++--------
 scripts/setup.sh  | 28 ++++++++++++++++++++--------
 2 files changed, 40 insertions(+), 16 deletions(-)

diff --git a/scripts/setup.ps1 b/scripts/setup.ps1
index 6e1cf30..e3f5e2b 100644
--- a/scripts/setup.ps1
+++ b/scripts/setup.ps1
@@ -58,22 +58,34 @@ if (-not (Test-Path ".env")) {
     Write-Host "Creating .env file..."
     @"
 # PentestAgent Configuration
-# Add your API keys here
 
-# OpenAI API Key (required for GPT models)
+# API Keys (set at least one for chat model)
 OPENAI_API_KEY=
-
-# Anthropic API Key (required for Claude models)
 ANTHROPIC_API_KEY=
+GEMINI_API_KEY=
 
-# Model Configuration
+# 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
 
-# Debug Mode
+# Embeddings (for RAG knowledge base)
+# Options: openai, local (default: openai if OPENAI_API_KEY set, else local)
+# PENTESTAGENT_EMBEDDINGS=local
+
+# Settings
 PENTESTAGENT_DEBUG=false
 
-# Max Iterations
-PENTESTAGENT_MAX_ITERATIONS=50
+# 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
 "@ | Set-Content -Path ".env" -Encoding UTF8
     Write-Host "[OK] .env file created"
     Write-Host "[!] Please edit .env and add your API keys"
diff --git a/scripts/setup.sh b/scripts/setup.sh
index bac4a3f..4808520 100644
--- a/scripts/setup.sh
+++ b/scripts/setup.sh
@@ -52,22 +52,34 @@ if [ ! -f ".env" ]; then
     echo "Creating .env file..."
     cat > .env << EOF
 # PentestAgent Configuration
-# Add your API keys here
 
-# OpenAI API Key (required for GPT models)
+# API Keys (set at least one for chat model)
 OPENAI_API_KEY=
-
-# Anthropic API Key (required for Claude models)
 ANTHROPIC_API_KEY=
+GEMINI_API_KEY=
 
-# Model Configuration
+# 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
 
-# Debug Mode
+# Embeddings (for RAG knowledge base)
+# Options: openai, local (default: openai if OPENAI_API_KEY set, else local)
+# PENTESTAGENT_EMBEDDINGS=local
+
+# Settings
 PENTESTAGENT_DEBUG=false
 
-# Max Iterations
-PENTESTAGENT_MAX_ITERATIONS=50
+# 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"