From 673a991da211feb2488a72fdd26862e7954e3c7c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 1 Aug 2024 17:16:02 -0700 Subject: [PATCH] fix test_traceparent_not_added_by_default --- litellm/tests/test_proxy_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_proxy_utils.py b/litellm/tests/test_proxy_utils.py index 30a1aac5cec..c8f6916e2c5 100644 --- a/litellm/tests/test_proxy_utils.py +++ b/litellm/tests/test_proxy_utils.py @@ -102,3 +102,5 @@ async def test_traceparent_not_added_by_default(endpoint, mock_request): _extra_headers = data.get("extra_headers") or {} assert "traceparent" not in _extra_headers + + setattr(litellm.proxy.proxy_server, "open_telemetry_logger", None)