scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

content filter test fix

Browse source
This commit is contained in:
yuneng-jiang 2026-02-12 15:31:19 -08:00
parent 1f61794c12
commit 7021da85d2
2 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/guardrails/guardrail_hooks/litellm_content_filter/content_filter.py
View file

@ -29,10 +29,7 @@ from fastapi import HTTPException
from litellm import Router
from litellm._logging import verbose_proxy_logger
from litellm.integrations.custom_guardrail import (
CustomGuardrail,
log_guardrail_information,
)
from litellm.integrations.custom_guardrail import CustomGuardrail
from litellm.proxy._types import UserAPIKeyAuth
from litellm.types.utils import ModelResponseStream
@ -1056,7 +1053,6 @@ class ContentFilterGuardrail(CustomGuardrail):
masked_entity_count=masked_entity_count,
)
@log_guardrail_information
async def apply_guardrail(
self,
inputs: "GenericGuardrailAPIInputs",

6
tests/code_coverage_tests/check_guardrail_apply_decorator.py
View file

@ -83,6 +83,12 @@ def test_guardrail_apply_decorator():
if python_file.name == "bedrock_guardrails.py":
continue
# Skip content_filter.py - it implements its own detailed logging via
# _log_guardrail_information with detections, masked_entity_count, etc.
# Using the decorator would cause duplicate entries.
if python_file.name == "content_filter.py":
continue
results = find_apply_guardrail_methods(python_file)
for class_name, line_num, has_decorator in results: