scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

TestAzureOpenAIVectorStore

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-25 14:06:28 -07:00
parent 61ed655574
commit 667f2613de
2 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/llm_responses_api_testing/test_azure_responses_api.py
View file

@ -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?",
)

4
tests/vector_store_tests/test_azure_vector_store.py
View file

@ -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",
}