From ed64dd7f9d43a4c54ad91f9450d46738fe3401fc Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 23 Nov 2024 03:06:51 +0530 Subject: [PATCH] test: fix test --- .../test_unit_tests_init_callbacks.py | 7 +++++-- 1 file changed, 5 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 2c373772a..15c2118d8 100644 --- a/tests/logging_callback_tests/test_unit_tests_init_callbacks.py +++ b/tests/logging_callback_tests/test_unit_tests_init_callbacks.py @@ -285,6 +285,9 @@ def test_dynamic_logging_global_callback(): def test_get_combined_callback_list(): from litellm.litellm_core_utils.litellm_logging import get_combined_callback_list - assert get_combined_callback_list( + assert "langfuse" in get_combined_callback_list( dynamic_success_callbacks=["langfuse"], global_callbacks=["lago"] - ) == ["langfuse", "lago"] + ) + assert "lago" in get_combined_callback_list( + dynamic_success_callbacks=["langfuse"], global_callbacks=["lago"] + )