Commit a377e990 authored by Wang Lvyuan's avatar Wang Lvyuan
Browse files

fix: remove unused wildcard mapping helper

parent 01ef7340
...@@ -612,13 +612,6 @@ func matchWildcard(pattern, str string) bool { ...@@ -612,13 +612,6 @@ func matchWildcard(pattern, str string) bool {
return matchAntigravityWildcard(pattern, str) return matchAntigravityWildcard(pattern, str)
} }
// matchWildcardMapping 通配符映射匹配(最长优先)
// 如果没有匹配,返回原始字符串
func matchWildcardMapping(mapping map[string]string, requestedModel string) string {
mappedModel, _ := matchWildcardMappingResult(mapping, requestedModel)
return mappedModel
}
func matchWildcardMappingResult(mapping map[string]string, requestedModel string) (string, bool) { func matchWildcardMappingResult(mapping map[string]string, requestedModel string) (string, bool) {
// 收集所有匹配的 pattern,按长度降序排序(最长优先) // 收集所有匹配的 pattern,按长度降序排序(最长优先)
type patternMatch struct { type patternMatch struct {
......
...@@ -43,12 +43,13 @@ func TestMatchWildcard(t *testing.T) { ...@@ -43,12 +43,13 @@ func TestMatchWildcard(t *testing.T) {
} }
} }
func TestMatchWildcardMapping(t *testing.T) { func TestMatchWildcardMappingResult(t *testing.T) {
tests := []struct { tests := []struct {
name string name string
mapping map[string]string mapping map[string]string
requestedModel string requestedModel string
expected string expected string
matched bool
}{ }{
// 精确匹配优先于通配符 // 精确匹配优先于通配符
{ {
...@@ -59,6 +60,7 @@ func TestMatchWildcardMapping(t *testing.T) { ...@@ -59,6 +60,7 @@ func TestMatchWildcardMapping(t *testing.T) {
}, },
requestedModel: "claude-sonnet-4-5", requestedModel: "claude-sonnet-4-5",
expected: "claude-sonnet-4-5-exact", expected: "claude-sonnet-4-5-exact",
matched: true,
}, },
// 最长通配符优先 // 最长通配符优先
...@@ -71,6 +73,7 @@ func TestMatchWildcardMapping(t *testing.T) { ...@@ -71,6 +73,7 @@ func TestMatchWildcardMapping(t *testing.T) {
}, },
requestedModel: "claude-sonnet-4-5", requestedModel: "claude-sonnet-4-5",
expected: "claude-sonnet-4-series", expected: "claude-sonnet-4-series",
matched: true,
}, },
// 单个通配符 // 单个通配符
...@@ -81,6 +84,7 @@ func TestMatchWildcardMapping(t *testing.T) { ...@@ -81,6 +84,7 @@ func TestMatchWildcardMapping(t *testing.T) {
}, },
requestedModel: "claude-opus-4-5", requestedModel: "claude-opus-4-5",
expected: "claude-mapped", expected: "claude-mapped",
matched: true,
}, },
// 无匹配返回原始模型 // 无匹配返回原始模型
...@@ -91,6 +95,7 @@ func TestMatchWildcardMapping(t *testing.T) { ...@@ -91,6 +95,7 @@ func TestMatchWildcardMapping(t *testing.T) {
}, },
requestedModel: "gemini-3-flash", requestedModel: "gemini-3-flash",
expected: "gemini-3-flash", expected: "gemini-3-flash",
matched: false,
}, },
// 空映射返回原始模型 // 空映射返回原始模型
...@@ -99,6 +104,7 @@ func TestMatchWildcardMapping(t *testing.T) { ...@@ -99,6 +104,7 @@ func TestMatchWildcardMapping(t *testing.T) {
mapping: map[string]string{}, mapping: map[string]string{},
requestedModel: "claude-sonnet-4-5", requestedModel: "claude-sonnet-4-5",
expected: "claude-sonnet-4-5", expected: "claude-sonnet-4-5",
matched: false,
}, },
// Gemini 模型映射 // Gemini 模型映射
...@@ -110,14 +116,15 @@ func TestMatchWildcardMapping(t *testing.T) { ...@@ -110,14 +116,15 @@ func TestMatchWildcardMapping(t *testing.T) {
}, },
requestedModel: "gemini-3-flash-preview", requestedModel: "gemini-3-flash-preview",
expected: "gemini-3-pro-high", expected: "gemini-3-pro-high",
matched: true,
}, },
} }
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
result := matchWildcardMapping(tt.mapping, tt.requestedModel) result, matched := matchWildcardMappingResult(tt.mapping, tt.requestedModel)
if result != tt.expected { if result != tt.expected || matched != tt.matched {
t.Errorf("matchWildcardMapping(%v, %q) = %q, want %q", tt.mapping, tt.requestedModel, result, tt.expected) t.Errorf("matchWildcardMappingResult(%v, %q) = (%q, %v), want (%q, %v)", tt.mapping, tt.requestedModel, result, matched, tt.expected, tt.matched)
} }
}) })
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment