From 667f2613de4fd20bf793f95f675520ef069004e1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 25 Oct 2025 14:06:28 -0700 Subject: [PATCH] TestAzureOpenAIVectorStore --- .../test_azure_responses_api.py | 12 ++++++------ tests/vector_store_tests/test_azure_vector_store.py | 4 ++-- 2 files changed, 8 insertions(+), 8 deletions(-) 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