phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix init presidio guardrail

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-04 15:22:37 -07:00
parent 4ab8e52bfa
commit 7712aa652a
2 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/litellm_core_utils/litellm_logging.py
View file

@ -1361,12 +1361,13 @@ class Logging:
) )
is not True is not True
): ):
self.model_call_details, result = await callback.async_logging_hook(
kwargs=self.model_call_details,
result=result,
call_type=self.call_type,
)
continue continue
self.model_call_details, result = await callback.async_logging_hook(
kwargs=self.model_call_details,
result=result,
call_type=self.call_type,
)
elif isinstance(callback, CustomLogger): elif isinstance(callback, CustomLogger):
self.model_call_details, result = await callback.async_logging_hook( self.model_call_details, result = await callback.async_logging_hook(
kwargs=self.model_call_details, kwargs=self.model_call_details,

1
litellm/proxy/proxy_config.yaml
View file

@ -13,6 +13,7 @@ guardrails:
litellm_params: litellm_params:
guardrail: presidio # supported values: "aporia", "lakera", "presidio" guardrail: presidio # supported values: "aporia", "lakera", "presidio"
mode: "pre_call" # pre_call, during_call, post_call mode: "pre_call" # pre_call, during_call, post_call
output_parse_pii: True
general_settings: general_settings:
master_key: sk-1234 master_key: sk-1234