From 3f30f935169a2c151c24c8d31b05c8e8dfbcf130 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 20 Nov 2023 18:26:03 -0800 Subject: [PATCH] (fix) pydantic errors openai usage --- litellm/utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 634c093297..6021cc5332 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -3078,9 +3078,11 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model choice_list.append(choice) model_response_object.choices = choice_list - if "usage" in response_object: - model_response_object.usage = response_object["usage"] - + 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"] + if "id" in response_object: model_response_object.id = response_object["id"]