tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-06-17 22:48:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix test

Browse Source
This commit is contained in:
Krrish Dholakia
2025-01-22 18:19:49 -08:00
parent 5c88644b87
commit 4e672f6269
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/logging_callback_tests/test_unit_tests_init_callbacks.py
+2 -2
View File
@@ -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