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

style: format streaming usage normalization

Browse source
This commit is contained in:
Mohammad Javad Naderi 2026-08-16 16:40:47 +03:30
parent 58d7a321fa
commit 678e89fc2e
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/litellm_core_utils/streaming_handler.py
View file

@ -1571,9 +1571,7 @@ class CustomStreamWrapper:
if hasattr(chunk, "usage") and chunk.usage is not None:
model_response.usage = (
litellm.Usage(**chunk.usage.model_dump())
if isinstance(chunk.usage, BaseModel)
else chunk.usage
litellm.Usage(**chunk.usage.model_dump()) if isinstance(chunk.usage, BaseModel) else chunk.usage
)
## RETURN ARG