phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix test otel logger

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-08 18:24:00 -07:00
parent 62d3141103
commit bcfe62b801
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_opentelemetry.py → litellm/tests/test_async_opentelemetry.py
View file

@ -12,19 +12,20 @@ verbose_logger.setLevel(logging.DEBUG)
# @pytest.mark.skip(reason="new test")
def test_otel_callback():
@pytest.mark.asyncio
async def test_otel_callback():
exporter = InMemorySpanExporter()
litellm.set_verbose = True
litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))]
litellm.completion(
await litellm.acompletion(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "hi"}],
temperature=0.1,
user="OTEL_USER",
)
time.sleep(4)
await asyncio.sleep(4)
spans = exporter.get_finished_spans()
print("spans", spans)