From e4014d3e0e0e929da4ca69ee94ded4297f309aa9 Mon Sep 17 00:00:00 2001 From: Simon Sanchez Viloria Date: Thu, 4 Jul 2024 14:26:17 +0200 Subject: [PATCH] fix linting error --- litellm/llms/watsonx.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/llms/watsonx.py b/litellm/llms/watsonx.py index 0ac9b4dca88..fbee32e6eb1 100644 --- a/litellm/llms/watsonx.py +++ b/litellm/llms/watsonx.py @@ -397,7 +397,7 @@ class IBMWatsonXAI(BaseLLM): } def _process_text_gen_response( - self, json_resp: dict, model_response: Union[ModelResponse, None] = None + self, json_resp: dict, model_response = None ) -> ModelResponse: if "results" not in json_resp: raise WatsonXAIError( @@ -430,7 +430,7 @@ class IBMWatsonXAI(BaseLLM): model: str, messages: list, custom_prompt_dict: dict, - model_response: ModelResponse, + model_response, print_verbose: Callable, encoding, logging_obj, @@ -545,7 +545,7 @@ class IBMWatsonXAI(BaseLLM): except Exception as e: raise WatsonXAIError(status_code=500, message=str(e)) - def _process_embedding_response(self, json_resp: dict, model_response:Union[ModelResponse,None]=None) -> ModelResponse: + def _process_embedding_response(self, json_resp: dict, model_response=None) -> ModelResponse: if model_response is None: model_response = ModelResponse(model=json_resp.get("model_id", None)) results = json_resp.get("results", [])