scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test_guardrail_status_fields_computation

Browse source
This commit is contained in:
Ishaan Jaffer 2025-11-06 16:56:50 -08:00
parent e6d7a0a153
commit 0a8ae52014
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/guardrails_tests/test_tracing_guardrails.py
View file

@ -605,7 +605,8 @@ def test_guardrail_status_fields_computation():
from litellm.litellm_core_utils.litellm_logging import _get_status_fields
# Test guardrail_intervened status (content was blocked by guardrail)
intervened_info = {"guardrail_status": "guardrail_intervened"}
# guardrail_information is now a list
intervened_info = [{"guardrail_status": "guardrail_intervened"}]
status_fields_intervened = _get_status_fields(
status="success",
guardrail_information=intervened_info,
@ -615,7 +616,7 @@ def test_guardrail_status_fields_computation():
assert status_fields_intervened["guardrail_status"] == "guardrail_intervened"
# Test legacy blocked status (for backward compatibility)
blocked_info = {"guardrail_status": "blocked"}
blocked_info = [{"guardrail_status": "blocked"}]
status_fields_blocked = _get_status_fields(
status="success",
guardrail_information=blocked_info,
@ -625,7 +626,7 @@ def test_guardrail_status_fields_computation():
assert status_fields_blocked["guardrail_status"] == "guardrail_intervened"
# Test success status
success_info = {"guardrail_status": "success"}
success_info = [{"guardrail_status": "success"}]
status_fields_success = _get_status_fields(
status="success",
guardrail_information=success_info,
@ -635,7 +636,7 @@ def test_guardrail_status_fields_computation():
assert status_fields_success["guardrail_status"] == "success"
# Test guardrail_failed_to_respond status
failed_info = {"guardrail_status": "guardrail_failed_to_respond"}
failed_info = [{"guardrail_status": "guardrail_failed_to_respond"}]
status_fields_failed = _get_status_fields(
status="failure",
guardrail_information=failed_info,
@ -645,7 +646,7 @@ def test_guardrail_status_fields_computation():
assert status_fields_failed["guardrail_status"] == "guardrail_failed_to_respond"
# Test legacy failure status (for backward compatibility)
failure_info = {"guardrail_status": "failure"}
failure_info = [{"guardrail_status": "failure"}]
status_fields_failure = _get_status_fields(
status="failure",
guardrail_information=failure_info,