From 2d618087bf9b19867b3bca77da21927878c02273 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 8 Jun 2024 15:24:14 -0700 Subject: [PATCH] test_otel_callback --- litellm/tests/test_opentelemetry.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/litellm/tests/test_opentelemetry.py b/litellm/tests/test_opentelemetry.py index ebeecf254ac..92d7b851531 100644 --- a/litellm/tests/test_opentelemetry.py +++ b/litellm/tests/test_opentelemetry.py @@ -17,13 +17,11 @@ def test_otel_callback(): litellm.set_verbose = True litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))] - asyncio.run( - litellm.acompletion( - model="gpt-3.5-turbo", - messages=[{"role": "user", "content": "hi"}], - temperature=0.1, - user="OTEL_USER", - ) + litellm.completion( + model="gpt-3.5-turbo", + messages=[{"role": "user", "content": "hi"}], + temperature=0.1, + user="OTEL_USER", ) time.sleep(4)