Litellm fix proxy unit testing p2 (#12779)

* test: update tests

* test: update test

* test: update unit tests
This commit is contained in:
Krish Dholakia 2025-07-19 16:35:05 -07:00 committed by GitHub
parent ec01ec923b
commit 0c461b2719
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 0 deletions

View File

@ -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()

View File

@ -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()