Merge pull request #30 from jdlms/update/config-GPT4.1

Kujtim Hoxha created 7 months ago

OpenAI model config changed to GPT-4.1

Change summary

internal/config/config.go      |  4 ++--
internal/config/config_test.go |  6 +++---
internal/llm/models/models.go  | 16 ++++++++--------
3 files changed, 13 insertions(+), 13 deletions(-)

Detailed changes

internal/config/config.go

@@ -116,8 +116,8 @@ func Load(debug bool) error {
 		viper.SetDefault("providers.openai.apiKey", os.Getenv("OPENAI_API_KEY"))
 		viper.SetDefault("providers.openai.enabled", true)
 		if !defaultModelSet {
-			viper.SetDefault("model.coder", models.GPT4o)
-			viper.SetDefault("model.task", models.GPT4o)
+			viper.SetDefault("model.coder", models.GPT41)
+			viper.SetDefault("model.task", models.GPT41)
 			defaultModelSet = true
 		}
 	}

internal/config/config_test.go

@@ -223,7 +223,7 @@ func TestLoad(t *testing.T) {
 				anthropicKey:  "",
 				openaiKey:     "test-key",
 				geminiKey:     "test-key",
-				expectedModel: models.GPT4o,
+				expectedModel: models.GPT41,
 			},
 			{
 				name:          "fallback to gemini when no others",
@@ -237,9 +237,9 @@ func TestLoad(t *testing.T) {
 				anthropicKey:     "test-key",
 				openaiKey:        "test-key",
 				geminiKey:        "test-key",
-				explicitModel:    models.GPT4o,
+				explicitModel:    models.GPT41,
 				useExplicitModel: true,
-				expectedModel:    models.GPT4o,
+				expectedModel:    models.GPT41,
 			},
 		}
 

internal/llm/models/models.go

@@ -23,7 +23,7 @@ const (
 	Claude3Haiku   ModelID = "claude-3-haiku"
 	Claude37Sonnet ModelID = "claude-3.7-sonnet"
 	// OpenAI
-	GPT4o ModelID = "gpt-4o"
+	GPT41 ModelID = "gpt-4.1"
 
 	// GEMINI
 	GEMINI25      ModelID = "gemini-2.5"
@@ -78,15 +78,15 @@ var SupportedModels = map[ModelID]Model{
 	},
 
 	// OpenAI
-	GPT4o: {
-		ID:                 GPT4o,
-		Name:               "GPT-4o",
+	GPT41: {
+		ID:                 GPT41,
+		Name:               "GPT-4.1",
 		Provider:           ProviderOpenAI,
-		APIModel:           "gpt-4o",
-		CostPer1MIn:        2.50,
-		CostPer1MInCached:  1.25,
+		APIModel:           "gpt-4.1",
+		CostPer1MIn:        2.00,
+		CostPer1MInCached:  0.50,
 		CostPer1MOutCached: 0,
-		CostPer1MOut:       10.00,
+		CostPer1MOut:       8.00,
 	},
 
 	// GEMINI