diff --git a/litellm/utils.py b/litellm/utils.py index 6021cc533..b71587693 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"]