1package models
2
3import "maps"
4
5type (
6 ModelID string
7 ModelProvider string
8)
9
10type Model struct {
11 ID ModelID `json:"id"`
12 Name string `json:"name"`
13 Provider ModelProvider `json:"provider"`
14 APIModel string `json:"api_model"`
15 CostPer1MIn float64 `json:"cost_per_1m_in"`
16 CostPer1MOut float64 `json:"cost_per_1m_out"`
17 CostPer1MInCached float64 `json:"cost_per_1m_in_cached"`
18 CostPer1MOutCached float64 `json:"cost_per_1m_out_cached"`
19 ContextWindow int64 `json:"context_window"`
20 DefaultMaxTokens int64 `json:"default_max_tokens"`
21 CanReason bool `json:"can_reason"`
22}
23
24// Model IDs
25const ( // GEMINI
26 // Bedrock
27 BedrockClaude37Sonnet ModelID = "bedrock.claude-3.7-sonnet"
28)
29
30const (
31 ProviderBedrock ModelProvider = "bedrock"
32 // ForTests
33 ProviderMock ModelProvider = "__mock"
34)
35
36var SupportedModels = map[ModelID]Model{
37 //
38 // // GEMINI
39 // GEMINI25: {
40 // ID: GEMINI25,
41 // Name: "Gemini 2.5 Pro",
42 // Provider: ProviderGemini,
43 // APIModel: "gemini-2.5-pro-exp-03-25",
44 // CostPer1MIn: 0,
45 // CostPer1MInCached: 0,
46 // CostPer1MOutCached: 0,
47 // CostPer1MOut: 0,
48 // },
49 //
50 // GRMINI20Flash: {
51 // ID: GRMINI20Flash,
52 // Name: "Gemini 2.0 Flash",
53 // Provider: ProviderGemini,
54 // APIModel: "gemini-2.0-flash",
55 // CostPer1MIn: 0.1,
56 // CostPer1MInCached: 0,
57 // CostPer1MOutCached: 0.025,
58 // CostPer1MOut: 0.4,
59 // },
60 //
61 // // Bedrock
62 BedrockClaude37Sonnet: {
63 ID: BedrockClaude37Sonnet,
64 Name: "Bedrock: Claude 3.7 Sonnet",
65 Provider: ProviderBedrock,
66 APIModel: "anthropic.claude-3-7-sonnet-20250219-v1:0",
67 CostPer1MIn: 3.0,
68 CostPer1MInCached: 3.75,
69 CostPer1MOutCached: 0.30,
70 CostPer1MOut: 15.0,
71 },
72}
73
74func init() {
75 maps.Copy(SupportedModels, AnthropicModels)
76 maps.Copy(SupportedModels, OpenAIModels)
77 maps.Copy(SupportedModels, GeminiModels)
78 maps.Copy(SupportedModels, GroqModels)
79 maps.Copy(SupportedModels, AzureModels)
80}