Litellm fix proxy unit testing p2 (#12779)
* test: update tests * test: update test * test: update unit tests
This commit is contained in:
parent
ec01ec923b
commit
0c461b2719
@ -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()
|
||||
|
||||
@ -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()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user