Skip to content

Commit daf0305

Browse files
l0gicgateadamdotdevin
authored andcommitted
fix: split out bedrock models
1 parent 307982a commit daf0305

File tree

2 files changed

+27
-48
lines changed

2 files changed

+27
-48
lines changed

internal/llm/models/bedrock.go

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
package models
2+
3+
const (
4+
ProviderBedrock ModelProvider = "bedrock"
5+
6+
// Models
7+
BedrockClaude37Sonnet ModelID = "bedrock.claude-3.7-sonnet"
8+
)
9+
10+
var BedrockModels = map[ModelID]Model{
11+
BedrockClaude37Sonnet: {
12+
ID: BedrockClaude37Sonnet,
13+
Name: "Bedrock: Claude 3.7 Sonnet",
14+
Provider: ProviderBedrock,
15+
APIModel: "anthropic.claude-3-7-sonnet-20250219-v1:0",
16+
CostPer1MIn: 3.0,
17+
CostPer1MInCached: 3.75,
18+
CostPer1MOutCached: 0.30,
19+
CostPer1MOut: 15.0,
20+
ContextWindow: 200_000,
21+
DefaultMaxTokens: 50_000,
22+
CanReason: true,
23+
SupportsAttachments: true,
24+
},
25+
}

internal/llm/models/models.go

Lines changed: 2 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,7 @@ type Model struct {
2222
SupportsAttachments bool `json:"supports_attachments"`
2323
}
2424

25-
// Model IDs
26-
const ( // GEMINI
27-
// Bedrock
28-
BedrockClaude37Sonnet ModelID = "bedrock.claude-3.7-sonnet"
29-
)
30-
3125
const (
32-
ProviderBedrock ModelProvider = "bedrock"
3326
// ForTests
3427
ProviderMock ModelProvider = "__mock"
3528
)
@@ -46,50 +39,11 @@ var ProviderPopularity = map[ModelProvider]int{
4639
ProviderVertexAI: 8,
4740
}
4841

49-
var SupportedModels = map[ModelID]Model{
50-
//
51-
// // GEMINI
52-
// GEMINI25: {
53-
// ID: GEMINI25,
54-
// Name: "Gemini 2.5 Pro",
55-
// Provider: ProviderGemini,
56-
// APIModel: "gemini-2.5-pro-exp-03-25",
57-
// CostPer1MIn: 0,
58-
// CostPer1MInCached: 0,
59-
// CostPer1MOutCached: 0,
60-
// CostPer1MOut: 0,
61-
// },
62-
//
63-
// GRMINI20Flash: {
64-
// ID: GRMINI20Flash,
65-
// Name: "Gemini 2.0 Flash",
66-
// Provider: ProviderGemini,
67-
// APIModel: "gemini-2.0-flash",
68-
// CostPer1MIn: 0.1,
69-
// CostPer1MInCached: 0,
70-
// CostPer1MOutCached: 0.025,
71-
// CostPer1MOut: 0.4,
72-
// },
73-
//
74-
// // Bedrock
75-
BedrockClaude37Sonnet: {
76-
ID: BedrockClaude37Sonnet,
77-
Name: "Bedrock: Claude 3.7 Sonnet",
78-
Provider: ProviderBedrock,
79-
APIModel: "anthropic.claude-3-7-sonnet-20250219-v1:0",
80-
CostPer1MIn: 3.0,
81-
CostPer1MInCached: 3.75,
82-
CostPer1MOutCached: 0.30,
83-
CostPer1MOut: 15.0,
84-
ContextWindow: 200_000,
85-
DefaultMaxTokens: 50_000,
86-
CanReason: true,
87-
SupportsAttachments: true,
88-
},
89-
}
42+
var SupportedModels = map[ModelID]Model{}
9043

9144
func init() {
9245
maps.Copy(SupportedModels, AnthropicModels)
46+
maps.Copy(SupportedModels, BedrockModels)
9347
maps.Copy(SupportedModels, OpenAIModels)
9448
maps.Copy(SupportedModels, GeminiModels)
9549
maps.Copy(SupportedModels, GroqModels)

0 commit comments

Comments
 (0)