test(test_presidio_pii_masking.py): fix presidio test

2024-07-13 21:44:22 -07:00 · 2024-07-13 21:44:22 -07:00 · d475311eb3
commit d475311eb3
parent da4bd47e3e
1 changed files with 3 additions and 0 deletions
--- a/litellm/tests/test_presidio_masking.py
+++ b/litellm/tests/test_presidio_masking.py
@ -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": {