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

fix(utils.py): fix openrouter params

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/4488
This commit is contained in:
Krrish Dholakia 2024-07-01 08:23:46 -07:00
parent 49a72ffb51
commit 5e521bd36e
1 changed files with 8 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

21
litellm/utils.py
View file

@ -3842,23 +3842,18 @@ def get_supported_openai_params(
return litellm.AzureOpenAIConfig().get_supported_openai_params() return litellm.AzureOpenAIConfig().get_supported_openai_params()
elif custom_llm_provider == "openrouter": elif custom_llm_provider == "openrouter":
return [ return [
"functions",
"function_call",
"temperature", "temperature",
"top_p", "top_p",
"n",
"stream",
"stop",
"max_tokens",
"presence_penalty",
"frequency_penalty", "frequency_penalty",
"logit_bias", "presence_penalty",
"user", "repetition_penalty",
"response_format",
"seed", "seed",
"tools", "max_tokens",
"tool_choice", "logit_bias",
"max_retries", "logprobs",
"top_logprobs",
"response_format",
"stop",
] ]
elif custom_llm_provider == "mistral" or custom_llm_provider == "codestral": elif custom_llm_provider == "mistral" or custom_llm_provider == "codestral":
# mistal and codestral api have the exact same params # mistal and codestral api have the exact same params