From 0c461b27190467fd7fbc2e257467a3dc039601cf Mon Sep 17 00:00:00 2001 From: Krish Dholakia Date: Sat, 19 Jul 2025 16:35:05 -0700 Subject: [PATCH] Litellm fix proxy unit testing p2 (#12779) * test: update tests * test: update test * test: update unit tests --- tests/proxy_unit_tests/test_proxy_exception_mapping.py | 3 +++ tests/test_litellm/proxy/test_proxy_server.py | 1 + 2 files changed, 4 insertions(+) diff --git a/tests/proxy_unit_tests/test_proxy_exception_mapping.py b/tests/proxy_unit_tests/test_proxy_exception_mapping.py index 8171a9cb06e..5fbe389f35a 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 c429662ec57..a92a9104a3d 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()