From 75b74244a7790c0a8f3fb20c600511670c229377 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Thu, 16 Apr 2026 00:44:05 +0000
Subject: [PATCH 1/2] Initial plan


From 6c6f81bfaa3fd1da55d297e105f9129a430ddd83 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Thu, 16 Apr 2026 00:47:10 +0000
Subject: [PATCH 2/2] fix: remove type suppression comments from
 MessagePiece.to_message() using TYPE_CHECKING pattern

Agent-Logs-Url: https://github.com/microsoft/PyRIT/sessions/5a4d10d1-e063-41c1-aea3-fba586dacc17

Co-authored-by: spencrr <23708360+spencrr@users.noreply.github.com>
---
 pyrit/models/message_piece.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pyrit/models/message_piece.py b/pyrit/models/message_piece.py
index 91d01032b..59f0de336 100644
--- a/pyrit/models/message_piece.py
+++ b/pyrit/models/message_piece.py
@@ -12,6 +12,7 @@
 from pyrit.models.literals import ChatMessageRole, PromptDataType, PromptResponseError
 
 if TYPE_CHECKING:
+    from pyrit.models.message import Message
     from pyrit.models.score import Score
 
 Originator = Literal["attack", "converter", "undefined", "scorer"]
@@ -226,7 +227,7 @@ def get_role_for_storage(self) -> ChatMessageRole:
         """
         return self._role
 
-    def to_message(self) -> Message:  # type: ignore[name-defined] # noqa: F821
+    def to_message(self) -> Message:
         """
         Convert this message piece into a Message.
 
@@ -235,7 +236,7 @@ def to_message(self) -> Message:  # type: ignore[name-defined] # noqa: F821
         """
         from pyrit.models.message import Message
 
-        return Message([self])  # noqa: F821
+        return Message([self])
 
     def has_error(self) -> bool:
         """