scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test_models_by_provider

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-23 09:10:41 -07:00
parent 8e65f99ebb
commit ae7b13550e
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/litellm_utils_tests/test_utils.py
View file

@ -1371,6 +1371,9 @@ def test_models_by_provider():
or v["litellm_provider"] == "bedrock_converse"
):
continue
elif v.get("mode") == "search":
# Skip search providers as they don't have traditional models
continue
else:
providers.add(v["litellm_provider"])
@ -2351,7 +2354,7 @@ def test_get_whitelisted_models():
"""
whitelisted_models = []
for model, info in litellm.model_cost.items():
if info["litellm_provider"] == "bedrock" and info["mode"] == "chat":
if info.get("litellm_provider") == "bedrock" and info.get("mode") == "chat":
whitelisted_models.append(model)
# Write to a local file