model_config.py

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]