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

test(otel): add regression for list guardrail_mode causing unhashable TypeError

Browse source
This commit is contained in:
devteamaegis 2026-05-22 01:52:37 -04:00
parent 3a1bb525a4
commit 0f38d14350
1 changed files with 32 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
tests/test_litellm/integrations/test_opentelemetry.py
View file

@ -87,6 +87,38 @@ class TestOpenTelemetryGuardrails(unittest.TestCase):
# Verify that start_span was never called
otel.tracer.start_span.assert_not_called()
@patch("litellm.integrations.opentelemetry.datetime")
def test_guardrail_mode_as_list_does_not_crash(self, mock_datetime):
"""Regression test for https://github.com/BerriAI/litellm/issues/28486.
When a guardrail is configured with a list ``mode`` (e.g.
``["pre_call", "post_call"]``), ``_create_guardrail_span`` previously
crashed with ``TypeError: unhashable type: 'list'`` inside
``_emit_once``, because the list value was unpacked directly into the
tuple used as a ``dict`` key. Lists are not hashable; only tuples are.
The fix normalises each scope part in ``_emit_once`` to a tuple when
it is a list, so the dedupe key is always hashable.
"""
otel = OpenTelemetry()
otel.tracer = MagicMock()
mock_span = MagicMock()
otel.tracer.start_span.return_value = mock_span
guardrail_info = {
"guardrail_name": "test_guardrail",
"guardrail_mode": ["pre_call", "post_call"], # list-valued mode
"start_time": 1609459200.0,
"end_time": 1609459201.0,
}
kwargs = {
"standard_logging_object": {"guardrail_information": [guardrail_info]}
}
# Must not raise TypeError: unhashable type: 'list'
otel._create_guardrail_span(kwargs=kwargs, context=None)
otel.tracer.start_span.assert_called_once()
@patch("litellm.integrations.opentelemetry.datetime")
def test_guardrail_response_dict_is_json_serialized(self, mock_datetime):
"""Dict guardrail_response (e.g. OpenAI moderation result) must reach