scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-25 05:54:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 125

test(proxy): cover the list-valued text completion prompt at the hook layer

Browse source
This commit is contained in:
Shreyansh Jain 2026-10-14 14:33:46 +05:30
parent a9cd0d2672
commit cf284b7ebd
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
tests/unit/proxy/hooks/test_prompt_injection_detection.py
View file

@ -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}),
],
)