From 8970f127808c6e9ef554a7337cf11e68b3eee9d8 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 25 Nov 2023 13:48:50 -0800 Subject: [PATCH] fix(utils.py): fix linting issues --- litellm/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index c036ac90bb6..2dfe9302b2b 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -3270,7 +3270,7 @@ def handle_failure(exception, traceback_exception, start_time, end_time, args, k def convert_to_model_response_object(response_object: Optional[dict]=None, model_response_object: Optional[Union[ModelResponse, EmbeddingResponse]]=None, response_type: Literal["completion", "embedding"] = "completion"): try: - if response_type == "completion": + if response_type == "completion" and (model_response_object is None or isinstance(model_response_object, ModelResponse)): if response_object is None or model_response_object is None: raise Exception("Error in response object format") choice_list=[] @@ -3303,7 +3303,7 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model if "model" in response_object: model_response_object.model = response_object["model"] return model_response_object - elif response_type == "embedding": + elif response_type == "embedding" and (model_response_object is None or isinstance(model_response_object, EmbeddingResponse)): if response_object is None: raise Exception("Error in response object format")