From 770f2d80e0ea9dfe2dd2c504af972d97055ce095 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 22 Jul 2024 22:31:17 -0700 Subject: [PATCH] fix(init_callbacks.py): fix presidio optional param --- litellm/proxy/common_utils/init_callbacks.py | 2 +- litellm/proxy/guardrails/init_guardrails.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/common_utils/init_callbacks.py b/litellm/proxy/common_utils/init_callbacks.py index bd52efb19..10a76149f 100644 --- a/litellm/proxy/common_utils/init_callbacks.py +++ b/litellm/proxy/common_utils/init_callbacks.py @@ -56,7 +56,7 @@ def initialize_callbacks_on_proxy( params = { "logging_only": presidio_logging_only, - **callback_specific_params, + **callback_specific_params.get("presidio", {}), } pii_masking_object = _OPTIONAL_PresidioPIIMasking(**params) imported_list.append(pii_masking_object) diff --git a/litellm/proxy/guardrails/init_guardrails.py b/litellm/proxy/guardrails/init_guardrails.py index e98beb817..de6181868 100644 --- a/litellm/proxy/guardrails/init_guardrails.py +++ b/litellm/proxy/guardrails/init_guardrails.py @@ -48,7 +48,7 @@ def initialize_guardrails( if guardrail.logging_only is True: 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) if len(default_on_callbacks_list) > 0: