From 94d5fe0520e5318d618ec4d7d6d791a88cbe43b6 Mon Sep 17 00:00:00 2001
From: Kujtim Hoxha <kujtimii.h@gmail.com>
Date: Thu, 22 May 2025 20:05:52 +0200
Subject: [PATCH] add sonnet 4

---
 internal/config/config.go        |  8 ++++----
 internal/llm/models/anthropic.go | 29 +++++++++++++++++++++++++++++
 2 files changed, 33 insertions(+), 4 deletions(-)

diff --git a/internal/config/config.go b/internal/config/config.go
index ff7585db9a6008d082191b252527bd11107ad74d..351bc501ace11b8de763975fbedda4ab28e7cb1a 100644
--- a/internal/config/config.go
+++ b/internal/config/config.go
@@ -280,10 +280,10 @@ func setProviderDefaults() {
 
 	// Anthropic configuration
 	if key := viper.GetString("providers.anthropic.apiKey"); strings.TrimSpace(key) != "" {
-		viper.SetDefault("agents.coder.model", models.Claude37Sonnet)
-		viper.SetDefault("agents.summarizer.model", models.Claude37Sonnet)
-		viper.SetDefault("agents.task.model", models.Claude37Sonnet)
-		viper.SetDefault("agents.title.model", models.Claude37Sonnet)
+		viper.SetDefault("agents.coder.model", models.Claude4Sonnet)
+		viper.SetDefault("agents.summarizer.model", models.Claude4Sonnet)
+		viper.SetDefault("agents.task.model", models.Claude4Sonnet)
+		viper.SetDefault("agents.title.model", models.Claude4Sonnet)
 		return
 	}
 
diff --git a/internal/llm/models/anthropic.go b/internal/llm/models/anthropic.go
index 156924a47ece2f21cf698051413ae87584cf89f2..9da03a835126956a74ee16888397abb343811ec4 100644
--- a/internal/llm/models/anthropic.go
+++ b/internal/llm/models/anthropic.go
@@ -9,6 +9,8 @@ const (
 	Claude37Sonnet ModelID = "claude-3.7-sonnet"
 	Claude35Haiku  ModelID = "claude-3.5-haiku"
 	Claude3Opus    ModelID = "claude-3-opus"
+	Claude4Opus    ModelID = "claude-4-opus"
+	Claude4Sonnet  ModelID = "claude-4-sonnet"
 )
 
 // https://docs.anthropic.com/en/docs/about-claude/models/all-models
@@ -79,4 +81,31 @@ var AnthropicModels = map[ModelID]Model{
 		DefaultMaxTokens:    4096,
 		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,
+	},
+	Claude4Opus: {
+		ID:                  Claude4Opus,
+		Name:                "Claude 4 Opus",
+		Provider:            ProviderAnthropic,
+		APIModel:            "claude-opus-4-20250514",
+		CostPer1MIn:         15.0,
+		CostPer1MInCached:   18.75,
+		CostPer1MOutCached:  1.50,
+		CostPer1MOut:        75.0,
+		ContextWindow:       200000,
+		DefaultMaxTokens:    4096,
+		SupportsAttachments: true,
+	},
 }