diff --git a/tests/integration/targets/test_entra_auth_targets.py b/tests/integration/targets/test_entra_auth_targets.py index 82dd17793..04df3afdf 100644 --- a/tests/integration/targets/test_entra_auth_targets.py +++ b/tests/integration/targets/test_entra_auth_targets.py @@ -262,11 +262,11 @@ async def test_openai_realtime_target_entra_auth(sqlite_instance, endpoint, mode @pytest.mark.asyncio async def test_video_target_entra_auth(sqlite_instance): # Takes a long time and sometimes encounters retry errors. - # Note: OPENAI_VIDEO_ENDPOINT should be configured for Sora v2 API - endpoint = os.environ["OPENAI_VIDEO2_ENDPOINT"] + # Note: AZURE_OPENAI_VIDEO_ENDPOINT should be configured for Sora v2 API + endpoint = os.environ["AZURE_OPENAI_VIDEO_ENDPOINT"] target = OpenAIVideoTarget( endpoint=endpoint, - model_name=os.environ["OPENAI_VIDEO2_MODEL"], + model_name=os.environ["AZURE_OPENAI_VIDEO_MODEL"], api_key=get_azure_openai_auth(endpoint), ) attack = PromptSendingAttack(objective_target=target) diff --git a/tests/integration/targets/test_target_filters.py b/tests/integration/targets/test_target_filters.py index 0510ffdfd..dae5ce8ed 100644 --- a/tests/integration/targets/test_target_filters.py +++ b/tests/integration/targets/test_target_filters.py @@ -121,7 +121,7 @@ async def test_image_input_filters(sqlite_instance, endpoint, api_key, model_nam @pytest.mark.asyncio @pytest.mark.parametrize( ("endpoint", "api_key", "model_name"), - [("OPENAI_VIDEO2_ENDPOINT", "OPENAI_VIDEO2_KEY", "OPENAI_VIDEO2_MODEL")], + [("AZURE_OPENAI_VIDEO_ENDPOINT", "AZURE_OPENAI_VIDEO_KEY", "AZURE_OPENAI_VIDEO_MODEL")], ) async def test_video_input_filters(sqlite_instance, endpoint, api_key, model_name): target = OpenAIVideoTarget( diff --git a/tests/integration/targets/test_targets_and_secrets.py b/tests/integration/targets/test_targets_and_secrets.py index 31a3a9851..bb8b2c0b5 100644 --- a/tests/integration/targets/test_targets_and_secrets.py +++ b/tests/integration/targets/test_targets_and_secrets.py @@ -460,7 +460,7 @@ async def test_connect_tts(sqlite_instance, endpoint, api_key, model_name): @pytest.mark.parametrize( ("endpoint", "api_key", "model_name"), [ - ("OPENAI_VIDEO2_ENDPOINT", "OPENAI_VIDEO2_KEY", "OPENAI_VIDEO2_MODEL"), + ("AZURE_OPENAI_VIDEO_ENDPOINT", "AZURE_OPENAI_VIDEO_KEY", "AZURE_OPENAI_VIDEO_MODEL"), # OpenAI Platform endpoint returns HTTP 401 "Missing scopes: api.videos.write" for all requests # ("PLATFORM_OPENAI_VIDEO_ENDPOINT", "PLATFORM_OPENAI_VIDEO_KEY", "PLATFORM_OPENAI_VIDEO_MODEL"), ], @@ -491,9 +491,9 @@ async def test_video_multiple_prompts_create_separate_files(sqlite_instance): This verifies that each video generation creates a unique file based on the video ID mechanism. """ - endpoint_value = _get_required_env_var("OPENAI_VIDEO2_ENDPOINT") - api_key_value = _get_required_env_var("OPENAI_VIDEO2_KEY") - model_name_value = _get_required_env_var("OPENAI_VIDEO2_MODEL") + endpoint_value = _get_required_env_var("AZURE_OPENAI_VIDEO_ENDPOINT") + api_key_value = _get_required_env_var("AZURE_OPENAI_VIDEO_KEY") + model_name_value = _get_required_env_var("AZURE_OPENAI_VIDEO_MODEL") target = OpenAIVideoTarget( endpoint=endpoint_value,