phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix proxy server test

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-07 12:54:39 -07:00
parent 923cbed6ab
commit 630bc803e2
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_proxy_server.py
View file

@ -191,6 +191,7 @@ def test_engines_model_chat_completions(mock_acompletion, client_no_auth):
specific_deployment=True, specific_deployment=True,
metadata=mock.ANY, metadata=mock.ANY,
proxy_server_request=mock.ANY, proxy_server_request=mock.ANY,
litellm_parent_otel_span=mock.ANY,
) )
print(f"response - {response.text}") print(f"response - {response.text}")
assert response.status_code == 200 assert response.status_code == 200
@ -228,6 +229,7 @@ def test_chat_completion_azure(mock_acompletion, client_no_auth):
specific_deployment=True, specific_deployment=True,
metadata=mock.ANY, metadata=mock.ANY,
proxy_server_request=mock.ANY, proxy_server_request=mock.ANY,
litellm_parent_otel_span=mock.ANY,
) )
assert response.status_code == 200 assert response.status_code == 200
result = response.json() result = response.json()
@ -272,6 +274,7 @@ def test_openai_deployments_model_chat_completions_azure(
specific_deployment=True, specific_deployment=True,
metadata=mock.ANY, metadata=mock.ANY,
proxy_server_request=mock.ANY, proxy_server_request=mock.ANY,
litellm_parent_otel_span=mock.ANY,
) )
assert response.status_code == 200 assert response.status_code == 200
result = response.json() result = response.json()
@ -486,6 +489,7 @@ def test_chat_completion_optional_params(mock_acompletion, client_no_auth):
specific_deployment=True, specific_deployment=True,
metadata=mock.ANY, metadata=mock.ANY,
proxy_server_request=mock.ANY, proxy_server_request=mock.ANY,
litellm_parent_otel_span=mock.ANY,
) )
assert response.status_code == 200 assert response.status_code == 200
result = response.json() result = response.json()