phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test_awesome_otel_with_message_logging_off

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-17 16:43:25 +05:30
parent 4554bf760c
commit 202b5cc2cd
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/logging_callback_tests/test_otel_logging.py
View file

@ -200,6 +200,12 @@ def test_completion_claude_3_function_call_with_otel(model):
@pytest.mark.parametrize("streaming", [True, False]) @pytest.mark.parametrize("streaming", [True, False])
@pytest.mark.parametrize("global_redact", [True, False]) @pytest.mark.parametrize("global_redact", [True, False])
async def test_awesome_otel_with_message_logging_off(streaming, global_redact): async def test_awesome_otel_with_message_logging_off(streaming, global_redact):
"""
No content should be logged when message logging is off
tests when litellm.turn_off_message_logging is set to True
tests when OpenTelemetry(message_logging=False) is set
"""
litellm.set_verbose = True litellm.set_verbose = True
litellm.callbacks = [OpenTelemetry(config=OpenTelemetryConfig(exporter=exporter))] litellm.callbacks = [OpenTelemetry(config=OpenTelemetryConfig(exporter=exporter))]
if global_redact is False: if global_redact is False:
@ -256,4 +262,9 @@ def validate_redacted_message_span_attributes(span):
"gen_ai.usage.prompt_tokens", "gen_ai.usage.prompt_tokens",
] ]
_all_attributes = set([name for name in span.attributes.keys()])
print("all_attributes", _all_attributes)
assert _all_attributes == set(expected_attributes)
pass pass