scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 48

Avoid usage_input dict in-place

Browse source
This commit is contained in:
Liam McDonald 2026-05-08 15:15:48 -07:00
parent 35ca10e73a
commit 4b6a0c2356
1 changed files with 6 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
litellm/responses/utils.py
View file

@ -914,22 +914,12 @@ class ResponseAPILoggingUtils:
)
response_api_usage: ResponseAPIUsage
if isinstance(usage_input, dict):
# Realtime uses input_token_details and output_token_details; convert to
# input_tokens_details and output_tokens_details when those are unset.
if (
usage_input.get("input_tokens_details") is None
and "input_token_details" in usage_input
):
usage_input["input_tokens_details"] = usage_input[
"input_token_details"
]
if (
usage_input.get("output_tokens_details") is None
and "output_token_details" in usage_input
):
usage_input["output_tokens_details"] = usage_input[
"output_token_details"
]
usage_input = dict(usage_input) # shallow copy; avoid mutating caller
# Realtime *_token_details → *_tokens_details when unset.
if usage_input.get("input_tokens_details") is None and "input_token_details" in usage_input:
usage_input["input_tokens_details"] = usage_input["input_token_details"]
if usage_input.get("output_tokens_details") is None and "output_token_details" in usage_input:
usage_input["output_tokens_details"] = usage_input["output_token_details"]
total_tokens = usage_input.get("total_tokens")
if total_tokens is None:
input_tokens = usage_input.get("input_tokens")