ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: update test apis

Browse Source
This commit is contained in:
Krrish Dholakia 2026-03-28 18:57:27 -07:00
parent 564ad3195b
commit bee1607248
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/local_testing/test_embedding.py
View File

@ -198,8 +198,8 @@ def _azure_ai_image_mock_response(*args, **kwargs):
[
(
"azure_ai/Cohere-embed-v3-multilingual-2",
os.getenv("AZURE_AI_COHERE_API_BASE"),
os.getenv("AZURE_AI_COHERE_API_KEY"),
os.getenv("AZURE_AI_API_BASE"),
os.getenv("AZURE_AI_API_KEY"),
)
],
)

6
tests/vector_store_tests/test_azure_ai_vector_store.py
View File

@ -17,10 +17,10 @@ async def test_basic_search_vector_store(sync_mode):
"vector_store_id": "my-vector-index",
"custom_llm_provider": "azure_ai",
"azure_search_service_name": "azure-kb-search",
"litellm_embedding_model": "azure/text-embedding-3-large",
"litellm_embedding_model": "azure_ai/text-embedding-3-large",
"litellm_embedding_config": {
"api_base": os.getenv("AZURE_AI_SEARCH_EMBEDDING_API_BASE"),
"api_key": os.getenv("AZURE_AI_SEARCH_EMBEDDING_API_KEY"),
"api_base": os.getenv("AZURE_AI_API_BASE"),
"api_key": os.getenv("AZURE_AI_API_KEY"),
},
"api_key": os.getenv("AZURE_SEARCH_API_KEY"),
}