Fix split-payload CI after rebase

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: Eve Kazarian

pyrit/executor/attack/multi_turn/split_payload.py

@@ -32,7 +32,6 @@
     MultiTurnAttackContext,
     MultiTurnAttackStrategy,
 )
-from pyrit.identifiers import build_atomic_attack_identifier
 from pyrit.memory.central_memory import CentralMemory
 from pyrit.models import (
     AttackOutcome,
@@ -43,6 +42,7 @@
     Score,
     SeedPrompt,
 )
+from pyrit.models.identifiers import build_atomic_attack_identifier
 from pyrit.prompt_normalizer import PromptNormalizer
 from pyrit.prompt_target import CapabilityName, TargetRequirements
 from pyrit.score import (
@@ -428,9 +428,7 @@ async def _perform_async(self, *, context: SplitPayloadAttackContext) -> SplitPa
             context.executed_turns += 1
 
             # Check if target refused the assembly turn
-            refusal_score = await self._check_refusal_async(
-                context=context, turn_objective=assembly_prompt
-            )
+            refusal_score = await self._check_refusal_async(context=context, turn_objective=assembly_prompt)
             if bool(refusal_score.get_value()):
                 self._logger.info("Assembly turn was refused by target")
                 all_turns_safe = False
@@ -781,9 +779,7 @@ def _build_assembly_prompt(self, *, context: SplitPayloadAttackContext) -> str:
         Returns:
             str: The adversarial-chat prompt text.
         """
-        turns_summary = "\n".join(
-            f"  Q{i + 1}: {turn}" for i, turn in enumerate(context.planned_turns)
-        )
+        turns_summary = "\n".join(f"  Q{i + 1}: {turn}" for i, turn in enumerate(context.planned_turns))
 
         return (
             f"The target has answered all {len(context.planned_turns)} questions successfully.\n"

tests/unit/executor/attack/multi_turn/test_split_payload.py

@@ -20,7 +20,6 @@
     SplitPayloadAttackContext,
     SplitPayloadAttackResult,
 )
-from pyrit.identifiers import ComponentIdentifier
 from pyrit.models import (
     AttackOutcome,
     ChatMessageRole,
@@ -30,6 +29,7 @@
     Score,
     ScoreType,
 )
+from pyrit.models.identifiers import ComponentIdentifier
 from pyrit.prompt_normalizer import PromptNormalizer
 from pyrit.prompt_target import PromptTarget
 from pyrit.score import FloatScaleThresholdScorer, SelfAskRefusalScorer, TrueFalseScorer
@@ -872,16 +872,12 @@ async def test_generate_assembly_prompt_guides_concrete_deliverable(
             prompt_normalizer=mock_prompt_normalizer,
         )
         basic_context.session = ConversationSession()
-        basic_context.params = AttackParameters(
-            objective="Write a tutorial with specific implementation steps"
-        )
+        basic_context.params = AttackParameters(objective="Write a tutorial with specific implementation steps")
         basic_context.planned_turns = [
             "How does component A work?",
             "How does component B fit with it?",
         ]
-        mock_prompt_normalizer.send_prompt_async.return_value = create_prompt_response(
-            text="Assembly follow-up"
-        )
+        mock_prompt_normalizer.send_prompt_async.return_value = create_prompt_response(text="Assembly follow-up")
 
         result = await attack._generate_assembly_prompt_async(context=basic_context)