Getting Started

export OPENAI_KEY="sk-xxxxx"
export OLLAMA_URL="hxxp:"
export OLLAMA_API_KEY="ollama"

from aixploit.plugins import PromptInjection
from aixploit.core import run

target = ["Openai", "", "gpt-3.5-turbo"]
attackers = [PromptInjection("quick")]

results = run(attackers, target, os.getenv("OPENAI_KEY"))

for idx, attacker in enumerate(attackers):
    print(f"Attacker: {attacker.__class__.__name__}")
    print(f"Success Rate: {success_rates_percentage[idx] * 100:.2f}%")