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

fix: coerce server_tool_use dict to ServerToolUse in Usage.__init__ (#26153)

Browse source
This commit is contained in:
Ishaan Jaffer 2026-05-06 17:05:29 -07:00
parent 6ff668c7aa
commit 0baf01b90e
No known key found for this signature in database
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/types/utils.py
View file

@ -1655,7 +1655,18 @@ class Usage(SafeAttributeModel, CompletionUsage):
)
if server_tool_use is not None:
self.server_tool_use = server_tool_use
# Coerce dict -> ServerToolUse so downstream cost-calc code can
# safely access attributes (e.g. .web_search_requests). Some
# streaming paths round-trip Usage through ``model_dump()`` and
# then re-construct with ``Usage(**dump)``, which would otherwise
# leave server_tool_use as a plain dict.
# See https://github.com/BerriAI/litellm/issues/26153.
if isinstance(server_tool_use, dict):
self.server_tool_use = ServerToolUse(**server_tool_use)
elif isinstance(server_tool_use, ServerToolUse):
self.server_tool_use = server_tool_use
else:
self.server_tool_use = ServerToolUse.model_validate(server_tool_use)
else: # maintain openai compatibility in usage object if possible
del self.server_tool_use