diff --git a/litellm/tests/test_proxy_exception_mapping.py b/litellm/tests/test_proxy_exception_mapping.py index ccd071d01e..bd1cf4bb67 100644 --- a/litellm/tests/test_proxy_exception_mapping.py +++ b/litellm/tests/test_proxy_exception_mapping.py @@ -103,6 +103,7 @@ def test_chat_completion_exception_azure(mock_acompletion, client): request_timeout=mock.ANY, metadata=mock.ANY, proxy_server_request=mock.ANY, + litellm_parent_otel_span=mock.ANY, ) json_response = response.json() @@ -271,6 +272,7 @@ def test_chat_completion_exception_azure_context_window(mock_acompletion, client request_timeout=mock.ANY, metadata=mock.ANY, proxy_server_request=mock.ANY, + litellm_parent_otel_span=mock.ANY, ) json_response = response.json()