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

feat - allow accessing data post success call

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-19 11:35:33 -07:00
parent 6af497e383
commit b4bca8db82
12 changed files with 71 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_presidio_masking.py
View file

@ -88,7 +88,11 @@ 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
user_api_key_dict=UserAPIKeyAuth(),
data={
"messages": [{"role": "system", "content": "You are an helpfull assistant"}]
},
response=response,
)
assert (