fix gigachat_models
This commit is contained in:
parent
7aca09397a
commit
ebdc2747e7
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user