mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-05 08:07:05 +00:00
updates in memory custom guardrail liteLLM params
This commit is contained in:
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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue