updates in memory custom guardrail liteLLM params

2026-08-17 10:26:07 +00:00 · 2025-09-17 16:28:05 -07:00
parent 895c41efa3
commit 66202d0ab7
2 changed files with 3 additions and 1 deletions
@@ -487,7 +487,8 @@ class CustomGuardrail(CustomLogger):
        """
        Update the guardrails litellm params in memory
        """
-        pass
+        for key, value in vars(litellm_params).items():
+            setattr(self, key, value)


 def log_guardrail_information(func):
@@ -675,5 +675,6 @@ class _OPTIONAL_PresidioPIIMasking(CustomGuardrail):
        """
        Update the guardrails litellm params in memory
        """
+        super().update_in_memory_litellm_params(litellm_params)
        if litellm_params.pii_entities_config:
            self.pii_entities_config = litellm_params.pii_entities_config