mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-15 23:22:59 +08:00
feat(platform): 接入 Command Code 平台
转发： - 平台清单与 provider profile 新增 command_code：按模型分流，Chat Completions / Responses 基址 https://api.commandcode.ai/provider/v1， Anthropic 基址 https://api.commandcode.ai/provider；内置规则 claude-* → Anthropic、gpt-* → Responses（也支持 Chat Completions）。 - 原生 Anthropic 端点的版本感知拼接改按分流方式判断，转发与连接测试共用 （Anthropic 基址带 /v1 时连接测试不再误报为配置错误）。 按模型的协议集合（模型支持入站协议时直通，免去协议转换）： - 分流规则可带协议集合 protocols（首项为首选协议）：入站协议在集合中时同 协议直通，否则走首选协议；只有一个协议时与原规则相同，OpenCode 行为不变。 - profile 新增 model_catalog：Command Code 的 /models 为每个模型给出 supported_endpoints。官方默认地址的目录按地址全进程共享，自定义上游按账号 隔离；目录请求的 UA 收敛与账号请求头覆写与转发一致。目录缓存 30 分钟、后台 刷新，失败退避 5 分钟；从未加载成功时首批请求最多等 2 秒，冷启动也按目录 分流。账号规则未命中的模型按目录取协议集合；目录不可用时回落内置规则（账号 未配置规则时）或 Chat Completions。 - 三个入站入口共用的 resolveUpstreamProtocol 增加目录参数。 - 前端分流规则编辑器可为每条规则勾选"也支持"的协议；提供模型目录的平台， 兜底行改为说明按上游模型列表选协议。 连接测试： - 按模型分流的平台共用一条测试路径（协议与网关同一判定，含模型目录；默认 测试模型取自 profile），不再落入 Claude 测试路径。 - 首个分支由"国产厂商"改为"按入站协议分流"；adaptive 测试跳过 profile 未 提供的 Anthropic 端点。现有国产厂商行为不变。 - 连接测试是 OpenCode 专属分流入口的最后一个调用方，改走统一判定后删除 ResolveOpenCodeGoUpstreamProtocol、openCodeGoNativeProtocol，以及此前就只剩 测试使用的 OpenCodeGoModelProtocol；路由矩阵的旧行为基准改为测试内独立复述。 额度与余额（/alpha/whoami、billing/credits、billing/subscriptions、 usage/summary）： - 额度探测写入 5 小时 / 每周窗口（限额套餐）与还原出的月度窗口，同时写入 积分余额；余额探测只取积分。仅对官方主机上的 API Key 账号生效。 - 订阅周期或本周期用量取数失败时保留上一次的月度快照，不当作没有月度窗口 清掉；窗口尚未开始（没有重置时间）时清掉上一个窗口的重置时间。 - whoami 取不到时探测整体失败，不按个人账户继续（组织账号会查到成员的个人 钱包，并据此按余额停调）。 - 阈值停调与定时检测纳入 Command Code；有充值积分时上游跳过窗口检查，窗口 快照不参与停调与冷却；积分低于余额阈值时临时停调。 - 上游计费探测的官方域名加入 commandcode.ai，不把 Key 发往官方主机上并不 存在的中转计费接口。 上游错误（依据 commandcode.ai/docs/troubleshooting/errors，按文案分类， 402 / 403 / 429 同样处理）： - 5 小时 / 每周窗口用满：带原因的临时停调，到文案中的 "Resets in …"，解析 不出时回落额度快照；充值积分不受窗口限制，额度 / 余额探测发现充值积分后 只解除这一类停调。 - 积分耗尽（含 usage_exceeded）：可恢复的临时停调，积分恢复后由周期检测 解除，不再永久置为 error。 - 组织花费上限：文案指明模型（"set for <模型>"）且请求模型已知时只写模型级 限流；成员总额度或作用域不明时整个账号冷却，最长一小时。 - 普通频率限制走默认 429 逻辑，不再冷却到窗口重置点。 前端： - 账号列表显示额度窗口与积分余额，阈值设置增加 command_code。单元格可见性 改为按账号判定，与后端一致：只对官方主机上的 API Key 账号显示，自定义中转 显示占位符，不再每次打开列表都探测出一行报错。 - 使用官网的官方标志（圆角方块内镂空 ⌘）；配色为中性灰（品牌黑色已由 Grok 使用）。 测试基准中原先用作"仅服务端登记"示例的 command_code 改为虚构的 acme_router； 新增的前端测试加入 CI 前端关键测试列表。 Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
SavitarC
co-authored by
Claude Opus 5.5
Claude Opus 5.5
parent
7ffa38a8ae
commit
e2939cd8fa
@@ -0,0 +1,176 @@
|
||||
//go:build unit
|
||||
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/tidwall/gjson"
|
||||
)
|
||||
|
||||
// commandCodeTestAccount 只带 api_key：端点与分流规则全部来自 provider profile。
|
||||
func commandCodeTestAccount(id int64) *Account {
|
||||
return &Account{
|
||||
ID: id,
|
||||
Name: "cc",
|
||||
Platform: PlatformCommandCode,
|
||||
Type: AccountTypeAPIKey,
|
||||
Status: StatusActive,
|
||||
Concurrency: 1,
|
||||
Credentials: map[string]any{"api_key": "user_test_key"},
|
||||
}
|
||||
}
|
||||
|
||||
func TestCommandCodeProfileRoutesByModel(t *testing.T) {
|
||||
account := commandCodeTestAccount(1)
|
||||
require.True(t, account.IsMultiProtocolAPIKey())
|
||||
require.True(t, account.routesByModel())
|
||||
require.False(t, account.RoutesProtocolByInbound())
|
||||
require.Equal(t, APIProtocolAdaptive, account.GetAPIProtocol())
|
||||
require.Equal(t, DefaultCommandCodeBaseURL, account.GetOpenAIBaseURL())
|
||||
require.Equal(t, DefaultCommandCodeAnthropicBaseURL, account.GetCNProtocolBaseURL(APIProtocolAnthropic))
|
||||
require.Equal(t, DefaultCommandCodeBaseURL, account.GetCNProtocolBaseURL(APIProtocolResponses))
|
||||
|
||||
// 未取到模型目录时按内置规则：Claude 只走 Anthropic；GPT 支持 Responses 与 Chat
|
||||
// Completions（同协议直通，Anthropic 入站走首选 Responses）；其余只保证 Chat Completions。
|
||||
for _, inbound := range []string{APIProtocolChatCompletions, APIProtocolResponses, APIProtocolAnthropic} {
|
||||
require.Equal(t, APIProtocolAnthropic, resolveUpstreamProtocol(account, inbound, "claude-sonnet-4-6", nil), inbound)
|
||||
require.Equal(t, APIProtocolChatCompletions, resolveUpstreamProtocol(account, inbound, "deepseek/deepseek-v4-flash", nil), inbound)
|
||||
}
|
||||
require.Equal(t, APIProtocolResponses, resolveUpstreamProtocol(account, APIProtocolResponses, "gpt-5.5", nil))
|
||||
require.Equal(t, APIProtocolChatCompletions, resolveUpstreamProtocol(account, APIProtocolChatCompletions, "gpt-5.5", nil))
|
||||
require.Equal(t, APIProtocolResponses, resolveUpstreamProtocol(account, APIProtocolAnthropic, "gpt-5.5", nil))
|
||||
|
||||
// 显式协议优先于内置规则。
|
||||
account.Credentials["api_protocol"] = APIProtocolChatCompletions
|
||||
require.Equal(t, APIProtocolChatCompletions, resolveUpstreamProtocol(account, APIProtocolAnthropic, "claude-sonnet-4-6", nil))
|
||||
}
|
||||
|
||||
func TestCommandCodeNativeAnthropicTargetURL(t *testing.T) {
|
||||
svc := &OpenAIGatewayService{}
|
||||
targetURL, err := svc.nativeAnthropicTargetURL(commandCodeTestAccount(2))
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, "https://api.commandcode.ai/provider/v1/messages", targetURL)
|
||||
}
|
||||
|
||||
func TestAccountTestService_CommandCodeClaudeUsesAnthropicMessages(t *testing.T) {
|
||||
account := commandCodeTestAccount(501)
|
||||
svc, upstream := adaptiveCNAccountTestService(account, adaptiveCNAnthropicTestResponse())
|
||||
c, recorder := newTestContext()
|
||||
|
||||
err := svc.TestAccountConnection(c, account.ID, "claude-sonnet-4-6", "hi", AccountTestModeDefault)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Len(t, upstream.requests, 1)
|
||||
require.Equal(t, "https://api.commandcode.ai/provider/v1/messages", upstream.requests[0].URL.String())
|
||||
require.Equal(t, "user_test_key", upstream.requests[0].Header.Get("x-api-key"))
|
||||
require.Contains(t, recorder.Body.String(), `"type":"test_complete"`)
|
||||
}
|
||||
|
||||
// 连接测试与转发共用端点拼接：按模型分流的平台，Anthropic 基址带 /v1 也合法。
|
||||
func TestAccountTestService_CommandCodeAnthropicBaseWithVersion(t *testing.T) {
|
||||
account := commandCodeTestAccount(504)
|
||||
account.Credentials["api_base_urls"] = map[string]any{APIProtocolAnthropic: "https://api.commandcode.ai/provider/v1"}
|
||||
svc, upstream := adaptiveCNAccountTestService(account, adaptiveCNAnthropicTestResponse())
|
||||
c, recorder := newTestContext()
|
||||
|
||||
err := svc.TestAccountConnection(c, account.ID, "claude-sonnet-4-6", "hi", AccountTestModeDefault)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Len(t, upstream.requests, 1)
|
||||
require.Equal(t, "https://api.commandcode.ai/provider/v1/messages", upstream.requests[0].URL.String())
|
||||
require.Contains(t, recorder.Body.String(), `"type":"test_complete"`)
|
||||
|
||||
target, err := (&OpenAIGatewayService{cfg: rawChatCompletionsTestConfig()}).nativeAnthropicTargetURL(account)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, upstream.requests[0].URL.String(), target)
|
||||
}
|
||||
|
||||
// 连接测试与网关同一判定：模型目录决定协议时，测试也按目录选择端点。
|
||||
func TestAccountTestService_CommandCodeUsesModelCatalog(t *testing.T) {
|
||||
base := fmt.Sprintf("http://cc-test-%d.example/provider/v1", time.Now().UnixNano())
|
||||
account := commandCodeTestAccount(505)
|
||||
account.Credentials["api_base_urls"] = map[string]any{
|
||||
APIProtocolChatCompletions: base,
|
||||
APIProtocolResponses: base,
|
||||
}
|
||||
url := buildOpenAIModelsURL(base)
|
||||
upstreamModelProtocols.store(modelProtocolCatalogKey(account, base, url),
|
||||
map[string][]string{"vendor/responses-only": {APIProtocolResponses}}, nil, time.Now())
|
||||
svc, upstream := adaptiveCNAccountTestService(account, adaptiveCNResponsesTestResponse())
|
||||
svc.openaiGatewayService = &OpenAIGatewayService{cfg: rawChatCompletionsTestConfig(), httpUpstream: upstream}
|
||||
c, recorder := newTestContext()
|
||||
|
||||
err := svc.TestAccountConnection(c, account.ID, "vendor/responses-only", "hi", AccountTestModeDefault)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Len(t, upstream.requests, 1)
|
||||
require.Equal(t, base+"/responses", upstream.requests[0].URL.String())
|
||||
require.Contains(t, recorder.Body.String(), `"type":"test_complete"`)
|
||||
}
|
||||
|
||||
func TestAccountTestService_CommandCodeGPTUsesResponses(t *testing.T) {
|
||||
account := commandCodeTestAccount(502)
|
||||
svc, upstream := adaptiveCNAccountTestService(account, adaptiveCNResponsesTestResponse())
|
||||
c, recorder := newTestContext()
|
||||
|
||||
err := svc.TestAccountConnection(c, account.ID, "gpt-5.5", "hi", AccountTestModeDefault)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Len(t, upstream.requests, 1)
|
||||
require.Equal(t, "https://api.commandcode.ai/provider/v1/responses", upstream.requests[0].URL.String())
|
||||
require.Equal(t, "Bearer user_test_key", upstream.requests[0].Header.Get("Authorization"))
|
||||
require.Contains(t, recorder.Body.String(), `"type":"test_complete"`)
|
||||
}
|
||||
|
||||
func TestAccountTestService_CommandCodeEmptyModelUsesProfileDefault(t *testing.T) {
|
||||
account := commandCodeTestAccount(503)
|
||||
svc, upstream := adaptiveCNAccountTestService(account, adaptiveCNChatTestResponse())
|
||||
c, recorder := newTestContext()
|
||||
|
||||
err := svc.TestAccountConnection(c, account.ID, "", "hi", AccountTestModeDefault)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Len(t, upstream.requests, 1)
|
||||
require.Equal(t, "https://api.commandcode.ai/provider/v1/chat/completions", upstream.requests[0].URL.String())
|
||||
require.Equal(t, DefaultCommandCodeTestModel, gjson.GetBytes(upstream.lastBody, "model").String())
|
||||
require.Contains(t, recorder.Body.String(), `"type":"test_complete"`)
|
||||
}
|
||||
|
||||
// 仅在 profile 登记、按入站协议分流的新供应商：adaptive 测试走原生端点探测，
|
||||
// 跳过 profile 未提供的 Anthropic 端点，不再落入 Claude 测试路径。
|
||||
func TestAccountTestService_ByInboundProviderSkipsMissingNativeEndpoints(t *testing.T) {
|
||||
const platform = "test_inbound_only_chat"
|
||||
providerProfiles[platform] = &ProviderProfile{
|
||||
Platform: platform, DefaultMode: AccountModePayG, Routing: ProviderRoutingByInbound,
|
||||
DefaultTestModel: "inbound-test-model",
|
||||
Modes: map[string]ProviderEndpoints{AccountModePayG: {BaseURLs: map[string]string{
|
||||
APIProtocolChatCompletions: "https://inbound.example.com/v1",
|
||||
}}},
|
||||
}
|
||||
t.Cleanup(func() { delete(providerProfiles, platform) })
|
||||
|
||||
account := &Account{
|
||||
ID: 601, Name: "inbound", Platform: platform, Type: AccountTypeAPIKey, Status: StatusActive, Concurrency: 1,
|
||||
Credentials: map[string]any{"api_key": "sk-inbound", "api_protocol": APIProtocolAdaptive},
|
||||
}
|
||||
svc, upstream := adaptiveCNAccountTestService(account, adaptiveCNChatTestResponse())
|
||||
c, recorder := newTestContext()
|
||||
|
||||
err := svc.TestAccountConnection(c, account.ID, "", "hi", AccountTestModeDefault)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Len(t, upstream.requests, 1)
|
||||
require.Equal(t, "https://inbound.example.com/v1/chat/completions", upstream.requests[0].URL.String())
|
||||
require.Equal(t, "inbound-test-model", gjson.GetBytes(upstream.lastBody, "model").String())
|
||||
require.Contains(t, recorder.Body.String(), `"type":"test_complete"`)
|
||||
}
|
||||
|
||||
// 官方主机不做上游计费探测：中转站的计费接口在官方 API 上不存在，只会把 Key 发往官方主机。
|
||||
func TestCommandCodeOfficialHostSkipsUpstreamBillingProbe(t *testing.T) {
|
||||
require.True(t, upstreamBillingProbeTargetIsOfficialAPI(DefaultCommandCodeBaseURL))
|
||||
require.True(t, upstreamBillingProbeTargetIsOfficialAPI(DefaultCommandCodeAnthropicBaseURL))
|
||||
}
|
||||
Reference in New Issue
Block a user