chaos/chaos-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: restore pre-3400 OpenRouter billing semantics

Browse Source
This commit is contained in:
Seefs
2026-03-25 13:24:52 +08:00
parent 9f61407bf0
commit d4a470a638
2 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
service/text_quota.go
+5 -4
View File
@@ -113,8 +113,9 @@ func calculateTextQuotaSummary(ctx *gin.Context, relayInfo *relaycommon.RelayInf
summary.ImageTokens = usage.PromptTokensDetails.ImageTokens summary.ImageTokens = usage.PromptTokensDetails.ImageTokens
summary.AudioTokens = usage.PromptTokensDetails.AudioTokens summary.AudioTokens = usage.PromptTokensDetails.AudioTokens
legacyClaudeDerived := isLegacyClaudeDerivedOpenAIUsage(relayInfo, usage) legacyClaudeDerived := isLegacyClaudeDerivedOpenAIUsage(relayInfo, usage)
isOpenRouter := relayInfo.ChannelMeta != nil && relayInfo.ChannelType == constant.ChannelTypeOpenRouter isOpenRouterClaudeBilling := relayInfo.ChannelMeta != nil &&
isOpenRouterClaudeBilling := isOpenRouter && summary.IsClaudeUsageSemantic relayInfo.ChannelType == constant.ChannelTypeOpenRouter &&
summary.IsClaudeUsageSemantic
if isOpenRouterClaudeBilling { if isOpenRouterClaudeBilling {
summary.PromptTokens -= summary.CacheTokens summary.PromptTokens -= summary.CacheTokens
@@ -199,7 +200,7 @@ func calculateTextQuotaSummary(ctx *gin.Context, relayInfo *relaycommon.RelayInf
var cachedTokensWithRatio decimal.Decimal var cachedTokensWithRatio decimal.Decimal
if !dCacheTokens.IsZero() { if !dCacheTokens.IsZero() {
if !summary.IsClaudeUsageSemantic && !legacyClaudeDerived && !isOpenRouter { if !summary.IsClaudeUsageSemantic && !legacyClaudeDerived {
baseTokens = baseTokens.Sub(dCacheTokens) baseTokens = baseTokens.Sub(dCacheTokens)
} }
cachedTokensWithRatio = dCacheTokens.Mul(dCacheRatio) cachedTokensWithRatio = dCacheTokens.Mul(dCacheRatio)
@@ -208,7 +209,7 @@ func calculateTextQuotaSummary(ctx *gin.Context, relayInfo *relaycommon.RelayInf
var cachedCreationTokensWithRatio decimal.Decimal var cachedCreationTokensWithRatio decimal.Decimal
hasSplitCacheCreationTokens := summary.CacheCreationTokens5m > 0 || summary.CacheCreationTokens1h > 0 hasSplitCacheCreationTokens := summary.CacheCreationTokens5m > 0 || summary.CacheCreationTokens1h > 0
if !dCachedCreationTokens.IsZero() || hasSplitCacheCreationTokens { if !dCachedCreationTokens.IsZero() || hasSplitCacheCreationTokens {
if !summary.IsClaudeUsageSemantic && !legacyClaudeDerived && !isOpenRouter { if !summary.IsClaudeUsageSemantic && !legacyClaudeDerived {
baseTokens = baseTokens.Sub(dCachedCreationTokens) baseTokens = baseTokens.Sub(dCachedCreationTokens)
cachedCreationTokensWithRatio = dCachedCreationTokens.Mul(dCacheCreationRatio) cachedCreationTokensWithRatio = dCachedCreationTokens.Mul(dCacheCreationRatio)
} else { } else {
service/text_quota_test.go
+9 -8
View File
@@ -206,7 +206,7 @@ func TestCalculateTextQuotaSummaryHandlesLegacyClaudeDerivedOpenAIUsage(t *testi
require.Equal(t, 1624, summary.Quota) require.Equal(t, 1624, summary.Quota)
} }
func TestCalculateTextQuotaSummaryDoesNotSubtractOpenRouterCacheTokensFromPrompt(t *testing.T) { func TestCalculateTextQuotaSummarySeparatesOpenRouterCacheReadFromPromptBilling(t *testing.T) {
gin.SetMode(gin.TestMode) gin.SetMode(gin.TestMode)
w := httptest.NewRecorder() w := httptest.NewRecorder()
ctx, _ := gin.CreateTestContext(w) ctx, _ := gin.CreateTestContext(w)
@@ -236,13 +236,14 @@ func TestCalculateTextQuotaSummaryDoesNotSubtractOpenRouterCacheTokensFromPrompt
summary := calculateTextQuotaSummary(ctx, relayInfo, usage) summary := calculateTextQuotaSummary(ctx, relayInfo, usage)
// OpenRouter usage is already normalized. prompt_tokens should stay intact. // OpenRouter OpenAI-format display keeps prompt_tokens as total input,
// quota = 2604 + 2432*0.1 + 383 = 3230.2 => 3230 // but billing still separates normal input from cache read tokens.
// quota = (2604 - 2432) + 2432*0.1 + 383 = 798.2 => 798
require.Equal(t, 2604, summary.PromptTokens) require.Equal(t, 2604, summary.PromptTokens)
require.Equal(t, 3230, summary.Quota) require.Equal(t, 798, summary.Quota)
} }
func TestCalculateTextQuotaSummaryDoesNotSubtractOpenRouterCacheCreationTokensFromPrompt(t *testing.T) { func TestCalculateTextQuotaSummarySeparatesOpenRouterCacheCreationFromPromptBilling(t *testing.T) {
gin.SetMode(gin.TestMode) gin.SetMode(gin.TestMode)
w := httptest.NewRecorder() w := httptest.NewRecorder()
ctx, _ := gin.CreateTestContext(w) ctx, _ := gin.CreateTestContext(w)
@@ -271,10 +272,10 @@ func TestCalculateTextQuotaSummaryDoesNotSubtractOpenRouterCacheCreationTokensFr
summary := calculateTextQuotaSummary(ctx, relayInfo, usage) summary := calculateTextQuotaSummary(ctx, relayInfo, usage)
// OpenRouter usage is already normalized. prompt_tokens should stay intact. // prompt_tokens is still logged as total input, but cache creation is billed separately.
// quota = 2604 + 100*1.25 + 383 = 3112 // quota = (2604 - 100) + 100*1.25 + 383 = 3012
require.Equal(t, 2604, summary.PromptTokens) require.Equal(t, 2604, summary.PromptTokens)
require.Equal(t, 3112, summary.Quota) require.Equal(t, 3012, summary.Quota)
} }
func TestCalculateTextQuotaSummaryKeepsPrePRClaudeOpenRouterBilling(t *testing.T) { func TestCalculateTextQuotaSummaryKeepsPrePRClaudeOpenRouterBilling(t *testing.T) {