From 0a2de5993285612d5050b76fbae8994ddd926c68 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 27 Oct 2023 09:32:25 -0700 Subject: [PATCH] (feat) use usage class for anthropic --- litellm/llms/anthropic.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/litellm/llms/anthropic.py b/litellm/llms/anthropic.py index 7a2b3d8d845..8cce80826bd 100644 --- a/litellm/llms/anthropic.py +++ b/litellm/llms/anthropic.py @@ -164,11 +164,9 @@ def completion( model_response["created"] = time.time() model_response["model"] = model - model_response["usage"] = { - "prompt_tokens": prompt_tokens, - "completion_tokens": completion_tokens, - "total_tokens": prompt_tokens + completion_tokens, - } + model_response.usage.completion_tokens = completion_tokens + model_response.usage.prompt_tokens = prompt_tokens + model_response.usage.total_tokens = prompt_tokens + completion_tokens return model_response def embedding():