From a343c4d22fc9be79c1141fddd443f3031211d007 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 22 Jan 2024 14:06:32 -0800 Subject: [PATCH] refactor(utils.py): fix linting errors --- litellm/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index f42c8a02494..2b589898c32 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5252,9 +5252,9 @@ def convert_to_model_response_object( model_response_object.model = response_object["model"] if start_time is not None and end_time is not None: - model_response_object._response_ms = ( + model_response_object._response_ms = ( # type: ignore end_time - start_time - ).total_seconds() * 1000 # return response latency in ms like openai + ).total_seconds() * 1000 return model_response_object elif response_type == "embedding" and ( @@ -5281,7 +5281,7 @@ def convert_to_model_response_object( model_response_object.usage.total_tokens = response_object["usage"].get("total_tokens", 0) # type: ignore if start_time is not None and end_time is not None: - model_response_object._response_ms = ( + model_response_object._response_ms = ( # type: ignore end_time - start_time ).total_seconds() * 1000 # return response latency in ms like openai