scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test(test_presidio_pii_masking.py): fix presidio test

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-13 21:44:22 -07:00
parent da4bd47e3e
commit d475311eb3
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_presidio_masking.py
View file

@ -273,6 +273,9 @@ async def test_presidio_pii_masking_logging_output_only_logged_response_guardrai
from litellm.proxy.guardrails.init_guardrails import initialize_guardrails
from litellm.types.guardrails import GuardrailItem, GuardrailItemSpec
os.environ["PRESIDIO_ANALYZER_API_BASE"] = "http://localhost:5002"
os.environ["PRESIDIO_ANONYMIZER_API_BASE"] = "http://localhost:5001"
guardrails_config: List[Dict[str, GuardrailItemSpec]] = [
{
"pii_masking": {