|
8 | 8 | import logging |
9 | 9 | from dataclasses import dataclass, field |
10 | 10 | from pathlib import Path |
11 | | -from typing import TYPE_CHECKING, Any, Optional, cast |
| 11 | +from typing import TYPE_CHECKING, Any, cast |
12 | 12 |
|
13 | 13 | from pyrit.common.apply_defaults import REQUIRED_VALUE, apply_defaults |
14 | 14 | from pyrit.common.path import EXECUTOR_SEED_PROMPT_PATH |
@@ -80,7 +80,6 @@ class SplitPayloadAttackContext(MultiTurnAttackContext[Any]): |
80 | 80 | design_pattern: str = "component_assembly" |
81 | 81 |
|
82 | 82 |
|
83 | | -@dataclass |
84 | 83 | class SplitPayloadAttackResult(AttackResult): |
85 | 84 | """Result of the split-payload attack strategy execution.""" |
86 | 85 |
|
@@ -161,9 +160,9 @@ def __init__( |
161 | 160 | *, |
162 | 161 | objective_target: PromptTarget = REQUIRED_VALUE, # type: ignore[ty:invalid-parameter-default] |
163 | 162 | attack_adversarial_config: AttackAdversarialConfig, |
164 | | - attack_converter_config: Optional[AttackConverterConfig] = None, |
165 | | - attack_scoring_config: Optional[AttackScoringConfig] = None, |
166 | | - prompt_normalizer: Optional[PromptNormalizer] = None, |
| 163 | + attack_converter_config: AttackConverterConfig | None = None, |
| 164 | + attack_scoring_config: AttackScoringConfig | None = None, |
| 165 | + prompt_normalizer: PromptNormalizer | None = None, |
167 | 166 | design_pattern: str = "component_assembly", |
168 | 167 | max_turns: int = 5, |
169 | 168 | max_regenerations: int = 3, |
@@ -270,7 +269,7 @@ def __init__( |
270 | 269 | self._max_turns = max_turns |
271 | 270 | self._max_regenerations = max_regenerations |
272 | 271 |
|
273 | | - def get_attack_scoring_config(self) -> Optional[AttackScoringConfig]: |
| 272 | + def get_attack_scoring_config(self) -> AttackScoringConfig | None: |
274 | 273 | """ |
275 | 274 | Get the attack scoring configuration. |
276 | 275 |
|
|
0 commit comments