From 940642d39f92e823c314e75ab377ea542a3ff00e Mon Sep 17 00:00:00 2001
From: Utkarsh Bahuguna <utkarshbahuguna10@gmail.com>
Date: Sun, 11 Oct 2026 00:44:20 +0530
Subject: [PATCH 1/2] label the sequential attack result

---
 pyrit/executor/attack/compound/sequential_attack.py  |  1 +
 .../attack/compound/test_sequential_attack.py        | 12 ++++++++++++
 2 files changed, 13 insertions(+)

diff --git a/pyrit/executor/attack/compound/sequential_attack.py b/pyrit/executor/attack/compound/sequential_attack.py
index 51be7926b4..d7d9d94f18 100644
--- a/pyrit/executor/attack/compound/sequential_attack.py
+++ b/pyrit/executor/attack/compound/sequential_attack.py
@@ -290,6 +290,7 @@ async def _perform_async(self, *, context: AttackContext[AttackParameters]) -> S
             automated_score=None,
             executed_turns=sum(r.executed_turns for r in results),
             outcome=outcome,
+            labels=context.memory_labels,
             child_attack_results=results,
             completion_policy=self._completion_policy,
             metadata={
diff --git a/tests/unit/executor/attack/compound/test_sequential_attack.py b/tests/unit/executor/attack/compound/test_sequential_attack.py
index be8afd7d77..49dc64fdec 100644
--- a/tests/unit/executor/attack/compound/test_sequential_attack.py
+++ b/tests/unit/executor/attack/compound/test_sequential_attack.py
@@ -489,6 +489,18 @@ async def test_context_labels_passed_through(self, target, seed_group):
 
         assert calls[0]["memory_labels"]["foo"] == "bar"
 
+    async def test_completed_result_carries_context_labels(self, target, seed_group):
+        a = _make_strategy(outcomes=[AttackOutcome.FAILURE], name="a")
+        compound = SequentialAttack(
+            objective_target=target, child_attacks=[SequentialChildAttack(strategy=a, seed_group=seed_group)]
+        )
+        patcher, _ = _patch_run_child_attack(strategies_by_id={id(a): a})
+
+        with patcher:
+            result = await compound._perform_async(context=_make_context(labels={"op": "seq"}))
+
+        assert result.labels == {"op": "seq"}
+
     async def test_child_attack_labels_override_context_labels(self, target, seed_group):
         a = _make_strategy(outcomes=[AttackOutcome.SUCCESS], name="a")
         child_attacks = [

From 90696316166cd3a406773a556b970e0356a014e9 Mon Sep 17 00:00:00 2001
From: Utkarsh Bahuguna <utkarshbahuguna10@gmail.com>
Date: Sun, 11 Oct 2026 10:43:15 +0530
Subject: [PATCH 2/2] tighten tests: async names, typed args, public paths

---
 .../attack/compound/test_sequential_attack.py | 24 ++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/tests/unit/executor/attack/compound/test_sequential_attack.py b/tests/unit/executor/attack/compound/test_sequential_attack.py
index 49dc64fdec..10f4d691e3 100644
--- a/tests/unit/executor/attack/compound/test_sequential_attack.py
+++ b/tests/unit/executor/attack/compound/test_sequential_attack.py
@@ -489,7 +489,9 @@ async def test_context_labels_passed_through(self, target, seed_group):
 
         assert calls[0]["memory_labels"]["foo"] == "bar"
 
-    async def test_completed_result_carries_context_labels(self, target, seed_group):
+    async def test_completed_result_carries_context_labels_async(
+        self, *, target: MagicMock, seed_group: AttackSeedGroup
+    ) -> None:
         a = _make_strategy(outcomes=[AttackOutcome.FAILURE], name="a")
         compound = SequentialAttack(
             objective_target=target, child_attacks=[SequentialChildAttack(strategy=a, seed_group=seed_group)]
@@ -501,6 +503,26 @@ async def test_completed_result_carries_context_labels(self, target, seed_group)
 
         assert result.labels == {"op": "seq"}
 
+    async def test_label_query_finds_the_completed_result_async(
+        self, *, seed_group: AttackSeedGroup, sqlite_instance: MemoryInterface
+    ) -> None:
+        target = MockPromptTarget()
+        compound = SequentialAttack(
+            objective_target=target,
+            child_attacks=[
+                SequentialChildAttack(strategy=_OkChildStrategy(objective_target=target), seed_group=seed_group)
+            ],
+        )
+
+        result = await compound.execute_async(objective="obj", memory_labels={"op": "seq"})
+
+        labelled = await sqlite_instance.get_attack_results_async(labels={"op": "seq"})
+        assert result.attack_result_id in {row.attack_result_id for row in labelled}
+        assert {row.attack_result_id for row in labelled} == {
+            result.attack_result_id,
+            *result.metadata[SequentialAttack.CHILD_ATTACK_RESULT_IDS_KEY],
+        }
+
     async def test_child_attack_labels_override_context_labels(self, target, seed_group):
         a = _make_strategy(outcomes=[AttackOutcome.SUCCESS], name="a")
         child_attacks = [