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

fix otel logging test

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-09 17:51:35 -07:00
parent 407bdf10ce
commit 16b6b56c8b
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_async_opentelemetry.py
View file

@ -18,7 +18,7 @@ class TestOpenTelemetry(OpenTelemetry):
self.kwargs = None self.kwargs = None
async def async_log_success_event(self, kwargs, response_obj, start_time, end_time): async def async_log_success_event(self, kwargs, response_obj, start_time, end_time):
print("in async_log_success_event for TestOpenTelemetry kwargs=", self.kwargs) print("in async_log_success_event for TestOpenTelemetry kwargs=", kwargs)
self.kwargs = kwargs self.kwargs = kwargs
await super().async_log_success_event( await super().async_log_success_event(
kwargs, response_obj, start_time, end_time kwargs, response_obj, start_time, end_time
@ -26,9 +26,8 @@ class TestOpenTelemetry(OpenTelemetry):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_otel_with_message_logging_off(): async def test_awesome_otel_with_message_logging_off():
litellm.set_verbose = True litellm.set_verbose = True
from litellm.integrations.opentelemetry import OpenTelemetry
otel_logger = TestOpenTelemetry( otel_logger = TestOpenTelemetry(
message_logging=False, config=OpenTelemetryConfig(exporter="console") message_logging=False, config=OpenTelemetryConfig(exporter="console")