From b620b853a4480b25d34b4a22f65f8f193b32105f Mon Sep 17 00:00:00 2001 From: hannahwestra25 Date: Thu, 19 Mar 2026 15:14:02 -0400 Subject: [PATCH] fix missing custom_capabilities in integration tests --- .../targets/test_openai_chat_target_integration.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/integration/targets/test_openai_chat_target_integration.py b/tests/integration/targets/test_openai_chat_target_integration.py index ae7c98919..6e0bf7a4a 100644 --- a/tests/integration/targets/test_openai_chat_target_integration.py +++ b/tests/integration/targets/test_openai_chat_target_integration.py @@ -18,6 +18,7 @@ from pyrit.common.path import HOME_PATH from pyrit.models import MessagePiece from pyrit.prompt_target import OpenAIChatAudioConfig, OpenAIChatTarget +from pyrit.prompt_target.common.target_capabilities import TargetCapabilities # Path to sample audio file for testing SAMPLE_AUDIO_FILE = HOME_PATH / "assets" / "converted_audio.wav" @@ -87,6 +88,11 @@ async def test_openai_chat_target_audio_multi_turn(sqlite_instance, platform_ope target = OpenAIChatTarget( **platform_openai_audio_args, audio_response_config=audio_config, + custom_capabilities=TargetCapabilities( + input_modalities=frozenset( + {frozenset({"text", "audio_path"}), frozenset({"text"}), frozenset({"audio_path"})} + ), + ), ) conv_id = str(uuid.uuid4())