From 973d4b71e7c72116d9ca00bf88bfce01ca68f8a3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 12:34:49 -0700 Subject: [PATCH] test_embedding_caching_azure --- tests/local_testing/test_caching.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_caching.py b/tests/local_testing/test_caching.py index 8bd73af609..3fb3ae3db4 100644 --- a/tests/local_testing/test_caching.py +++ b/tests/local_testing/test_caching.py @@ -527,7 +527,7 @@ def test_embedding_caching_azure(): print(api_key) print(api_base) embedding1 = embedding( - model="text-embedding-ada-002", + model="azure/text-embedding-ada-002", input=["good morning from litellm", "this is another item"], api_key=api_key, api_base=api_base, @@ -540,7 +540,7 @@ def test_embedding_caching_azure(): time.sleep(1) start_time = time.time() embedding2 = embedding( - model="text-embedding-ada-002", + model="azure/text-embedding-ada-002", input=["good morning from litellm", "this is another item"], api_key=api_key, api_base=api_base,