Skip to content

Commit 66202d0

Browse files
committed
updates in memory custom guardrail liteLLM params
1 parent 895c41e commit 66202d0

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

litellm/integrations/custom_guardrail.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -487,7 +487,8 @@ def update_in_memory_litellm_params(self, litellm_params: LitellmParams) -> None
487487
"""
488488
Update the guardrails litellm params in memory
489489
"""
490-
pass
490+
for key, value in vars(litellm_params).items():
491+
setattr(self, key, value)
491492

492493

493494
def log_guardrail_information(func):

litellm/proxy/guardrails/guardrail_hooks/presidio.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -675,5 +675,6 @@ def update_in_memory_litellm_params(self, litellm_params: LitellmParams) -> None
675675
"""
676676
Update the guardrails litellm params in memory
677677
"""
678+
super().update_in_memory_litellm_params(litellm_params)
678679
if litellm_params.pii_entities_config:
679680
self.pii_entities_config = litellm_params.pii_entities_config

0 commit comments

Comments
 (0)