diff --git a/examples/airt/agentic_red_teaming_attacks.ipynb b/examples/airt/agentic_red_teaming_attacks.ipynb index 913dcf47..5c2e9aa0 100644 --- a/examples/airt/agentic_red_teaming_attacks.ipynb +++ b/examples/airt/agentic_red_teaming_attacks.ipynb @@ -20,7 +20,7 @@ "\n", "```bash\n", "export CRUCIBLE_API_KEY=\"your-api-key\" # from https://platform.dreadnode.io/account\n", - "export GROQ_API_KEY=\"your-groq-api-key\"\n", + "export OPENAI_API_KEY=\"your-openai-api-key\"\n", "```" ] }, @@ -52,8 +52,9 @@ "outputs": [], "source": [ "CRUCIBLE_API_KEY = os.environ[\"CRUCIBLE_API_KEY\"] # https://platform.dreadnode.io/account\n", + "OPENAI_API_KEY = os.environ[\"OPENAI_API_KEY\"]\n", "CRUCIBLE_URL = \"https://platform.dreadnode.io\"\n", - "ATTACKER_MODEL = \"groq/meta-llama/llama-4-scout-17b-16e-instruct\"\n", + "ATTACKER_MODEL = \"openai/gpt-4o\"\n", "EVALUATOR_MODEL = ATTACKER_MODEL\n", "\n", "TOOLSHED_URL = \"https://toolshed.platform.dreadnode.io\"\n", @@ -339,7 +340,7 @@ " frontier_size=10,\n", " branching_factor=5,\n", " on_topic_threshold=0.3,\n", - " hooks=[apply_input_transforms(transforms_toolshed)],\n", + " hooks=[apply_input_transforms(transforms_vaultguard)],\n", " )\n", " .with_(max_trials=VAULTGUARD_MAX_TRIALS)\n", " .add_objective(vaultguard_scorer, direction=\"maximize\", name=\"flag_capture\")\n",