diff --git a/litellm/__init__.py b/litellm/__init__.py index 8af9a7d10a..1bc690e561 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -557,6 +557,8 @@ stability_models: Set = set() github_copilot_models: Set = set() minimax_models: Set = set() aws_polly_models: Set = set() +gigachat_models: Set = set() +llamagate_models: Set = set() def is_bedrock_pricing_only_model(key: str) -> bool: @@ -809,6 +811,10 @@ def add_known_models(): minimax_models.add(key) elif value.get("litellm_provider") == "aws_polly": aws_polly_models.add(key) + elif value.get("litellm_provider") == "gigachat": + gigachat_models.add(key) + elif value.get("litellm_provider") == "llamagate": + llamagate_models.add(key) add_known_models() @@ -1015,6 +1021,8 @@ models_by_provider: dict = { "github_copilot": github_copilot_models, "minimax": minimax_models, "aws_polly": aws_polly_models, + "gigachat": gigachat_models, + "llamagate": llamagate_models, } # mapping for those models which have larger equivalents