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 2d618087bf
commit 03dcf366a1
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") # @pytest.mark.skip(reason="new test")
def test_otel_callback(): @pytest.mark.asyncio
async def test_otel_callback():
exporter = InMemorySpanExporter() exporter = InMemorySpanExporter()
litellm.set_verbose = True litellm.set_verbose = True
litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))] litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))]
litellm.completion( await litellm.acompletion(
model="gpt-3.5-turbo", model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "hi"}], messages=[{"role": "user", "content": "hi"}],
temperature=0.1, temperature=0.1,
user="OTEL_USER", user="OTEL_USER",
) )
time.sleep(4) await asyncio.sleep(4)
spans = exporter.get_finished_spans() spans = exporter.get_finished_spans()
print("spans", spans) print("spans", spans)