phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(main.py): fix linting error

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-09 18:12:28 -07:00
parent 9083d8e490
commit 425efc60f4
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/main.py
View file

@ -1810,7 +1810,7 @@ def completion(
or get_secret("PREDIBASE_API_KEY")
)
model_response = predibase_chat_completions.completion(
_model_response = predibase_chat_completions.completion(
model=model,
messages=messages,
model_response=model_response,
@ -1832,8 +1832,8 @@ def completion(
and optional_params["stream"] == True
and acompletion == False
):
return response
response = model_response
return _model_response
response = _model_response
elif custom_llm_provider == "ai21":
custom_llm_provider = "ai21"
ai21_key = (