From 4e672f62691df5111e04aed0650dcd507018463d Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 22 Jan 2025 18:19:49 -0800 Subject: [PATCH] fix: fix test --- .../logging_callback_tests/test_unit_tests_init_callbacks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/logging_callback_tests/test_unit_tests_init_callbacks.py b/tests/logging_callback_tests/test_unit_tests_init_callbacks.py index 69bee7d402..8b549f6180 100644 --- a/tests/logging_callback_tests/test_unit_tests_init_callbacks.py +++ b/tests/logging_callback_tests/test_unit_tests_init_callbacks.py @@ -193,8 +193,8 @@ async def use_callback_in_llm_call( elif used_in == "success_callback": print(f"litellm.success_callback: {litellm.success_callback}") print(f"litellm._async_success_callback: {litellm._async_success_callback}") - assert isinstance(litellm.success_callback[1], expected_class) - assert len(litellm.success_callback) == 2 # ["lago", LagoLogger] + assert isinstance(litellm.success_callback[0], expected_class) + assert len(litellm.success_callback) == 1 # [LagoLogger] assert isinstance(litellm._async_success_callback[0], expected_class) assert len(litellm._async_success_callback) == 1