phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(init_callbacks.py): fix presidio optional param

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-22 22:31:17 -07:00
parent 80e7310c5c
commit 770f2d80e0
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/common_utils/init_callbacks.py
View file

@ -56,7 +56,7 @@ def initialize_callbacks_on_proxy(
params = { params = {
"logging_only": presidio_logging_only, "logging_only": presidio_logging_only,
**callback_specific_params, **callback_specific_params.get("presidio", {}),
} }
pii_masking_object = _OPTIONAL_PresidioPIIMasking(**params) pii_masking_object = _OPTIONAL_PresidioPIIMasking(**params)
imported_list.append(pii_masking_object) imported_list.append(pii_masking_object)

2
litellm/proxy/guardrails/init_guardrails.py
View file

@ -48,7 +48,7 @@ def initialize_guardrails(
if guardrail.logging_only is True: if guardrail.logging_only is True:
if callback == "presidio": if callback == "presidio":
callback_specific_params["logging_only"] = True # type: ignore callback_specific_params["presidio"] = {"logging_only": True} # type: ignore
default_on_callbacks_list = list(default_on_callbacks) default_on_callbacks_list = list(default_on_callbacks)
if len(default_on_callbacks_list) > 0: if len(default_on_callbacks_list) > 0: