Spaces:
Running
Running
File size: 835 Bytes
e91e2b4 3b1bcbe e91e2b4 3b1bcbe e91e2b4 3b1bcbe e91e2b4 3b1bcbe e91e2b4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
PROVIDER_MODEL_MAP = {
"openai": [
"gpt-4.1-mini",
"gpt-5-mini",
"gpt-5-nano",
],
"anthropic": [
"anthropic.claude-3-5-sonnet-20241022-v2:0",
"anthropic.claude-3-haiku-20240307-v1:0",
],
}
MODEL_TO_PROVIDER = {
model: provider
for provider, models in PROVIDER_MODEL_MAP.items()
for model in models
}
# Model display names with categories
MODEL_DISPLAY_NAMES = {
"gpt-4.1-mini": "OpenAI - GPT-4.1 Mini",
"gpt-5-mini": "OpenAI - GPT-5 Mini",
"gpt-5-nano": "OpenAI - GPT-5 Nano",
"anthropic.claude-3-5-sonnet-20241022-v2:0": "AWS Bedrock - Anthropic - Claude 3.5 Sonnet",
"anthropic.claude-3-haiku-20240307-v1:0": "AWS Bedrock - Anthropic - Claude 3 Haiku",
}
MODEL_CHOICES = [model for models in PROVIDER_MODEL_MAP.values() for model in models]
|