lywsvip/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-09-03 06:35:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(usage): harden canonical token normalization

Browse Source
This commit is contained in:
Luis Pater
2026-07-24 01:18:18 +08:00
parent 885ffac42c
commit 42f36b94e0
5 changed files with 254 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/redisqueue/plugin_test.go
View File

@@ -144,7 +144,7 @@ func TestUsageQueuePluginPayloadDefaultsGenerateTrueWhenOmitted(t *testing.T) {
})
}
func TestUsageQueuePluginMarksCanonicalZeroCacheRead(t *testing.T) {
func TestUsageQueuePluginPreservesLegacyCachedOnlyUsage(t *testing.T) {
withEnabledQueue(t, func() {
ctx := internallogging.WithResponseStatusHolder(context.Background())
internallogging.SetResponseStatus(ctx, http.StatusOK)
@@ -153,21 +153,16 @@ func TestUsageQueuePluginMarksCanonicalZeroCacheRead(t *testing.T) {
Provider: "openai",
Model: "gpt-5.4",
Detail: coreusage.Detail{
CachedTokens: 13,
CacheReadTokens: 0,
CachedTokens: 13,
},
})
payload := popSinglePayload(t)
requireTokensBoolField(t, payload, "cache_read_tokens_present", true)
tokens := requireTokensPayload(t, payload)
var cacheReadTokens int64
if errUnmarshal := json.Unmarshal(tokens["cache_read_tokens"], &cacheReadTokens); errUnmarshal != nil {
t.Fatalf("unmarshal cache_read_tokens: %v", errUnmarshal)
}
if cacheReadTokens != 0 {
t.Fatalf("cache_read_tokens = %d, want 0", cacheReadTokens)
}
requireIntField(t, tokens, "cache_read_tokens", 13)
requireIntField(t, tokens, "total_tokens", 13)
requireTokenBreakdown(t, payload, coreusage.TokenAccountingQualityUnclassified, 13)
})
}