scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

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
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/logging_callback_tests/test_unit_tests_init_callbacks.py
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