@@ -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-
3125const (
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
9144func 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