From bb38e9cbf82516b64c5918081a7d48e0ccb8e796 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 12 Sep 2024 15:24:04 -0700 Subject: [PATCH] fix gcs logging --- litellm/types/utils.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/litellm/types/utils.py b/litellm/types/utils.py index b5f1ba08051..1283a379b05 100644 --- a/litellm/types/utils.py +++ b/litellm/types/utils.py @@ -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,