From 9f76933201cc6d5e78e6b6f1c9607ff3078a4444 Mon Sep 17 00:00:00 2001
From: spencrr <23708360+spencrr@users.noreply.github.com>
Date: Mon, 20 Apr 2026 16:51:22 -0700
Subject: [PATCH] [MAINT]: Remove unnecessary cast after PyRIT bumped

---
 rampart/_pyrit/llm_bridge.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/rampart/_pyrit/llm_bridge.py b/rampart/_pyrit/llm_bridge.py
index 4de0d1b..1a8b292 100644
--- a/rampart/_pyrit/llm_bridge.py
+++ b/rampart/_pyrit/llm_bridge.py
@@ -12,10 +12,10 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, cast
+from typing import TYPE_CHECKING, Any
 from uuid import uuid4
 
-from pyrit.models import Message, MessagePiece
+from pyrit.models import MessagePiece
 from pyrit.prompt_target import OpenAIChatTarget, PromptChatTarget
 
 if TYPE_CHECKING:
@@ -144,8 +144,7 @@ async def send_generation_request_async(
         conversation_id=conversation_id,
     )
 
-    # Can remove after bumping to PyRIT v0.13.0
-    request = cast("Message", request_piece.to_message())  # pyright: ignore[reportUnknownMemberType]
+    request = request_piece.to_message()
 
     responses = await target.send_prompt_async(message=request)
     return responses[0].get_value()