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

fix supports_web_search

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-22 14:02:51 -07:00
parent 1d7accce9e
commit 7dd37a5b18
4 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -4544,6 +4544,7 @@ def _get_model_info_helper( # noqa: PLR0915
supports_native_streaming=_model_info.get(
"supports_native_streaming", None
),
supports_web_search=_model_info.get("supports_web_search", False),
tpm=_model_info.get("tpm", None),
rpm=_model_info.get("rpm", None),
)
@ -4612,6 +4613,7 @@ def get_model_info(model: str, custom_llm_provider: Optional[str] = None) -> Mod
supports_audio_input: Optional[bool]
supports_audio_output: Optional[bool]
supports_pdf_input: Optional[bool]
supports_web_search: Optional[bool]
Raises:
Exception: If the model is not mapped yet.