From bd77bdabe634fd9def69101c39bcb71b155d15a5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 11:27:34 -0700 Subject: [PATCH] test_call_one_endpoint --- tests/local_testing/test_router.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py index e7e61d378c..fbe0a2fb8a 100644 --- a/tests/local_testing/test_router.py +++ b/tests/local_testing/test_router.py @@ -544,7 +544,7 @@ def test_call_one_endpoint(): { "model_name": "gpt-3.5-turbo", # openai model name "litellm_params": { # params for litellm completion/embedding call - "model": "azure/gpt-4o-new-test", + "model": "azure/gpt-4.1-nano", "api_key": old_api_key, "api_version": os.getenv("AZURE_API_VERSION"), "api_base": os.getenv("AZURE_API_BASE"), @@ -555,7 +555,7 @@ def test_call_one_endpoint(): { "model_name": "text-embedding-ada-002", "litellm_params": { - "model": "azure/azure-embedding-model", + "model": "azure/text-embedding-ada-002", "api_key": os.environ["AZURE_API_KEY"], "api_base": os.environ["AZURE_API_BASE"], }, @@ -574,7 +574,7 @@ def test_call_one_endpoint(): async def call_azure_completion(): response = await router.acompletion( - model="azure/gpt-4o-new-test", + model="azure/gpt-4.1-nano", messages=[{"role": "user", "content": "hello this request will pass"}], specific_deployment=True, ) @@ -582,7 +582,7 @@ def test_call_one_endpoint(): async def call_azure_embedding(): response = await router.aembedding( - model="azure/azure-embedding-model", + model="azure/text-embedding-ada-002", input=["good morning from litellm"], specific_deployment=True, )