phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

✨ (Feat) Log Guardrails run, guardrail response on logging integrations (#7445)

Browse source 
* add guardrail_information to SLP

* use standard_logging_guardrail_information

* track StandardLoggingGuardrailInformation

* use log_guardrail_information

* use log_guardrail_information

* docs guardrails

* docs guardrails

* update quick start

* fix presidio logging for sync functions

* update Guardrail type

* enforce add_standard_logging_guardrail_information_to_request_data

* update gd docs
This commit is contained in:
Ishaan Jaff 2024-12-27 15:01:56 -08:00 • committed by GitHub
parent 9efb076037
commit 6ec5ed8b3c
14 changed files with 223 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/guardrails/guardrail_hooks/aporia_ai.py
View file

@ -19,7 +19,10 @@ from fastapi import HTTPException
import litellm
from litellm._logging import verbose_proxy_logger
from litellm.integrations.custom_guardrail import CustomGuardrail
from litellm.integrations.custom_guardrail import (
CustomGuardrail,
log_guardrail_information,
)
from litellm.litellm_core_utils.logging_utils import (
convert_litellm_response_object_to_str,
)
@ -142,6 +145,7 @@ class AporiaGuardrail(CustomGuardrail):
},
)
@log_guardrail_information
async def async_post_call_success_hook(
self,
data: dict,
@ -173,6 +177,7 @@ class AporiaGuardrail(CustomGuardrail):
pass
@log_guardrail_information
async def async_moderation_hook( ### 👈 KEY CHANGE ###
self,
data: dict,