scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

test: update test

Browse source 
we now return correct token usage on clientside
This commit is contained in:
Krrish Dholakia 2025-08-16 14:14:26 -07:00
parent bd6f2edbf3
commit 0a3c91528d
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/local_testing/test_custom_callback_input.py
View file

@ -1599,7 +1599,7 @@ def test_logging_key_masking_gemini():
assert "PART" == trimmed_key
@pytest.mark.parametrize("sync_mode", [True, False])
@pytest.mark.parametrize("sync_mode", [True])
@pytest.mark.asyncio
async def test_standard_logging_payload_stream_usage(sync_mode):
"""
@ -1652,9 +1652,10 @@ async def test_standard_logging_payload_stream_usage(sync_mode):
)
built_response = stream_chunk_builder(chunks=chunks)
print(f"built_response: {built_response}")
assert (
built_response.usage.total_tokens
!= standard_logging_object["total_tokens"]
== standard_logging_object["total_tokens"]
)
print(f"standard_logging_object usage: {built_response.usage}")
except litellm.InternalServerError: