Learn how to use Aixploit for LLM security testing
Example 1: Basic Prompt Injection
from aixploit.plugins import PromptInjection from aixploit.core import run # Setup basic prompt injection target = ["Openai", "", "gpt-3.5-turbo"] attackers = [PromptInjection("quick")] # Execute attack results = run(attackers, target, os.getenv("OPENAI_KEY"))
Example 2: Custom Attack Patterns
from aixploit.plugins import CustomAttack from aixploit.core import run # Define custom attack pattern custom_pattern = { "prompt": "Your custom prompt here", "target": "Expected vulnerable response" } # Execute custom attack attacker = CustomAttack(custom_pattern) results = run([attacker], target, os.getenv("OPENAI_KEY"))
Example 3: Advanced Configuration
from aixploit.plugins import AdvancedAttack from aixploit.core import run # Configure advanced settings config = { "max_tokens": 1000, "temperature": 0.7, "retry_count": 3 } # Execute with advanced configuration attacker = AdvancedAttack(config) results = run([attacker], target, os.getenv("OPENAI_KEY"))