45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package common
|
||
|
||
import "testing"
|
||
|
||
// TestTotalInputTokens 验证「含缓存总输入」归一化口径：
|
||
// OpenAI/Gemini 口径（CacheIncludedInPrompt=true，prompt 已含缓存）原样返回；
|
||
// Claude 口径（false，input_tokens 与缓存三项并列）补加 cache_read + cache_creation。
|
||
func TestTotalInputTokens(t *testing.T) {
|
||
// OpenAI 口径：prompt=2006 已含 cached=1920，总输入即 2006，不得再叠加
|
||
openai := &Usage{
|
||
PromptTokens: 2006,
|
||
CompletionTokens: 46,
|
||
CacheIncludedInPrompt: true,
|
||
PromptTokensDetails: &TokenDetails{CachedTokens: 1920},
|
||
}
|
||
if got := openai.TotalInputTokens(); got != 2006 {
|
||
t.Errorf("TotalInputTokens() = %d, want 2006 (included 口径不叠加)", got)
|
||
}
|
||
|
||
// Claude 口径：input=204 + cache_read=1800 + cache_creation=248 = 2252
|
||
claude := &Usage{
|
||
PromptTokens: 204,
|
||
CompletionTokens: 503,
|
||
PromptTokensDetails: &TokenDetails{
|
||
CachedTokens: 1800,
|
||
CachedCreationTokens: 248,
|
||
},
|
||
}
|
||
if got := claude.TotalInputTokens(); got != 2252 {
|
||
t.Errorf("TotalInputTokens() = %d, want 2252 (excluded 口径补加缓存)", got)
|
||
}
|
||
|
||
// Claude 口径但无缓存明细：原样返回
|
||
plain := &Usage{PromptTokens: 100}
|
||
if got := plain.TotalInputTokens(); got != 100 {
|
||
t.Errorf("TotalInputTokens() = %d, want 100", got)
|
||
}
|
||
|
||
// nil 安全
|
||
var nilUsage *Usage
|
||
if got := nilUsage.TotalInputTokens(); got != 0 {
|
||
t.Errorf("TotalInputTokens() on nil = %d, want 0", got)
|
||
}
|
||
}
|