fix: recognize claude cache creation input usage

Author: DouDOU-start

backend/internal/plugin/usage_adapter.go

@@ -52,11 +52,14 @@ func usageSnapshotFromSDK(usage *sdk.Usage) usageSnapshot {
 			snap.OutputTokens += int(metric.Value)
 		case "cached_input_tokens", "cached_input_token", "cache_read_tokens", "cache_read_token":
 			snap.CachedInputTokens += int(metric.Value)
-		case "cache_creation_tokens", "cache_creation_token":
+		case "cache_creation_tokens", "cache_creation_token",
+			"cache_creation_input_tokens", "cache_creation_input_token":
 			snap.CacheCreationTokens += int(metric.Value)
-		case "cache_creation_5m_tokens", "cache_creation_5m_token":
+		case "cache_creation_5m_tokens", "cache_creation_5m_token",
+			"cache_creation_5m_input_tokens", "cache_creation_5m_input_token":
 			snap.CacheCreation5mTokens += int(metric.Value)
-		case "cache_creation_1h_tokens", "cache_creation_1h_token":
+		case "cache_creation_1h_tokens", "cache_creation_1h_token",
+			"cache_creation_1h_input_tokens", "cache_creation_1h_input_token":
 			snap.CacheCreation1hTokens += int(metric.Value)
 		case "reasoning_output_tokens", "reasoning_tokens", "reasoning_token":
 			snap.ReasoningOutputTokens += int(metric.Value)
@@ -147,8 +150,11 @@ func applyUsageCost(snap *usageSnapshot, key string, cost float64, metadata map[
 	case "cached_input", "cached_input_tokens", "cached_input_token", "cache_read_tokens", "cache_read_token":
 		snap.CachedInputCost += cost
 	case "cache_creation", "cache_creation_tokens", "cache_creation_token",
+		"cache_creation_input_tokens", "cache_creation_input_token",
 		"cache_creation_5m", "cache_creation_5m_tokens", "cache_creation_5m_token",
-		"cache_creation_1h", "cache_creation_1h_tokens", "cache_creation_1h_token":
+		"cache_creation_5m_input_tokens", "cache_creation_5m_input_token",
+		"cache_creation_1h", "cache_creation_1h_tokens", "cache_creation_1h_token",
+		"cache_creation_1h_input_tokens", "cache_creation_1h_input_token":
 		snap.CacheCreationCost += cost
 	}
 }
@@ -170,9 +176,12 @@ func applyUsagePrice(snap *usageSnapshot, key string, metadata map[string]string
 	case "cached_input", "cached_input_tokens", "cached_input_token", "cache_read_tokens", "cache_read_token":
 		snap.CachedInputPrice = price
 	case "cache_creation", "cache_creation_tokens", "cache_creation_token",
-		"cache_creation_5m", "cache_creation_5m_tokens", "cache_creation_5m_token":
+		"cache_creation_input_tokens", "cache_creation_input_token",
+		"cache_creation_5m", "cache_creation_5m_tokens", "cache_creation_5m_token",
+		"cache_creation_5m_input_tokens", "cache_creation_5m_input_token":
 		snap.CacheCreationPrice = price
-	case "cache_creation_1h", "cache_creation_1h_tokens", "cache_creation_1h_token":
+	case "cache_creation_1h", "cache_creation_1h_tokens", "cache_creation_1h_token",
+		"cache_creation_1h_input_tokens", "cache_creation_1h_input_token":
 		snap.CacheCreation1hPrice = price
 	}
 }

backend/internal/plugin/usage_adapter_test.go

@@ -0,0 +1,45 @@
+package plugin
+
+import (
+	"math"
+	"testing"
+
+	sdk "github.com/DouDOU-start/airgate-sdk/sdkgo"
+)
+
+func TestUsageSnapshotFromSDK_RecognizesClaudeCacheCreationKeys(t *testing.T) {
+	usage := &sdk.Usage{
+		Metrics: []sdk.UsageMetric{
+			{Key: "input_tokens", Value: 2},
+			{Key: "output_tokens", Value: 163},
+			{Key: "cached_input_tokens", Value: 141779},
+			{Key: "cache_creation_input_tokens", Value: 1756},
+			{Key: "cache_creation_5m_input_tokens", Value: 1756},
+			{Key: "cache_creation_1h_input_tokens", Value: 0},
+		},
+		CostDetails: []sdk.UsageCostDetail{
+			{Key: "input_tokens", AccountCost: 0.00001, Metadata: map[string]string{"unit_price": "5"}},
+			{Key: "cached_input_tokens", AccountCost: 0.0708895, Metadata: map[string]string{"unit_price": "0.5"}},
+			{Key: "cache_creation_5m_input_tokens", AccountCost: 0.010975, Metadata: map[string]string{"unit_price": "6.25"}},
+			{Key: "output_tokens", AccountCost: 0.004075, Metadata: map[string]string{"unit_price": "25"}},
+		},
+	}
+
+	snap := usageSnapshotFromSDK(usage)
+
+	if snap.CacheCreationTokens != 1756 {
+		t.Fatalf("CacheCreationTokens = %d, want 1756", snap.CacheCreationTokens)
+	}
+	if snap.CacheCreation5mTokens != 1756 {
+		t.Fatalf("CacheCreation5mTokens = %d, want 1756", snap.CacheCreation5mTokens)
+	}
+	if snap.CacheCreation1hTokens != 0 {
+		t.Fatalf("CacheCreation1hTokens = %d, want 0", snap.CacheCreation1hTokens)
+	}
+	if math.Abs(snap.CacheCreationCost-0.010975) > 1e-12 {
+		t.Fatalf("CacheCreationCost = %.12f, want %.12f", snap.CacheCreationCost, 0.010975)
+	}
+	if math.Abs(snap.CacheCreationPrice-6.25) > 1e-12 {
+		t.Fatalf("CacheCreationPrice = %.12f, want 6.25", snap.CacheCreationPrice)
+	}
+}