File size: 934 Bytes
a0e37e2
f5c9c80
 
 
 
 
 
 
 
 
 
 
a0e37e2
 
 
 
9433533
f5c9c80
a0e37e2
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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",
})