@@ -31,24 +31,37 @@ class OpenAIModels(StrEnum):
3131class GeminiModels (StrEnum ):
3232 """Supported Google Gemini model identifiers."""
3333
34- # Gemini 2 models
34+ # Gemini 2.0 models
35+ gemini_2_0_flash_001 = "gemini-2.0-flash-001"
36+ gemini_2_0_pro_001 = "gemini-2.0-pro-001"
37+
38+ # Gemini 2.5 models
3539 gemini_2_5_pro = "gemini-2.5-pro"
3640 gemini_2_5_flash = "gemini-2.5-flash"
37- gemini_2_0_flash_001 = "gemini-2.0-flash-001"
3841
3942 # Gemini 3 models
4043 gemini_3_pro_preview = "gemini-3-pro-preview"
44+ gemini_3_flash_preview = "gemini-3-flash-preview"
4145
4246
4347class BedrockModels (StrEnum ):
4448 """Supported AWS Bedrock model identifiers."""
4549
4650 # Claude 3.7 models
47- anthropic_claude_3_7_sonnet = "anthropic.claude-3-7-sonnet-20250219-v1:0"
51+ anthropic_bedorck_claude_3_7_sonnet = "anthropic.claude-3-7-sonnet-20250219-v1:0"
4852
4953 # Claude 4 models
50- anthropic_claude_sonnet_4 = "anthropic.claude-sonnet-4-20250514-v1:0"
54+ anthropic_bedorck_claude_sonnet_4 = "anthropic.claude-sonnet-4-20250514-v1:0"
5155
5256 # Claude 4.5 models
53- anthropic_claude_sonnet_4_5 = "anthropic.claude-sonnet-4-5-20250929-v1:0"
54- anthropic_claude_haiku_4_5 = "anthropic.claude-haiku-4-5-20251001-v1:0"
57+ anthropic_bedorck_claude_sonnet_4_5 = "anthropic.claude-sonnet-4-5-20250929-v1:0"
58+ anthropic_bedorck_claude_haiku_4_5 = "anthropic.claude-haiku-4-5-20251001-v1:0"
59+ anthropic_bedorck_claude_opus_4_5 = "anthropic.claude-opus-4-5-20251101-v1:0"
60+
61+
62+ class VertexModels (StrEnum ):
63+ """Supported Vertex model identifiers."""
64+
65+ anthropic_vertex_claude_4_5_sonnet = "claude-opus-4-5@20251101"
66+ anthropic_vertex_claude_4_5_haiku = "claude-haiku-4-5@20251001"
67+ anthropic_vertex_claude_4_5_opus = "claude-opus-4-5@20251101"
0 commit comments