From d475311eb30f65dbbcfc560703577103a1d06522 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 13 Jul 2024 21:44:22 -0700
Subject: [PATCH] test(test_presidio_pii_masking.py): fix presidio test

---
 litellm/tests/test_presidio_masking.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/litellm/tests/test_presidio_masking.py b/litellm/tests/test_presidio_masking.py
index 2885c07f34b..193fcf113b7 100644
--- 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": {