tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-05 18:24:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

[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
2 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/__init__.py
+1 -1
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,
litellm/proxy/guardrails/guardrail_registry.py
+14 -2
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