From 7f037feedd6b525b8862ce5fbab838e92eedb239 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 19 Aug 2023 20:16:54 -0700 Subject: [PATCH] linting bug fix --- litellm/llms/huggingface_restapi.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/llms/huggingface_restapi.py b/litellm/llms/huggingface_restapi.py index 997c322bd79..624fb4f0556 100644 --- a/litellm/llms/huggingface_restapi.py +++ b/litellm/llms/huggingface_restapi.py @@ -77,7 +77,7 @@ class HuggingfaceRestAPILLM: "parameters": optional_params } ## LOGGING - logging.pre_call(input=prompt, api_key=self.api_key, additional_args={"complete_input_dict": data}) + self.logging_obj.pre_call(input=prompt, api_key=self.api_key, additional_args={"complete_input_dict": data}) ## COMPLETION CALL response = requests.post( completion_url, headers=self.headers, data=json.dumps(data) @@ -86,7 +86,7 @@ class HuggingfaceRestAPILLM: return response.iter_lines() else: ## LOGGING - logging.post_call(input=prompt, api_key=self.api_key, original_response=response.text, additional_args={"complete_input_dict": data}) + self.logging_obj.post_call(input=prompt, api_key=self.api_key, original_response=response.text, additional_args={"complete_input_dict": data}) ## RESPONSE OBJECT completion_response = response.json() print_verbose(f"response: {completion_response}")