From 6e7654fb540fc099e730c9c0193dc36bc68fd8c1 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 24 Aug 2024 17:26:56 -0700 Subject: [PATCH] fix: fix linting errors --- litellm/llms/OpenAI/openai.py | 3 +-- litellm/llms/azure_text.py | 14 +++++++++----- 2 files changed, 10 insertions(+), 7 deletions(-) diff --git a/litellm/llms/OpenAI/openai.py b/litellm/llms/OpenAI/openai.py index 8021ccd59e4..54bac565772 100644 --- a/litellm/llms/OpenAI/openai.py +++ b/litellm/llms/OpenAI/openai.py @@ -1800,12 +1800,11 @@ class OpenAITextCompletion(BaseLLM): headers: Optional[dict] = None, ): super().completion() - exception_mapping_worked = False try: if headers is None: headers = self.validate_environment(api_key=api_key) if model is None or messages is None: - raise OpenAIError(status_code=422, message=f"Missing model or messages") + raise OpenAIError(status_code=422, message="Missing model or messages") if ( len(messages) > 0 diff --git a/litellm/llms/azure_text.py b/litellm/llms/azure_text.py index 9a8d462e564..db8c516b265 100644 --- a/litellm/llms/azure_text.py +++ b/litellm/llms/azure_text.py @@ -162,11 +162,10 @@ class AzureTextCompletion(BaseLLM): client=None, ): super().completion() - exception_mapping_worked = False try: if model is None or messages is None: raise AzureOpenAIError( - status_code=422, message=f"Missing model or messages" + status_code=422, message="Missing model or messages" ) max_retries = optional_params.pop("max_retries", 2) @@ -293,7 +292,10 @@ class AzureTextCompletion(BaseLLM): "api-version", api_version ) - response = azure_client.completions.create(**data, timeout=timeout) # type: ignore + raw_response = azure_client.completions.with_raw_response.create( + **data, timeout=timeout + ) + response = raw_response.parse() stringified_response = response.model_dump() ## LOGGING logging_obj.post_call( @@ -380,13 +382,15 @@ class AzureTextCompletion(BaseLLM): "complete_input_dict": data, }, ) - response = await azure_client.completions.create(**data, timeout=timeout) + raw_response = await azure_client.completions.with_raw_response.create( + **data, timeout=timeout + ) + response = raw_response.parse() return openai_text_completion_config.convert_to_chat_model_response_object( response_object=response.model_dump(), model_response_object=model_response, ) except AzureOpenAIError as e: - exception_mapping_worked = True raise e except Exception as e: status_code = getattr(e, "status_code", 500)