refactor(antigravity): 简化模型映射逻辑,支持前缀匹配

- 删除精确映射表 antigravityModelMapping,统一使用前缀映射
- 前缀映射支持模型版本号变化(如 -20251111, -thinking, -preview)
- 简化 IsModelSupported 函数,所有 claude-/gemini- 前缀模型都支持
- 添加跨协议测试用例:Claude 端点调用 Gemini 模型、Gemini 端点调用 Claude 模型
This commit is contained in:
song
2025-12-31 21:16:32 +08:00
parent 312cc00d21
commit 7f7bbdf677
3 changed files with 96 additions and 59 deletions

View File

@@ -515,24 +515,10 @@ func (s *GatewayService) isModelSupportedByAccount(account *Account, requestedMo
}
// IsAntigravityModelSupported 检查 Antigravity 平台是否支持指定模型
// 所有 claude- 和 gemini- 前缀的模型都能通过映射或透传支持
func IsAntigravityModelSupported(requestedModel string) bool {
// 直接支持的模型
if antigravitySupportedModels[requestedModel] {
return true
}
// 可映射的模型
if _, ok := antigravityModelMapping[requestedModel]; ok {
return true
}
// Gemini 前缀透传
if strings.HasPrefix(requestedModel, "gemini-") {
return true
}
// Claude 模型支持(通过默认映射到 claude-sonnet-4-5
if strings.HasPrefix(requestedModel, "claude-") {
return true
}
return false
return strings.HasPrefix(requestedModel, "claude-") ||
strings.HasPrefix(requestedModel, "gemini-")
}
// GetAccessToken 获取账号凭证