From 4f46ac4ab5644b7e76b5e0630b4fc50a67df770c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 20 Nov 2023 18:32:43 -0800 Subject: [PATCH] (fix) linting error --- litellm/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 6021cc5332..b715876930 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -3079,9 +3079,9 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model model_response_object.choices = choice_list if "usage" in response_object: - model_response_object.usage.completion_tokens = response_object["usage"]["completion_tokens"] - model_response_object.usage.prompt_tokens = response_object["usage"]["prompt_tokens"] - model_response_object.usage.total_tokens = response_object["usage"]["total_tokens"] + model_response_object.usage.completion_tokens = response_object["usage"].get("completion_tokens", 0) + model_response_object.usage.prompt_tokens = response_object["usage"].get("prompt_tokens", 0) + model_response_object.usage.total_tokens = response_object["usage"].get("total_tokens", 0) if "id" in response_object: model_response_object.id = response_object["id"]