feat: sync deployed v1.0.0-rc.38-custom.42
Sync production source including routing, billing, Astra detection and public uptime. Update custom build instructions and license packaging.
This commit is contained in:
1 parent
aa6c3f50f4
commit
ceddfd3d6b
828 files changed
+122253
-13702
No files matched your search
@@ -49,6 +49,33 @@ func TestNewOpenAIChatBillingUsageRequiresTokenContent(t *testing.T) {
|
||||
assert.Equal(t, 1, billingUsage.OpenAIUsage.PromptTokens)
|
||||
}
|
||||
|
||||
func TestImageCacheDetailsSurviveUsageSnapshots(t *testing.T) {
|
||||
var original Usage
|
||||
require.NoError(t, kitutil.Unmarshal([]byte(`{"input_tokens":1000,"input_tokens_details":{"cached_tokens":300,"image_tokens":600,"cached_tokens_details":{"image_tokens":200,"text_tokens":100}}}`), &original))
|
||||
billing := NewOpenAIResponsesBillingUsage(&original)
|
||||
require.NotNil(t, billing)
|
||||
*original.InputTokensDetails.CachedTokensDetails.ImageTokens = 99
|
||||
canonical, ok := billing.CanonicalUsage()
|
||||
require.True(t, ok)
|
||||
require.NotNil(t, canonical.PromptTokensDetails.CachedTokensDetails)
|
||||
assert.Equal(t, 200, *canonical.PromptTokensDetails.CachedTokensDetails.ImageTokens)
|
||||
*canonical.PromptTokensDetails.CachedTokensDetails.ImageTokens = 42
|
||||
assert.Equal(t, 200, *billing.OpenAIUsage.InputTokensDetails.CachedTokensDetails.ImageTokens)
|
||||
|
||||
var incoming Usage
|
||||
require.NoError(t, kitutil.Unmarshal([]byte(`{"input_tokens_details":{"cached_tokens_details":{"image_tokens":0}}}`), &incoming))
|
||||
merged := MergeUsageNonZero(canonical, &incoming)
|
||||
require.NotNil(t, merged.InputTokensDetails.CachedTokensDetails.ImageTokens)
|
||||
assert.Zero(t, *merged.InputTokensDetails.CachedTokensDetails.ImageTokens)
|
||||
assert.Equal(t, 100, *merged.InputTokensDetails.CachedTokensDetails.TextTokens)
|
||||
*incoming.InputTokensDetails.CachedTokensDetails.ImageTokens = 9
|
||||
assert.Zero(t, *merged.InputTokensDetails.CachedTokensDetails.ImageTokens)
|
||||
encoded, err := kitutil.Marshal(merged)
|
||||
require.NoError(t, err)
|
||||
assert.Contains(t, string(encoded), `"cached_tokens_details":{"text_tokens":100,"image_tokens":0}`)
|
||||
assert.NotContains(t, string(encoded), `"audio_tokens":null`)
|
||||
}
|
||||
|
||||
func TestNewEstimatedGeminiChatBillingUsage(t *testing.T) {
|
||||
billingUsage := NewEstimatedGeminiChatBillingUsage(&Usage{
|
||||
PromptTokens: 11,
|
||||
|
||||
Reference in new issue
Block a user