diff --git a/doc/code/scenarios/0_scenarios.ipynb b/doc/code/scenarios/0_scenarios.ipynb index a68def78b7..82d2c1bc76 100644 --- a/doc/code/scenarios/0_scenarios.ipynb +++ b/doc/code/scenarios/0_scenarios.ipynb @@ -422,7 +422,7 @@ "logging.getLogger(\"pyrit\").setLevel(logging.ERROR)\n", "\n", "response = await get_scenario_service().list_scenarios_async(limit=200) # type: ignore\n", - "print_scenario_list(items=[s.model_dump() for s in response.items])" + "print_scenario_list(items=response.items)" ] }, { diff --git a/doc/code/scenarios/0_scenarios.py b/doc/code/scenarios/0_scenarios.py index e9ee9d603f..1a6fc2a0bf 100644 --- a/doc/code/scenarios/0_scenarios.py +++ b/doc/code/scenarios/0_scenarios.py @@ -168,7 +168,7 @@ async def _build_atomic_attacks_async(self, *, context): logging.getLogger("pyrit").setLevel(logging.ERROR) response = await get_scenario_service().list_scenarios_async(limit=200) # type: ignore -print_scenario_list(items=[s.model_dump() for s in response.items]) +print_scenario_list(items=response.items) # %% [markdown] # diff --git a/doc/code/scenarios/2_custom_scenario_parameters.ipynb b/doc/code/scenarios/2_custom_scenario_parameters.ipynb index 775635fa08..237a69eb62 100644 --- a/doc/code/scenarios/2_custom_scenario_parameters.ipynb +++ b/doc/code/scenarios/2_custom_scenario_parameters.ipynb @@ -464,7 +464,7 @@ "# Supported Parameters section is visible in one and absent in the other.\n", "demo_names = {\"airt.scam\", \"foundry.red_team_agent\"}\n", "response = await get_scenario_service().list_scenarios_async(limit=200) # type: ignore\n", - "print_scenario_list(items=[s.model_dump() for s in response.items if s.scenario_name in demo_names])" + "print_scenario_list(items=[s for s in response.items if s.scenario_name in demo_names])" ] }, { diff --git a/doc/code/scenarios/2_custom_scenario_parameters.py b/doc/code/scenarios/2_custom_scenario_parameters.py index 0481206a91..4605eed057 100644 --- a/doc/code/scenarios/2_custom_scenario_parameters.py +++ b/doc/code/scenarios/2_custom_scenario_parameters.py @@ -198,7 +198,7 @@ # Supported Parameters section is visible in one and absent in the other. demo_names = {"airt.scam", "foundry.red_team_agent"} response = await get_scenario_service().list_scenarios_async(limit=200) # type: ignore -print_scenario_list(items=[s.model_dump() for s in response.items if s.scenario_name in demo_names]) +print_scenario_list(items=[s for s in response.items if s.scenario_name in demo_names]) # %% [markdown] # Notice the `Supported Parameters:` section under `airt.scam`. It's absent diff --git a/doc/code/targets/6_custom_targets.ipynb b/doc/code/targets/6_custom_targets.ipynb index b1aec36f3f..4cbeeff8c3 100644 --- a/doc/code/targets/6_custom_targets.ipynb +++ b/doc/code/targets/6_custom_targets.ipynb @@ -174,7 +174,7 @@ ")\n", "adversarial_config = AttackAdversarialConfig(\n", " target=aoai_chat,\n", - " seed_prompt=initial_red_teaming_prompt,\n", + " first_message=initial_red_teaming_prompt,\n", ")\n", "\n", "gandalf_target = GandalfTarget(level=gandalf_level)\n", diff --git a/doc/code/targets/6_custom_targets.py b/doc/code/targets/6_custom_targets.py index a928db7781..5e0d6a1422 100644 --- a/doc/code/targets/6_custom_targets.py +++ b/doc/code/targets/6_custom_targets.py @@ -72,7 +72,7 @@ ) adversarial_config = AttackAdversarialConfig( target=aoai_chat, - seed_prompt=initial_red_teaming_prompt, + first_message=initial_red_teaming_prompt, ) gandalf_target = GandalfTarget(level=gandalf_level)