diff --git a/tests/proxy_unit_tests/test_proxy_exception_mapping.py b/tests/proxy_unit_tests/test_proxy_exception_mapping.py index 8171a9cb06..5fbe389f35 100644 --- a/tests/proxy_unit_tests/test_proxy_exception_mapping.py +++ b/tests/proxy_unit_tests/test_proxy_exception_mapping.py @@ -122,6 +122,7 @@ def test_chat_completion_exception_azure(mock_acompletion, client): request_timeout=mock.ANY, metadata=mock.ANY, proxy_server_request=mock.ANY, + secret_fields=mock.ANY, ) json_response = response.json() @@ -155,6 +156,7 @@ def test_embedding_auth_exception_azure(mock_aembedding, client): **test_data, metadata=mock.ANY, proxy_server_request=mock.ANY, + secret_fields=mock.ANY, ) print("Response from proxy=", response) @@ -295,6 +297,7 @@ def test_chat_completion_exception_azure_context_window(mock_acompletion, client request_timeout=mock.ANY, metadata=mock.ANY, proxy_server_request=mock.ANY, + secret_fields=mock.ANY, ) json_response = response.json() diff --git a/tests/test_litellm/proxy/test_proxy_server.py b/tests/test_litellm/proxy/test_proxy_server.py index c429662ec5..a92a9104a3 100644 --- a/tests/test_litellm/proxy/test_proxy_server.py +++ b/tests/test_litellm/proxy/test_proxy_server.py @@ -260,6 +260,7 @@ def test_embedding_input_array_of_tokens(mock_aembedding, client_no_auth): input=[[2046, 13269, 158208]], metadata=mock.ANY, proxy_server_request=mock.ANY, + secret_fields=mock.ANY, ) assert response.status_code == 200 result = response.json()