Skip to content

Commit c0a2dab

Browse files
author
hannahwestra25
committed
bug fix
1 parent 9fedf2f commit c0a2dab

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

pyrit/scenario/scenarios/adaptive/adaptive_scenario.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
from pyrit.scenario.core.atomic_attack import AtomicAttack
2727
from pyrit.scenario.core.attack_technique import AttackTechnique
2828
from pyrit.scenario.core.scenario import Scenario
29+
from pyrit.scenario.core.scenario_target_defaults import get_default_adversarial_target
2930
from pyrit.scenario.scenarios.adaptive.dispatcher import (
3031
AdaptiveTechniqueDispatcher,
3132
TechniqueBundle,
@@ -262,11 +263,14 @@ def _build_techniques_dict(
262263
logger.warning(f"Skipping technique '{technique_name}': {type(exc).__name__}: {exc}")
263264
continue
264265
eval_hash = compute_inner_attack_eval_hash(attack=technique.attack)
266+
adversarial_chat = factory.adversarial_chat
267+
if adversarial_chat is None and factory.uses_adversarial:
268+
adversarial_chat = get_default_adversarial_target()
265269
techniques[eval_hash] = TechniqueBundle(
266270
attack=technique.attack,
267271
name=technique_name,
268272
seed_technique=technique.seed_technique,
269-
adversarial_chat=factory.adversarial_chat,
273+
adversarial_chat=adversarial_chat,
270274
)
271275

272276
if not techniques:

0 commit comments

Comments
 (0)