From 75e433e4b823ed2f52cfd4cca4f5d3ad79c709d2 Mon Sep 17 00:00:00 2001
From: qiangmzsx <qiangmzsx@gmail.com>
Date: Sun, 1 Mar 2026 10:34:33 +0800
Subject: [PATCH] add test

---
 pkg/providers/factory_provider_test.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/pkg/providers/factory_provider_test.go b/pkg/providers/factory_provider_test.go
index e0c0eddef..b0d0780ed 100644
--- a/pkg/providers/factory_provider_test.go
+++ b/pkg/providers/factory_provider_test.go
@@ -34,6 +34,12 @@ func TestExtractProtocol(t *testing.T) {
 			wantProtocol: "anthropic",
 			wantModelID:  "claude-sonnet-4.6",
 		},
+		{
+			name:         "claude with prefix",
+			model:        "claude/claude-sonnet-4.6",
+			wantProtocol: "claude",
+			wantModelID:  "claude-sonnet-4.6",
+		},
 		{
 			name:         "no prefix - defaults to openai",
 			model:        "gpt-4o",
@@ -154,6 +160,25 @@ func TestCreateProviderFromConfig_Anthropic(t *testing.T) {
 	}
 }
 
+func TestCreateProviderFromConfig_Claudue(t *testing.T) {
+	cfg := &config.ModelConfig{
+		ModelName: "test-anthropic",
+		Model:     "claude/claude-sonnet-4.6",
+		APIKey:    "test-key",
+	}
+
+	provider, modelID, err := CreateProviderFromConfig(cfg)
+	if err != nil {
+		t.Fatalf("CreateProviderFromConfig() error = %v", err)
+	}
+	if provider == nil {
+		t.Fatal("CreateProviderFromConfig() returned nil provider")
+	}
+	if modelID != "claude-sonnet-4.6" {
+		t.Errorf("modelID = %q, want %q", modelID, "claude-sonnet-4.6")
+	}
+}
+
 func TestCreateProviderFromConfig_Antigravity(t *testing.T) {
 	cfg := &config.ModelConfig{
 		ModelName: "test-antigravity",