fix(routing): 对齐缓存亲和力硬规则与审计

apps/api/internal/store/tasks_runtime.go

@@ -996,14 +996,16 @@ func (s *Store) FinishTaskAttempt(ctx context.Context, input FinishTaskAttemptIn
 	if statusCode == 0 {
 		statusCode = taskAttemptMetricInt(input.Metrics, "statusCode")
 	}
+	usageJSON, _ := json.Marshal(sanitizeJSONForStorage(minimalTaskAttemptUsage(input.Usage)))
+	metricsJSON, _ := json.Marshal(sanitizeJSONForStorage(minimalTaskAttemptMetrics(input.Metrics)))
 	_, err := s.pool.Exec(ctx, `
 UPDATE gateway_task_attempts
 SET status = $2::text,
     retryable = $3,
     request_id = NULLIF($4::text, ''),
     status_code = NULLIF($5::int, 0),
-    usage = '{}'::jsonb,
-    metrics = '{}'::jsonb,
+    usage = $11::jsonb,
+    metrics = $12::jsonb,
     response_snapshot = '{}'::jsonb,
     pricing_snapshot = '{}'::jsonb,
     request_fingerprint = NULL,
@@ -1024,10 +1026,45 @@ WHERE id = $1::uuid`,
 		input.ResponseDurationMS,
 		input.ErrorCode,
 		truncateUTF8Bytes(input.ErrorMessage, 2048),
+		usageJSON,
+		metricsJSON,
 	)
 	return err
 }
 
+func minimalTaskAttemptUsage(usage map[string]any) map[string]any {
+	return whitelistedTaskAttemptMap(usage, []string{
+		"inputTokens", "promptTokens", "input_tokens", "prompt_tokens",
+		"cachedInputTokens", "cachedPromptTokens", "cached_input_tokens", "cached_tokens",
+		"cachedInputTokensKnown",
+		"outputTokens", "completionTokens", "output_tokens", "completion_tokens",
+		"totalTokens", "total_tokens",
+	})
+}
+
+func minimalTaskAttemptMetrics(metrics map[string]any) map[string]any {
+	return whitelistedTaskAttemptMap(metrics, []string{
+		"platformPriority", "currentPriority", "loadRatio", "loadAvoided",
+		"cacheAffinityKey", "cacheAdjustedPriority", "cacheAffinitySamples",
+		"cacheAffinityScore", "cacheAffinityConfidence", "cacheAffinityHitRatio",
+		"cacheAffinityEMAHitRatio", "cacheAffinityLastHitRatio",
+		"cacheAffinityCachedInputTokens", "cacheAffinityBoost",
+		"cacheAffinityApplied", "cacheAffinityMatched",
+		"cacheAffinityCandidateCount", "cacheAffinityMatchedPrefixDepth",
+		"cacheAffinityOverrideReason",
+	})
+}
+
+func whitelistedTaskAttemptMap(input map[string]any, keys []string) map[string]any {
+	out := map[string]any{}
+	for _, key := range keys {
+		if value, ok := input[key]; ok {
+			out[key] = value
+		}
+	}
+	return out
+}
+
 func (s *Store) FinishTaskSuccess(ctx context.Context, input FinishTaskSuccessInput) (GatewayTask, error) {
 	resultReport := sanitizeJSONForStorageWithReport(minimalTaskResult(input.Result))
 	if resultReport.BinaryCount > 0 {
@@ -1041,6 +1078,8 @@ func (s *Store) FinishTaskSuccess(ctx context.Context, input FinishTaskSuccessIn
 	billingsJSON, _ := json.Marshal(sanitizeJSONForStorage(input.Billings))
 	usageJSON, _ := json.Marshal(sanitizeJSONForStorage(emptyObjectIfNil(input.Usage)))
 	metricsJSON, _ := json.Marshal(sanitizeJSONForStorage(emptyObjectIfNil(input.Metrics)))
+	attemptUsageJSON, _ := json.Marshal(sanitizeJSONForStorage(minimalTaskAttemptUsage(input.Usage)))
+	attemptMetricsJSON, _ := json.Marshal(sanitizeJSONForStorage(minimalTaskAttemptMetrics(input.Metrics)))
 	billingSummaryJSON, _ := json.Marshal(sanitizeJSONForStorage(emptyObjectIfNil(input.BillingSummary)))
 	pricingSnapshotJSON, _ := json.Marshal(sanitizeJSONForStorage(emptyObjectIfNil(input.PricingSnapshot)))
 	finalChargeAmount := strings.TrimSpace(input.FinalChargeAmountText)
@@ -1056,8 +1095,8 @@ SET status = 'succeeded',
     retryable = false,
     request_id = NULLIF($2, ''),
     status_code = NULL,
-    usage = '{}'::jsonb,
-    metrics = '{}'::jsonb,
+    usage = $6::jsonb,
+    metrics = $7::jsonb,
     response_snapshot = '{}'::jsonb,
     pricing_snapshot = '{}'::jsonb,
     request_fingerprint = NULL,
@@ -1072,6 +1111,7 @@ SET status = 'succeeded',
 WHERE id = $1::uuid`,
 				input.AttemptID, input.RequestID,
 				nullableTime(input.ResponseStartedAt), nullableTime(input.ResponseFinishedAt), input.ResponseDurationMS,
+				attemptUsageJSON, attemptMetricsJSON,
 			); err != nil {
 				return err
 			}