From 2287f383a1bc97be2dbab62aac718671db0b8738 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 7 Jun 2024 13:31:18 -0700 Subject: [PATCH] test fix --- litellm/tests/test_proxy_exception_mapping.py | 2 ++ 1 file changed, 2 insertions(+) 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()