Litellm fix proxy unit testing (#12778)

* test: update tests

* test: update test
This commit is contained in:
Krish Dholakia 2025-07-19 16:13:03 -07:00 committed by GitHub
parent 55f6460c35
commit 014f4ef86b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 8 additions and 1 deletions

File diff suppressed because one or more lines are too long

View file

@ -165,6 +165,7 @@ def test_chat_completion(mock_acompletion, client_no_auth):
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,
)
print(f"response - {response.text}")
assert response.status_code == 200
@ -435,6 +436,7 @@ def test_engines_model_chat_completions(mock_acompletion, client_no_auth):
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,
)
print(f"response - {response.text}")
assert response.status_code == 200
@ -472,6 +474,7 @@ def test_chat_completion_azure(mock_acompletion, client_no_auth):
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,
)
assert response.status_code == 200
result = response.json()
@ -516,6 +519,7 @@ def test_openai_deployments_model_chat_completions_azure(
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,
)
assert response.status_code == 200
result = response.json()
@ -549,6 +553,7 @@ def test_embedding(mock_aembedding, client_no_auth):
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,
)
assert response.status_code == 200
result = response.json()
@ -576,6 +581,7 @@ def test_bedrock_embedding(mock_aembedding, client_no_auth):
input=["good morning from litellm"],
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,
)
assert response.status_code == 200
result = response.json()
@ -633,6 +639,7 @@ def test_img_gen(mock_aimage_generation, client_no_auth):
size="1024x1024",
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,
)
assert response.status_code == 200
result = response.json()
@ -735,6 +742,7 @@ def test_chat_completion_optional_params(mock_acompletion, client_no_auth):
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,
)
assert response.status_code == 200
result = response.json()