Просмотр исходного кода

switch default to claude sonnet 4

Dax Raad 9 месяцев назад
Родитель
Сommit
2b258b1473
2 измененных файлов с 19 добавлено и 4 удалено
  1. 4 4
      internal/config/config.go
  2. 15 0
      internal/llm/models/anthropic.go

+ 4 - 4
internal/config/config.go

@@ -246,9 +246,9 @@ func setProviderDefaults() {
 
 
 	// Anthropic configuration
 	// Anthropic configuration
 	if key := viper.GetString("providers.anthropic.apiKey"); strings.TrimSpace(key) != "" {
 	if key := viper.GetString("providers.anthropic.apiKey"); strings.TrimSpace(key) != "" {
-		viper.SetDefault("agents.primary.model", models.Claude37Sonnet)
-		viper.SetDefault("agents.task.model", models.Claude37Sonnet)
-		viper.SetDefault("agents.title.model", models.Claude37Sonnet)
+		viper.SetDefault("agents.primary.model", models.Claude4Sonnet)
+		viper.SetDefault("agents.task.model", models.Claude4Sonnet)
+		viper.SetDefault("agents.title.model", models.Claude4Sonnet)
 		return
 		return
 	}
 	}
 
 
@@ -596,7 +596,7 @@ func setDefaultModelForAgent(agent AgentName) bool {
 			maxTokens = 80
 			maxTokens = 80
 		}
 		}
 		cfg.Agents[agent] = Agent{
 		cfg.Agents[agent] = Agent{
-			Model:     models.Claude37Sonnet,
+			Model:     models.Claude4Sonnet,
 			MaxTokens: maxTokens,
 			MaxTokens: maxTokens,
 		}
 		}
 		return true
 		return true

+ 15 - 0
internal/llm/models/anthropic.go

@@ -9,6 +9,7 @@ const (
 	Claude37Sonnet ModelID = "claude-3.7-sonnet"
 	Claude37Sonnet ModelID = "claude-3.7-sonnet"
 	Claude35Haiku  ModelID = "claude-3.5-haiku"
 	Claude35Haiku  ModelID = "claude-3.5-haiku"
 	Claude3Opus    ModelID = "claude-3-opus"
 	Claude3Opus    ModelID = "claude-3-opus"
+	Claude4Sonnet  ModelID = "claude-4-sonnet"
 )
 )
 
 
 // https://docs.anthropic.com/en/docs/about-claude/models/all-models
 // https://docs.anthropic.com/en/docs/about-claude/models/all-models
@@ -53,6 +54,20 @@ var AnthropicModels = map[ModelID]Model{
 		CanReason:           true,
 		CanReason:           true,
 		SupportsAttachments: true,
 		SupportsAttachments: true,
 	},
 	},
+	Claude4Sonnet: {
+		ID:                  Claude4Sonnet,
+		Name:                "Claude 4 Sonnet",
+		Provider:            ProviderAnthropic,
+		APIModel:            "claude-sonnet-4-20250514",
+		CostPer1MIn:         3.0,
+		CostPer1MInCached:   3.75,
+		CostPer1MOutCached:  0.30,
+		CostPer1MOut:        15.0,
+		ContextWindow:       200000,
+		DefaultMaxTokens:    50000,
+		CanReason:           true,
+		SupportsAttachments: true,
+	},
 	Claude35Haiku: {
 	Claude35Haiku: {
 		ID:                  Claude35Haiku,
 		ID:                  Claude35Haiku,
 		Name:                "Claude 3.5 Haiku",
 		Name:                "Claude 3.5 Haiku",