Add minimax provider to models_by_provider

- Add minimax_models set to track minimax models
- Add handling in add_known_models() to populate minimax_models
- Add 'minimax' entry to models_by_provider dictionary
- Fixes test_models_by_provider assertion error where 'minimax' was missing
This commit is contained in:
Alexsander Hamir 2026-01-01 10:56:35 -08:00
parent 3f6312c0ef
commit 422a246ba4

View file

@ -553,6 +553,7 @@ docker_model_runner_models: Set = set()
amazon_nova_models: Set = set()
stability_models: Set = set()
github_copilot_models: Set = set()
minimax_models: Set = set()
def is_bedrock_pricing_only_model(key: str) -> bool:
@ -801,6 +802,8 @@ def add_known_models():
stability_models.add(key)
elif value.get("litellm_provider") == "github_copilot":
github_copilot_models.add(key)
elif value.get("litellm_provider") == "minimax":
minimax_models.add(key)
add_known_models()
@ -1005,6 +1008,7 @@ models_by_provider: dict = {
"amazon_nova": amazon_nova_models,
"stability": stability_models,
"github_copilot": github_copilot_models,
"minimax": minimax_models,
}
# mapping for those models which have larger equivalents