fix test litellm_parent_otel_span

This commit is contained in:
Ishaan Jaff 2024-06-07 14:07:58 -07:00
parent d9dacc1f43
commit 7dcf8fc67e
2 changed files with 6 additions and 8 deletions

View file

@ -190,7 +190,6 @@ def test_engines_model_chat_completions(mock_acompletion, client_no_auth):
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
litellm_parent_otel_span=mock.ANY,
)
print(f"response - {response.text}")
assert response.status_code == 200
@ -228,7 +227,6 @@ def test_chat_completion_azure(mock_acompletion, client_no_auth):
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
litellm_parent_otel_span=mock.ANY,
)
assert response.status_code == 200
result = response.json()
@ -273,7 +271,6 @@ def test_openai_deployments_model_chat_completions_azure(
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
litellm_parent_otel_span=mock.ANY,
)
assert response.status_code == 200
result = response.json()
@ -488,7 +485,6 @@ def test_chat_completion_optional_params(mock_acompletion, client_no_auth):
specific_deployment=True,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
litellm_parent_otel_span=mock.ANY,
)
assert response.status_code == 200
result = response.json()