phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): handle else block for get optional params

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-03 07:45:44 -07:00
parent 5e12307a48
commit ea30359b38
1 changed files with 26 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

29
litellm/utils.py
View file

@ -6062,9 +6062,9 @@ def get_optional_params(
model=model,
api_version=api_version, # type: ignore
)
else: # assume passing in params for azure openai
else: # assume passing in params for text-completion openai
supported_params = get_supported_openai_params(
model=model, custom_llm_provider="azure"
model=model, custom_llm_provider="custom_openai"
)
_check_valid_arg(supported_params=supported_params)
if functions is not None:
@ -6614,7 +6614,30 @@ def get_supported_openai_params(
]
elif custom_llm_provider == "watsonx":
return litellm.IBMWatsonXAIConfig().get_supported_openai_params()
elif custom_llm_provider == "custom_openai" or "text-completion-openai":
return [
"functions",
"function_call",
"temperature",
"top_p",
"n",
"stream",
"stream_options",
"stop",
"max_tokens",
"presence_penalty",
"frequency_penalty",
"logit_bias",
"user",
"response_format",
"seed",
"tools",
"tool_choice",
"max_retries",
"logprobs",
"top_logprobs",
"extra_headers",
]
return None