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

fix gcs logging

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-12 15:24:04 -07:00
parent 46ce4995b8
commit bb38e9cbf8
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/types/utils.py
View file

@ -484,9 +484,12 @@ class Usage(CompletionUsage):
):
prompt_tokens = params["prompt_cache_miss_tokens"]
completion_tokens_details = CompletionTokensDetails(
reasoning_tokens=reasoning_tokens
)
# handle reasoning_tokens
completion_tokens_details = None
if reasoning_tokens:
completion_tokens_details = CompletionTokensDetails(
reasoning_tokens=reasoning_tokens
)
super().__init__(
prompt_tokens=prompt_tokens or 0,
completion_tokens=completion_tokens or 0,