scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-25 05:54:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 126

fix: add base_model param to _supports_factory signature

Browse source
This commit is contained in:
factnn 2026-06-30 00:51:44 +08:00
parent fef8fa4466
commit 9e47d34bf8
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/utils.py
View file

@ -2329,13 +2329,21 @@ def _supports_provider_info_factory(
return None
def _supports_factory(model: str, custom_llm_provider: Optional[str], key: str) -> bool:
def _supports_factory(
model: str,
custom_llm_provider: Optional[str],
key: str,
base_model: Optional[str] = None,
) -> bool:
"""
Check if the given model supports function calling and return a boolean value.
Parameters:
model (str): The model name to be checked.
custom_llm_provider (Optional[str]): The provider to be checked.
base_model (Optional[str]): If set, used as a fallback when the model
string itself is not found in the registry (e.g. custom Azure
deployment names).
Returns:
bool: True if the model supports function calling, False otherwise.