scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 131

test(proxy): assert shared callback is _ProxyDBLogger instance

Browse source 
Adds isinstance check to verify the shared object is actually a
_ProxyDBLogger, not just that two args share identity.
This commit is contained in:
yryzhan 2026-05-20 17:49:10 +02:00
parent 38660efdfc
commit d19b37c97b
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/proxy/test_proxy_server.py
View file

@ -7501,10 +7501,13 @@ class TestCostTrackingSingleDBLogger:
with patch("litellm.logging_callback_manager", mock_callback_manager):
cost_tracking()
from litellm.proxy.hooks.proxy_track_cost_callback import _ProxyDBLogger
add_callback_call = mock_callback_manager.add_litellm_callback.call_args[0][0]
add_async_call = (
mock_callback_manager.add_litellm_async_success_callback.call_args[0][0]
)
assert isinstance(add_callback_call, _ProxyDBLogger)
assert add_callback_call is add_async_call
def test_cost_tracking_noop_when_prisma_client_is_none(self):