From 24032bb9bfe04e6922fb52395eebd48ed57b0630 Mon Sep 17 00:00:00 2001 From: Raja Sekhar Rao Dheekonda Date: Tue, 17 Mar 2026 11:37:17 -0700 Subject: [PATCH 1/2] fix: Switch notebook to OpenAI gpt-4o and read keys from env vars --- examples/airt/agentic_red_teaming_attacks.ipynb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/airt/agentic_red_teaming_attacks.ipynb b/examples/airt/agentic_red_teaming_attacks.ipynb index 913dcf47..e8742fb9 100644 --- a/examples/airt/agentic_red_teaming_attacks.ipynb +++ b/examples/airt/agentic_red_teaming_attacks.ipynb @@ -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", From cf3efc71e109fe8978351e4d7eff8fa4b4adcb26 Mon Sep 17 00:00:00 2001 From: Raja Sekhar Rao Dheekonda Date: Tue, 17 Mar 2026 11:47:12 -0700 Subject: [PATCH 2/2] fix: Add organization param and fix VaultGuard transforms reference --- examples/airt/agentic_red_teaming_attacks.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/airt/agentic_red_teaming_attacks.ipynb b/examples/airt/agentic_red_teaming_attacks.ipynb index e8742fb9..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", "```" ] }, @@ -340,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",