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

feat - show openai params on model hub ui

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-27 08:49:51 -07:00
parent 3be6a07e35
commit b5f883ab74
3 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/router.py
View file

@ -3072,7 +3072,7 @@ class Router:
model=litellm_params.model,
custom_llm_provider=litellm_params.custom_llm_provider,
)
except Exception as e:
except litellm.exceptions.BadRequestError as e:
continue
if model_group_info is None:
@ -3124,19 +3124,23 @@ class Router:
if (
model_info.get("supports_parallel_function_calling", None)
is not None
and model_info["supports_parallel_function_calling"] == True # type: ignore
and model_info["supports_parallel_function_calling"] is True # type: ignore
):
model_group_info.supports_parallel_function_calling = True
if (
model_info.get("supports_vision", None) is not None
and model_info["supports_vision"] == True # type: ignore
and model_info["supports_vision"] is True # type: ignore
):
model_group_info.supports_vision = True
if (
model_info.get("supports_function_calling", None) is not None
and model_info["supports_function_calling"] == True # type: ignore
and model_info["supports_function_calling"] is True # type: ignore
):
model_group_info.supports_function_calling = True
if model_info.get("supported_openai_params", None) is not None:
model_group_info.supported_openai_params = model_info[
"supported_openai_params"
]
return model_group_info