From 81282d14b5196eccdc59c0c993ea16b19d08bcff Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 13:03:17 -0700 Subject: [PATCH] test_async_embedding_openai --- tests/local_testing/test_custom_callback_input.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/local_testing/test_custom_callback_input.py b/tests/local_testing/test_custom_callback_input.py index 36fa31f4ac..cd73045cb2 100644 --- a/tests/local_testing/test_custom_callback_input.py +++ b/tests/local_testing/test_custom_callback_input.py @@ -774,7 +774,9 @@ async def test_async_embedding_openai(): customHandler_failure = CompletionCustomHandler() litellm.callbacks = [customHandler_success] response = await litellm.aembedding( - model="azure/text-embedding-ada-002", + model="text-embedding-ada-002", + input=["good morning from litellm"], + ) await asyncio.sleep(1) print(f"customHandler_success.errors: {customHandler_success.errors}") print(f"customHandler_success.states: {customHandler_success.states}")