BUG Fix for responses API (microsoft#1156)

Co-authored-by: Richard Lundeen <rlundeen@microsoft.com>

Author: riedgar-ms

.env_example

@@ -54,6 +54,11 @@ OPENAI_CHAT_MODEL=${PLATFORM_OPENAI_CHAT_GPT4O_MODEL}
 # OPENAI RESPONSES TARGET SECRETS
 ##################################
 
+AZURE_OPENAI_GPT5_RESPONSES_ENDPOINT="https://xxxxxxxxx.azure.com/openai/deployments/gpt-5/responses"
+AZURE_OPENAI_GPT5_COMPLETION_ENDPOINT="https://xxxxxxxxx.azure.com/openai/deployments/gpt-5/chat/completions?api-version=2025-01-01-preview"
+AZURE_OPENAI_GPT5_KEY="xxxxxxx"
+AZURE_OPENAI_GPT5_MODEL="gpt-5"
+
 PLATFORM_OPENAI_RESPONSES_ENDPOINT="https://api.openai.com/v1/responses"
 PLATFORM_OPENAI_RESPONSES_KEY="sk-xxxxx"
 PLATFORM_OPENAI_RESPONSES_MODEL="o4-mini"

pyrit/prompt_target/openai/openai_response_target.py

@@ -188,7 +188,10 @@ async def _construct_input_item_from_piece(self, piece: MessagePiece) -> Dict[st
                 image paths.
         """
         if piece.converted_value_data_type == "text":
-            return {"type": "input_text" if piece.role == "user" else "output_text", "text": piece.converted_value}
+            return {
+                "type": "input_text" if piece.role in ["developer", "user"] else "output_text",
+                "text": piece.converted_value,
+            }
         if piece.converted_value_data_type == "image_path":
             data_url = await convert_local_image_to_data_url(piece.converted_value)
             return {"type": "input_image", "image_url": {"url": data_url}}

tests/integration/targets/test_openai_responses_gpt5.py

@@ -0,0 +1,48 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+
+import os
+import uuid
+
+import pytest
+
+from pyrit.models import MessagePiece
+from pyrit.prompt_target import OpenAIResponseTarget
+
+
+@pytest.mark.asyncio
+async def test_openai_responses_gpt5(sqlite_instance):
+    args = {
+        "endpoint": os.getenv("AZURE_OPENAI_GPT5_RESPONSES_ENDPOINT"),
+        "model_name": os.getenv("AZURE_OPENAI_GPT5_MODEL"),
+        "api_key": os.getenv("AZURE_OPENAI_GPT5_RESPONSES_KEY"),
+    }
+
+    target = OpenAIResponseTarget(**args)
+
+    conv_id = str(uuid.uuid4())
+
+    developer_piece = MessagePiece(
+        role="developer",
+        original_value="You are a helpful assistant.",
+        original_value_data_type="text",
+        conversation_id=conv_id,
+        attack_identifier={"id": str(uuid.uuid4())},
+    )
+    sqlite_instance.add_message_to_memory(request=developer_piece.to_message())
+
+    user_piece = MessagePiece(
+        role="user",
+        original_value="What is the capital of France?",
+        original_value_data_type="text",
+        conversation_id=conv_id,
+    )
+
+    result = await target.send_prompt_async(prompt_request=user_piece.to_message())
+    assert result is not None
+    assert len(result.message_pieces) == 2
+    assert result.message_pieces[0].role == "assistant"
+    assert result.message_pieces[1].role == "assistant"
+    # Hope that the model manages to give the correct answer somewhere (GPT-5 really should)
+    assert "Paris" in result.message_pieces[1].converted_value