scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

fix linting error

Browse source
This commit is contained in:
Simon Sanchez Viloria 2024-07-04 14:26:17 +02:00
parent 574de1d7fd
commit e4014d3e0e
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/watsonx.py
View file

@ -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", [])