From 35ecf3c06b7591b7de61af3603df658130fa8476 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 27 Jul 2024 15:30:58 -0700 Subject: [PATCH] fix otel test --- tests/otel_tests/test_otel.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/otel_tests/test_otel.py b/tests/otel_tests/test_otel.py index 1c38e8125..d0d312128 100644 --- a/tests/otel_tests/test_otel.py +++ b/tests/otel_tests/test_otel.py @@ -97,6 +97,8 @@ async def test_chat_completion_check_otel_spans(): key = key_gen["key"] await chat_completion(session=session, key=key, model="fake-openai-endpoint") + await asyncio.sleep(3) + otel_spans = await get_otel_spans(session=session, key=key) print("otel_spans: ", otel_spans) @@ -108,8 +110,8 @@ async def test_chat_completion_check_otel_spans(): print("Parent trace spans: ", parent_trace_spans) - # either 4 or 5 traces depending on how many redis calls were made - assert len(parent_trace_spans) == 5 or len(parent_trace_spans) == 4 + # either 5 or 6 traces depending on how many redis calls were made + assert len(parent_trace_spans) == 6 or len(parent_trace_spans) == 5 # 'postgres', 'redis', 'raw_gen_ai_request', 'litellm_request', 'Received Proxy Server Request' in the span assert "postgres" in parent_trace_spans