scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

[Bug Fix] Content Filter Guard (#16414)

Browse source 
* fix add_guardrail_to_db

* bump commit
This commit is contained in:
Ishaan Jaff 2025-11-08 16:30:05 -08:00 • committed by GitHub
parent 6759ea0198
commit b242567778
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 15 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/__init__.py
View file

@ -34,7 +34,7 @@ from litellm.types.utils import (
all_litellm_params as _litellm_completion_params,
CredentialItem,
PriorityReservationDict,
) # maintain backwards compatibility for root param
) # maintain backwards compatibility for root param.
from litellm._logging import (
set_verbose,
_turn_on_debug,

16
litellm/proxy/guardrails/guardrail_registry.py
View file

@ -232,7 +232,13 @@ class GuardrailRegistry:
"""
try:
guardrail_name = guardrail.get("guardrail_name")
litellm_params: str = safe_dumps(dict(guardrail.get("litellm_params", {})))
# Properly serialize LitellmParams Pydantic model to dict
litellm_params_obj = guardrail.get("litellm_params", {})
if hasattr(litellm_params_obj, 'model_dump'):
litellm_params_dict = litellm_params_obj.model_dump()
else:
litellm_params_dict = dict(litellm_params_obj) if litellm_params_obj else {}
litellm_params: str = safe_dumps(litellm_params_dict)
guardrail_info: str = safe_dumps(guardrail.get("guardrail_info", {}))
# Create guardrail in DB
@ -278,7 +284,13 @@ class GuardrailRegistry:
"""
try:
guardrail_name = guardrail.get("guardrail_name")
litellm_params: str = safe_dumps(dict(guardrail.get("litellm_params", {})))
# Properly serialize LitellmParams Pydantic model to dict
litellm_params_obj = guardrail.get("litellm_params", {})
if hasattr(litellm_params_obj, 'model_dump'):
litellm_params_dict = litellm_params_obj.model_dump()
else:
litellm_params_dict = dict(litellm_params_obj) if litellm_params_obj else {}
litellm_params: str = safe_dumps(litellm_params_dict)
guardrail_info: str = safe_dumps(guardrail.get("guardrail_info", {}))
# Update in DB