diff --git a/tests/unit/proxy/hooks/test_prompt_injection_detection.py b/tests/unit/proxy/hooks/test_prompt_injection_detection.py index 6df1f6f8cf8..e68f2e20fa7 100644 --- a/tests/unit/proxy/hooks/test_prompt_injection_detection.py +++ b/tests/unit/proxy/hooks/test_prompt_injection_detection.py @@ -107,6 +107,7 @@ async def test_acompletion_call_type_allows_safe_prompt(): ("atranscription", {"model": "test-model", "prompt": INJECTION_ATTEMPT}), ("transcription", {"model": "test-model", "prompt": INJECTION_ATTEMPT}), ("atext_completion", {"model": "test-model", "prompt": INJECTION_ATTEMPT}), + ("atext_completion", {"model": "test-model", "prompt": [INJECTION_ATTEMPT]}), ("amoderation", {"model": "test-model", "input": INJECTION_ATTEMPT}), ], )