Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions pyrit/prompt_target/openai/openai_chat_target.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@
data_serializer_factory,
)
from pyrit.models.json_response_config import _JsonResponseConfig
from pyrit.prompt_target.common.prompt_target import PromptTarget
from pyrit.prompt_target.common.target_capabilities import TargetCapabilities
from pyrit.prompt_target.common.target_configuration import TargetConfiguration
from pyrit.prompt_target.common.utils import limit_requests_per_minute, validate_temperature, validate_top_p
Expand All @@ -33,7 +32,7 @@
logger = logging.getLogger(__name__)


class OpenAIChatTarget(OpenAITarget, PromptTarget):
class OpenAIChatTarget(OpenAITarget):
"""
Facilitates multimodal (image and text) input and text output generation.

Expand Down
3 changes: 1 addition & 2 deletions pyrit/prompt_target/openai/openai_realtime_target.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@
construct_response_from_request,
data_serializer_factory,
)
from pyrit.prompt_target.common.prompt_target import PromptTarget
from pyrit.prompt_target.common.target_capabilities import TargetCapabilities
from pyrit.prompt_target.common.target_configuration import TargetConfiguration
from pyrit.prompt_target.common.utils import limit_requests_per_minute
Expand Down Expand Up @@ -58,7 +57,7 @@ def flatten_transcripts(self) -> str:
return "".join(self.transcripts)


class RealtimeTarget(OpenAITarget, PromptTarget):
class RealtimeTarget(OpenAITarget):
"""
A prompt target for Azure OpenAI Realtime API.

Expand Down
3 changes: 1 addition & 2 deletions pyrit/prompt_target/openai/openai_response_target.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,6 @@
PromptResponseError,
)
from pyrit.models.json_response_config import _JsonResponseConfig
from pyrit.prompt_target.common.prompt_target import PromptTarget
from pyrit.prompt_target.common.target_capabilities import TargetCapabilities
from pyrit.prompt_target.common.target_configuration import TargetConfiguration
from pyrit.prompt_target.common.utils import limit_requests_per_minute, validate_temperature, validate_top_p
Expand Down Expand Up @@ -59,7 +58,7 @@ class MessagePieceType(str, Enum):
MCP_APPROVAL_REQUEST = "mcp_approval_request"


class OpenAIResponseTarget(OpenAITarget, PromptTarget):
class OpenAIResponseTarget(OpenAITarget):
"""
Enables communication with endpoints that support the OpenAI Response API.

Expand Down
Loading