ask_candid/base/config/models.py

from types import MappingProxyType
from enum import Enum


class BedrockEndpoints(Enum):
    claude_35_haiku = "us.anthropic.claude-3-5-haiku-20241022-v1:0"
    claude_4_sonnet = "us.anthropic.claude-sonnet-4-20250514-v1:0"
    claude_45_sonnet = "us.anthropic.claude-sonnet-4-5-20250929-v1:0"
    llama_31_70b_instruct = "us.meta.llama3-1-70b-instruct-v1:0"
    mistral_large = "mistral.mistral-large-2402-v1:0"
    mixtral_8x7b = "mistral.mixtral-8x7b-instruct-v0:1"


Name2Endpoint = MappingProxyType({
    "gpt-4o": "gpt-4o",
    "claude-3.5-haiku": "us.anthropic.claude-3-5-haiku-20241022-v1:0",
    "claude-4-sonnet": "us.anthropic.claude-sonnet-4-20250514-v1:0",
    "claude-4.5-sonnet": "us.anthropic.claude-sonnet-4-5-20250929-v1:0",
    # "llama-3.1-70b-instruct": "us.meta.llama3-1-70b-instruct-v1:0",
    # "mistral-large": "mistral.mistral-large-2402-v1:0",
    # "mixtral-8x7B": "mistral.mixtral-8x7b-instruct-v0:1",
})