From 2aacf64db198c4d2251b014d4898db5b811fea49 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 11 Aug 2025 22:58:58 -0700 Subject: [PATCH] test: add unit tests --- .../test_openai_responses_transformation.py | 53 ++++++++++++++++--- 1 file changed, 45 insertions(+), 8 deletions(-) diff --git a/tests/test_litellm/llms/openai/responses/test_openai_responses_transformation.py b/tests/test_litellm/llms/openai/responses/test_openai_responses_transformation.py index 1ed2266d1f..a587832a1a 100644 --- a/tests/test_litellm/llms/openai/responses/test_openai_responses_transformation.py +++ b/tests/test_litellm/llms/openai/responses/test_openai_responses_transformation.py @@ -147,7 +147,7 @@ class TestOpenAIResponsesAPIConfig: assert result.type == ResponsesAPIStreamEvents.RESPONSE_COMPLETED assert result.response.id == "resp_123" - + @pytest.mark.serial def test_validate_environment(self): """Test that validate_environment correctly sets the Authorization header""" @@ -292,27 +292,64 @@ class TestAzureResponsesAPIConfig: def test_azure_get_complete_url_with_version_types(self): """Test Azure get_complete_url with different API version types""" base_url = "https://litellm8397336933.openai.azure.com" - + # Test with preview version - should use openai/v1/responses result_preview = self.config.get_complete_url( api_base=base_url, litellm_params={"api_version": "preview"}, ) - assert result_preview == "https://litellm8397336933.openai.azure.com/openai/v1/responses?api-version=preview" - - # Test with latest version - should use openai/v1/responses + assert ( + result_preview + == "https://litellm8397336933.openai.azure.com/openai/v1/responses?api-version=preview" + ) + + # Test with latest version - should use openai/v1/responses result_latest = self.config.get_complete_url( api_base=base_url, litellm_params={"api_version": "latest"}, ) - assert result_latest == "https://litellm8397336933.openai.azure.com/openai/v1/responses?api-version=latest" - + assert ( + result_latest + == "https://litellm8397336933.openai.azure.com/openai/v1/responses?api-version=latest" + ) + # Test with date-based version - should use openai/responses result_date = self.config.get_complete_url( api_base=base_url, litellm_params={"api_version": "2025-01-01"}, ) - assert result_date == "https://litellm8397336933.openai.azure.com/openai/responses?api-version=2025-01-01" + assert ( + result_date + == "https://litellm8397336933.openai.azure.com/openai/responses?api-version=2025-01-01" + ) + + def test_azure_get_complete_url_with_default_api_version(self): + """Test Azure get_complete_url uses default API version when none is provided""" + from litellm.constants import AZURE_DEFAULT_RESPONSES_API_VERSION + + base_url = "https://litellm8397336933.openai.azure.com" + + # Test with no api_version provided - should use default + result_no_version = self.config.get_complete_url( + api_base=base_url, + litellm_params={}, + ) + expected_url = f"https://litellm8397336933.openai.azure.com/openai/responses?api-version={AZURE_DEFAULT_RESPONSES_API_VERSION}" + assert result_no_version == expected_url + + # Test with empty litellm_params - should use default + result_empty_params = self.config.get_complete_url( + api_base=base_url, + litellm_params={}, + ) + assert result_empty_params == expected_url + + # Test with None api_version - should use default + result_none_version = self.config.get_complete_url( + api_base=base_url, + litellm_params={"api_version": None}, + ) + assert result_none_version == expected_url class TestTransformListInputItemsRequest: