From 77670fa419415d199b98c0c67364b6a1f8eb4e36 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 27 Sep 2025 14:14:40 -0700 Subject: [PATCH] fix: fix test --- tests/local_testing/test_custom_callback_input.py | 3 ++- 1 file changed, 2 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..87e4d640f7 100644 --- a/tests/local_testing/test_custom_callback_input.py +++ b/tests/local_testing/test_custom_callback_input.py @@ -775,6 +775,8 @@ async def test_async_embedding_openai(): litellm.callbacks = [customHandler_success] response = await litellm.aembedding( model="azure/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}") @@ -913,7 +915,6 @@ async def test_async_embedding_bedrock(): pytest.fail(f"An exception occurred: {str(e)}") - # Image Generation