From 22c033d3d77344fae4f6987c07efade338921fb7 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 09:28:24 -0700 Subject: [PATCH] test_acompletion_caching_on_router_caching_groups --- tests/local_testing/test_router_caching.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/tests/local_testing/test_router_caching.py b/tests/local_testing/test_router_caching.py index 574f133ace..d9ee4fdeb5 100644 --- a/tests/local_testing/test_router_caching.py +++ b/tests/local_testing/test_router_caching.py @@ -213,10 +213,8 @@ async def test_acompletion_caching_with_ttl_on_router(): { "model_name": "gpt-3.5-turbo", "litellm_params": { - "model": "azure/chatgpt-v-3", - "api_key": os.getenv("AZURE_API_KEY"), - "api_base": os.getenv("AZURE_API_BASE"), - "api_version": os.getenv("AZURE_API_VERSION"), + "model": "gpt-4.1-nano", + "api_key": os.getenv("OPENAI_API_KEY"), }, "tpm": 100000, "rpm": 10000, @@ -277,9 +275,9 @@ async def test_acompletion_caching_on_router_caching_groups(): "rpm": 10000, }, { - "model_name": "azure-gpt-3.5-turbo", + "model_name": "azure-gpt-4.1-nano", "litellm_params": { - "model": "azure/chatgpt-v-3", + "model": "azure/gpt-4.1-nano", "api_key": os.getenv("AZURE_API_KEY"), "api_base": os.getenv("AZURE_API_BASE"), "api_version": os.getenv("AZURE_API_VERSION"),