diff --git a/tests/llm_responses_api_testing/test_azure_responses_api.py b/tests/llm_responses_api_testing/test_azure_responses_api.py index daeb288e4e8..5f3e7f2c399 100644 --- a/tests/llm_responses_api_testing/test_azure_responses_api.py +++ b/tests/llm_responses_api_testing/test_azure_responses_api.py @@ -23,10 +23,10 @@ from base_responses_api import BaseResponsesAPITest class TestAzureResponsesAPITest(BaseResponsesAPITest): def get_base_completion_call_args(self): return { - "model": "azure/gpt-5-nano", + "model": "azure/gpt-5-mini", "truncation": "auto", - "api_base": os.getenv("AZURE_RESPONSES_OPENAI_ENDPOINT"), - "api_key": os.getenv("AZURE_RESPONSES_OPENAI_API_KEY"), + "api_base": os.getenv("AZURE_API_BASE"), + "api_key": os.getenv("AZURE_API_KEY"), "api_version": os.getenv("AZURE_RESPONSES_OPENAI_API_VERSION"), } @@ -38,11 +38,11 @@ async def test_azure_responses_api_preview_api_version(): """ litellm._turn_on_debug() response = await litellm.aresponses( - model="azure/gpt-5-nano", + model="azure/gpt-5-mini", truncation="auto", api_version="preview", - api_base=os.getenv("AZURE_RESPONSES_OPENAI_ENDPOINT"), - api_key=os.getenv("AZURE_RESPONSES_OPENAI_API_KEY"), + api_base=os.getenv("AZURE_API_BASE"), + api_key=os.getenv("AZURE_API_KEY"), input="Hello, can you tell me a short joke?", ) diff --git a/tests/vector_store_tests/test_azure_vector_store.py b/tests/vector_store_tests/test_azure_vector_store.py index 783417cb742..29e2cfc9203 100644 --- a/tests/vector_store_tests/test_azure_vector_store.py +++ b/tests/vector_store_tests/test_azure_vector_store.py @@ -19,7 +19,7 @@ class TestAzureOpenAIVectorStore(BaseVectorStoreTest): """ return { "custom_llm_provider": "azure", - "api_base": os.getenv("AZURE_RESPONSES_OPENAI_ENDPOINT"), - "api_key": os.getenv("AZURE_RESPONSES_OPENAI_API_KEY"), + "api_base": os.getenv("AZURE_API_BASE"), + "api_key": os.getenv("AZURE_API_KEY"), "api_version": "2025-04-01-preview", } \ No newline at end of file