From b835165742826a1f33d325b3c26e9192ca4d866f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 16 Jul 2024 20:58:34 -0700 Subject: [PATCH] fix otel trace --- tests/otel_tests/test_otel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/otel_tests/test_otel.py b/tests/otel_tests/test_otel.py index e372ddb95..d673553ae 100644 --- a/tests/otel_tests/test_otel.py +++ b/tests/otel_tests/test_otel.py @@ -107,7 +107,7 @@ async def test_chat_completion_check_otel_spans(): print("Parent trace spans: ", parent_trace_spans) - assert len(parent_trace_spans) == 4 + assert 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