From f7213bacfe242637aaab9119f4857bfbf4d1c248 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 12:18:17 -0700 Subject: [PATCH] test_async_custom_handler_embedding_optional_param --- tests/local_testing/test_custom_logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/local_testing/test_custom_logger.py b/tests/local_testing/test_custom_logger.py index ba9973e11d1..ee212658c89 100644 --- a/tests/local_testing/test_custom_logger.py +++ b/tests/local_testing/test_custom_logger.py @@ -392,7 +392,7 @@ async def test_async_custom_handler_embedding_optional_param(): customHandler_optional_params = MyCustomHandler() litellm.callbacks = [customHandler_optional_params] response = await litellm.aembedding( - model="azure/azure-embedding-model", input=["hello world"], user="John" + model="text-embedding-ada-002", input=["hello world"], user="John" ) await asyncio.sleep(1) # success callback is async assert customHandler_optional_params.user == "John"