From 7dcf8fc67e668e19a5df5303c379f36b496a48d8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 7 Jun 2024 14:07:58 -0700 Subject: [PATCH] fix test litellm_parent_otel_span --- litellm/tests/test_proxy_exception_mapping.py | 10 ++++++---- litellm/tests/test_proxy_server.py | 4 ---- 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/litellm/tests/test_proxy_exception_mapping.py b/litellm/tests/test_proxy_exception_mapping.py index bd1cf4bb679..4988426616a 100644 --- a/litellm/tests/test_proxy_exception_mapping.py +++ b/litellm/tests/test_proxy_exception_mapping.py @@ -103,7 +103,6 @@ def test_chat_completion_exception_azure(mock_acompletion, client): request_timeout=mock.ANY, metadata=mock.ANY, proxy_server_request=mock.ANY, - litellm_parent_otel_span=mock.ANY, ) json_response = response.json() @@ -211,7 +210,9 @@ def test_chat_completion_exception_any_model(client): ) assert isinstance(openai_exception, openai.BadRequestError) _error_message = openai_exception.message - assert "chat_completion: Invalid model name passed in model=Lite-GPT-12" in str(_error_message) + assert "chat_completion: Invalid model name passed in model=Lite-GPT-12" in str( + _error_message + ) except Exception as e: pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}") @@ -239,7 +240,9 @@ def test_embedding_exception_any_model(client): print("Exception raised=", openai_exception) assert isinstance(openai_exception, openai.BadRequestError) _error_message = openai_exception.message - assert "embeddings: Invalid model name passed in model=Lite-GPT-12" in str(_error_message) + assert "embeddings: Invalid model name passed in model=Lite-GPT-12" in str( + _error_message + ) except Exception as e: pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}") @@ -272,7 +275,6 @@ def test_chat_completion_exception_azure_context_window(mock_acompletion, client request_timeout=mock.ANY, metadata=mock.ANY, proxy_server_request=mock.ANY, - litellm_parent_otel_span=mock.ANY, ) json_response = response.json() diff --git a/litellm/tests/test_proxy_server.py b/litellm/tests/test_proxy_server.py index a37f8adbd1b..114b96872f6 100644 --- a/litellm/tests/test_proxy_server.py +++ b/litellm/tests/test_proxy_server.py @@ -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()