99from pyrit .common .path import EXECUTOR_RED_TEAM_PATH , SCORER_SEED_PROMPT_PATH
1010from pyrit .executor .attack import ContextComplianceAttack , RedTeamingAttack , RolePlayAttack , RolePlayPaths
1111from pyrit .executor .attack .core .attack_config import AttackAdversarialConfig , AttackScoringConfig
12- from pyrit .models import Parameter , SeedAttackGroup
12+ from pyrit .models import Parameter , SeedAttackGroup , SeedPrompt
1313from pyrit .prompt_target import PromptTarget
1414from pyrit .scenario .core .atomic_attack import AtomicAttack
1515from pyrit .scenario .core .attack_technique import AttackTechnique
@@ -174,9 +174,9 @@ def _get_atomic_attack_from_strategy(self, *, strategy: str, seed_groups: list[S
174174
175175 if strategy == "persuasive_rta" :
176176 # Set system prompt to generic persuasion persona
177- self ._adversarial_config .system_prompt_path = Path (
178- PERSUASION_DECEPTION_PATH , "persuasion_persona_generic.yaml"
179- ). resolve ()
177+ self ._adversarial_config .system_prompt = SeedPrompt . from_yaml_file (
178+ Path ( PERSUASION_DECEPTION_PATH , "persuasion_persona_generic.yaml" ). resolve ()
179+ )
180180
181181 attack_strategy = RedTeamingAttack (
182182 objective_target = self ._objective_target ,
@@ -193,7 +193,7 @@ def _get_atomic_attack_from_strategy(self, *, strategy: str, seed_groups: list[S
193193 )
194194 elif strategy == "context_compliance" :
195195 # Set system prompt to default
196- self ._adversarial_config .system_prompt_path = None
196+ self ._adversarial_config .system_prompt = None
197197
198198 attack_strategy = ContextComplianceAttack (
199199 objective_target = self ._objective_target ,
0 commit comments