phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(presidio_pii_masking.py): fix conditional check

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-13 22:11:03 -08:00
parent f68b656040
commit 6b91f48c64
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_presidio_masking.py
View file

@ -53,7 +53,7 @@ async def test_output_parsing():
mock_response="Hello <PERSON>! How can I assist you today?",
)
new_response = await pii_masking.async_post_call_success_hook(
user_api_key_dict=UserAPIKeyAuth(), response=response, call_type="completion"
user_api_key_dict=UserAPIKeyAuth(), response=response
)
assert (