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

fix: add aws_polly to models_by_provider to fix test_models_by_provider test

Browse source
This commit is contained in:
Alexsander Hamir 2026-01-01 11:59:02 -08:00
parent 422a246ba4
commit e4c2a8d22f
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/__init__.py
View file

@ -554,6 +554,7 @@ amazon_nova_models: Set = set()
stability_models: Set = set()
github_copilot_models: Set = set()
minimax_models: Set = set()
aws_polly_models: Set = set()
def is_bedrock_pricing_only_model(key: str) -> bool:
@ -804,6 +805,8 @@ def add_known_models():
github_copilot_models.add(key)
elif value.get("litellm_provider") == "minimax":
minimax_models.add(key)
elif value.get("litellm_provider") == "aws_polly":
aws_polly_models.add(key)
add_known_models()
@ -1009,6 +1012,7 @@ models_by_provider: dict = {
"stability": stability_models,
"github_copilot": github_copilot_models,
"minimax": minimax_models,
"aws_polly": aws_polly_models,
}
# mapping for those models which have larger equivalents